class SlackSmartBot
  module Commands
    module General
      module AI
        module OpenAI
          def open_ai_edit_image(message, files)
            save_stats(__method__)
            get_personal_settings()
            @ai_open_ai, message_connect = SlackSmartBot::AI::OpenAI.connect(@ai_open_ai, config, @personal_settings)
            respond message_connect if message_connect
            user = Thread.current[:user]
            if !@ai_open_ai[user.name].nil? and !@ai_open_ai[user.name][:client].nil?
              @ai_open_ai_image ||= {}
              @ai_open_ai_image[user.name] ||= []
              react :art
              begin
                @ai_open_ai_image[user.name] = [] if !files.nil? and files.size == 1
                if files.nil? or files.size != 1
                  respond "*OpenAI*: Sorry, I need an image to edit. Please upload an image and try again."
                else
                  require "nice_http"
                  image = "#{config.path}/tmp/#{user.name}_#{@ai_open_ai_image[user.name].object_id}.png"
                  http = NiceHttp.new(host: "https://files.slack.com", headers: { "Authorization" => "Bearer #{config.token}" })
                  res = http.get(files[0].url_private_download, save_data: image)
                  success, res = SlackSmartBot::AI::OpenAI.send_image_edit(@ai_open_ai[user.name].client, image, message, size: @ai_open_ai[user.name][:image_size])

                  if success
                    urls = res
                    urls = [urls] if urls.is_a?(String)
                    if urls.nil? or urls.empty?
                      respond "*OpenAI*: Sorry, I'm having some problems. OpenAI was not able to generate an image."
                    else
                      if @ai_open_ai_image[user.name].empty?
                        session_name = "Edit"
                      else
                        session_name = @ai_open_ai_image[user.name].first[0..29]
                      end
                      messagersp = "OpenAI Session: _<#{session_name}...>_ (id:#{@ai_open_ai_image[user.name].object_id})"
                      message = "Edit"
                      require "uri"
                      urls.each do |url|
                        uri = URI.parse(url)
                        require "nice_http"
                        http = NiceHttp.new(host: "https://#{uri.host}")
                        file_path_name = "#{config.path}/tmp/#{user.name}_#{@ai_open_ai_image[user.name].object_id}.png"
                        res = http.get(uri.path + "?#{uri.query}", save_data: file_path_name)
                        send_file(Thread.current[:dest], messagersp, file_path_name, message, "image/png", "png")
                        http.close unless http.nil?
                      end
                    end
                  else
                    respond res
                  end
                end
              rescue => exception
                respond "*OpenAI*: Sorry, I'm having some problems. OpenAI probably is not available. Please try again later."
                @logger.warn exception
              end
              unreact :art
            end
          end
        end
      end
    end
  end
end