<?php
/*
* OpenAI Rest Controller
*/
class BDTHEMES_OPENAI_REST_CONTROLLER extends WP_REST_CONTROLLER {
    public function __construct() {
        $this->namespace = 'bdthemes/v1';
        $this->rest_base = 'openai';
        add_action('rest_api_init', [$this, 'register_rest_routes']);
    }
    public function register_rest_routes() {
        register_rest_route($this->namespace, $this->rest_base . '/images/generations', [
            'methods' => WP_REST_Server::CREATABLE,
            'callback' => [$this, 'image_generation_request'],
            'permission_callback' => [$this, 'image_generation_permission_check'],
        ]);
    }
    public function image_generation_request(WP_REST_Request $request) {
        $paramaters = $request->get_params();
        $api_key = $this->get_openai_api_key();
        if (!$api_key) {
            return new WP_Error('no_api_key', 'No API Key Provided', ['status' => 401]);
        }
        $url = 'https://api.openai.com/v1/images/generations';
        $headers = [
            'Authorization' => 'Bearer ' . $api_key,
            'Content-Type' => 'application/json',
        ];

        $body = [
            'prompt' => isset($paramaters['prompt']) ? $paramaters['prompt'] : 'A beautiful sunset over the city',
            'n' => isset($paramaters['number']) ? (int) $paramaters['number'] : 4,
            'size' => isset($paramaters['size']) ? (string) $paramaters['size'] : '512x512',
            'response_format' => 'b64_json',
            'user' => strval(get_current_user_id()),
        ];

		if( isset( $paramaters['method'] ) && $paramaters['method'] == 'generator' ) {
			$body = [
				'prompt' => isset( $paramaters['prompt'] ) ? $paramaters['prompt'] : 'A beautiful sunset over the city',
				'n'      => isset( $paramaters['number'] ) ? (int) $paramaters['number'] : 4,
				'size'   => isset( $paramaters['size'] ) ? (string) $paramaters['size'] : '512x512',
				'user'   => strval( get_current_user_id() ),
			];
		}


        $args = [
            'headers' => $headers,
            'body' => json_encode($body),
            'method' => 'POST',
            'timeout' => 60,
            'data_format' => 'body',
        ];

        $api_request = wp_remote_post($url, $args);
        $response = json_decode($api_request['body'], true);
        return  $response;
    }
    public function image_generation_permission_check() {
		return true; //todo: change this to a proper permission check
        return current_user_can('edit_posts');
    }
    private function get_openai_api_key() {
        $api_key = get_option('bdthemes_openai_api_key');
        if (!empty($api_key)) {
            return $api_key;
        }
        return false;
    }
}

new BDTHEMES_OPENAI_REST_CONTROLLER();