<?php

class Meow_MWAI_OpenAI {
  private $core = null;
  private $apiKey = null;

  public function __construct( $core ) {
    $this->core = $core;
    $this->apiKey = $this->core->get_option( 'openai_apikey' );
  }

  public function listFiles() {
    return $this->run( 'GET', '/files' );
  }

  public function listFineTunes() {
    return $this->run( 'GET', '/fine-tunes' );
  }

  public function uploadFile( $filename, $data ) {
    $result = $this->run( 'POST', '/files', null, [ 'data' => $data, 'filename' => $filename ] );
    return $result;
  }

  public function deleteFile( $fileId ) {
    return $this->run( 'DELETE', '/files/' . $fileId );
  }

  public function downloadFile( $fileId ) {
    return $this->run( 'GET', '/files/' . $fileId . '/content', null, null, false );
  }

  public function fineTuneFile( $fileId, $model, $suffix ) {
    $result = $this->run( 'POST', '/fine-tunes', [
      'training_file' => $fileId,
      'model' => $model,
      'suffix' => $suffix
    ] );
    return $result;
  }

  public function create_body_for_file( $file, $boundary ) {
    $fields = array(
      'purpose' => 'fine-tune',
      'file' => $file['filename']
    );

    $body = '';
    foreach ( $fields as $name => $value ) {
        $body .= "--$boundary\r\n";
        $body .= "Content-Disposition: form-data; name=\"$name\"";
        if ( $name == 'file' ) {
            $body .= "; filename=\"{$value}\"\r\n";
            $body .= "Content-Type: application/json\r\n\r\n";
            $body .= $file['data'] . "\r\n";
        } else {
            $body .= "\r\n\r\n$value\r\n";
        }
    }
    $body .= "--$boundary--\r\n";
  

    // $body  = '';
    // $body .= '--' . $boundary;
    // $body .= "\r\n";
    // // '"; filename="' . $file['filename'] . '"'
    // $body .= 'Content-Disposition: form-data; name="photo_upload_file_name"; filename="' . $_FILES['resume']['name'] . '"' . "\r\n";
    // //$body .= 'Content-Disposition: form-data; baba="yo"; file="' . $file['filename'] . '"; purpose="fine-tune"' .  "\r\n";
    // //$body .= 'Content-Type: ' . $format . '\r\n'; //
    // //$body .= 'Content-Transfer-Encoding: binary' . "\r\n";
    // $body .= "\r\n";
    // $body .= $file['data'];
    // $body .= "\r\n";
    // $body .= '--' . $boundary . '--';
    // $body .= "\r\n\r\n";
    return $body;
  } 

  public function run( $method, $url, $query = null, $file = null, $json = true ) {
    $apiKey = $this->apiKey;
    $headers = "Content-Type: application/json\r\n" . "Authorization: Bearer " . $apiKey . "\r\n";
    $body = $query ? json_encode( $query ) : null;
    if ( !empty( $file ) ) {
      $boundary = wp_generate_password( 24, false );
      $headers  = [ 
        'Content-Type' => 'multipart/form-data; boundary=' . $boundary,
        'Authorization' => 'Bearer ' . $this->apiKey,
      ];
      $body = $this->create_body_for_file( $file, $boundary );
    }
    
    $url = 'https://api.openai.com/v1' . $url;
    $options = [
      "headers" => $headers,
      "method" => $method,
      "timeout" => 60,
      "body" => $body,
      "sslverify" => false
    ];

    try {
      $response = wp_remote_request( $url, $options );
      if ( is_wp_error( $response ) ) {
        throw new Exception( $response->get_error_message() );
      }
      $response = wp_remote_retrieve_body( $response );
      $data = $json ? json_decode( $response, true ) : $response;
      
      // Error handling
      if ( isset( $data['error'] ) ) {
        $message = $data['error']['message'];
        // If the message contains "Incorrect API key provided: THE_KEY.", replace the key by "----".
        if ( preg_match( '/API key provided(: .*)\./', $message, $matches ) ) {
          $message = str_replace( $matches[1], '', $message );
        }
        throw new Exception( $message );
      }
      
      return $data;
    }
    catch ( Exception $e ) {
      error_log( $e->getMessage() );
      throw new Exception( 'Error while calling OpenAI: ' . $e->getMessage() );
    }
  }
}