class OpenAI::Files

def default_version
  "v1".freeze
end

def delete(id:, version: default_version)
  self.class.delete(
    "/#{version}/files/#{id}",
    headers: {
      "Content-Type" => "application/json",
      "Authorization" => "Bearer #{@access_token}",
      "OpenAI-Organization" => @organization_id
    }
  )
end

def initialize(access_token: nil, organization_id: nil)
  @access_token = access_token || ENV.fetch("OPENAI_ACCESS_TOKEN")
  @organization_id = organization_id || ENV.fetch("OPENAI_ORGANIZATION_ID", nil)
end

def list(version: default_version)
  self.class.get(
    "/#{version}/files",
    headers: {
      "Content-Type" => "application/json",
      "Authorization" => "Bearer #{@access_token}",
      "OpenAI-Organization" => @organization_id
    }
  )
end

def retrieve(id:, version: default_version)
  self.class.get(
    "/#{version}/files/#{id}",
    headers: {
      "Content-Type" => "application/json",
      "Authorization" => "Bearer #{@access_token}",
      "OpenAI-Organization" => @organization_id
    }
  )
end

def upload(version: default_version, parameters: {})
  validate(file: parameters[:file])
  self.class.post(
    "/#{version}/files",
    headers: {
      "Content-Type" => "application/json",
      "Authorization" => "Bearer #{@access_token}",
      "OpenAI-Organization" => @organization_id
    },
    body: parameters.merge(file: File.open(parameters[:file]))
  )
end

def validate(file:)
  File.open(file).each_line.with_index do |line, index|
    JSON.parse(line)
  rescue JSON::ParserError => e
    raise JSON::ParserError, "#{e.message} - found on line #{index + 1} of #{file}"
  end
end