cloud_api = $cloud_api; } /** * Make a POST request. * * @param string $endpoint Endpoint to contact. * @param array $data Data to include in POST request.. * * @return array|WP_Error Response data on success or error on failure. */ private function send_post_request( string $endpoint, array $data ) { if ( ! $this->cloud_api->is_cloud_key_verified() ) { return new WP_Error( 'cloud_ai_key_error', __( 'Cannot access Cloud API without a verified cloud key.', 'code-snippets' ) ); } $freemius_license = code_snippets()->licensing->get_license_key(); if ( ! $freemius_license ) { return new WP_Error( 'cloud_ai_license_error', __( 'Could not retrieve license details for API request.', 'code-snippets' ) ); } $multipart_data = [ [ 'name' => 'fs_key', 'contents' => $freemius_license, ], ]; foreach ( $data as $name => $contents ) { $multipart_data[] = [ 'name' => $name, 'contents' => $contents, ]; } $url = sprintf( '%s/%s', self::API_URL, ltrim( $endpoint, '/\\' ) ); $client = new Client(); $request = new Request( 'POST', $url, $this->cloud_api->build_request_headers() ); try { $response = $client->send( $request, [ 'multipart' => $multipart_data ] ); } catch ( GuzzleException $exception ) { return new WP_Error( 'cloud_ai_request_error', __( 'Failed to send request to Cloud API', 'code-snippets' ), $exception->getMessage() ); } $body = json_decode( $response->getBody(), true ); return $body['response'] && is_array( $body['response'] ) ? $body['response'] : new WP_Error( 'cloud_ai_response_error', __( 'Did not receive a valid response from the Cloud API.', 'code-snippets' ), $body ); } /** * Unpack response data by filtering out invalid and missing data, and rewriting field names. * * @param array $response Response data. * @param array $key_map Map of valid keys from response key to result key. * * @return array */ private function unpack_response_data( array $response, array $key_map ): array { $result = []; foreach ( $response as $key => $value ) { if ( isset( $key_map[ $key ] ) && ! empty( $value ) ) { $result[ $key_map[ $key ] ] = $value; } } return $result; } /** * Make a POST request to the Cloud AI prompt endpoint. * * @param string $prompt Prompt to use for generating code. * @param string $type Snippet type. * * @return array|WP_Error Request response. */ public function prompt( string $prompt, string $type ) { if ( empty( $prompt ) ) { return new WP_Error( 'generate_snippet_missing_prompt', esc_html__( 'Cannot generate snippet for an empty prompt.', 'code-snippets' ), [ 'status' => 400 ] ); } if ( ! in_array( $type, self::VALID_PROMPT_TYPES, true ) ) { return new WP_Error( 'generate_snippet_invalid_type', esc_html__( 'Cannot generate code for invalid snippet type.', 'code-snippets' ), [ 'status' => 400, 'type' => $type, ] ); } $response = $this->send_post_request( sprintf( '%s/%s', self::PROMPT_PATH, $type ), [ 'prompt' => $prompt ] ); return is_wp_error( $response ) ? $response : $this->unpack_response_data( $response, [ 'n' => 'name', 'c' => 'code', 'd' => 'desc', ] ); } /** * Make a POST request to the Cloud AI explain endpoint. * * @param string $code Snippet code to explain. * @param string $field Snippet field. * * @return array|WP_Error */ public function explain( string $code, string $field ) { if ( empty( $code ) ) { return new WP_Error( 'explain_snippet_missing_code', esc_html__( 'Cannot generate an explanation for empty snippet code.', 'code-snippets' ), [ 'status' => 400 ] ); } if ( ! in_array( $field, self::VALID_EXPLAIN_FIELDS, true ) ) { return new WP_Error( 'explain_snippet_invalid_field', esc_html__( 'Cannot generate explanation for invalid snippet field.', 'code-snippets' ), [ 'status' => 400, 'field' => $field, ] ); } $response = $this->send_post_request( sprintf( '%s/%s', self::EXPLAIN_PATH, $field ), [ 'prompt' => $code ] ); return is_wp_error( $response ) ? $response : $this->unpack_response_data( $response, [ 'n' => 'name', 'c' => 'lines', 't' => 'tags', 'd' => 'desc', ] ); } }