intraxia/wp-gistpen

<?php
namespace Intraxia\Gistpen\Http;

use Intraxia\Gistpen\Model\Blob;
use Intraxia\Gistpen\Model\Repo;
use Intraxia\Gistpen\Model\Language;
use Intraxia\Jaxion\Contract\Axolotl\EntityManager;
use Intraxia\Jaxion\Contract\Core\HasFilters;
use WP_Error;
use WP_REST_Request;
use WP_REST_Response;

/**
 * Class BlobController
 *
 * @package    Intraxia\Gistpen
 * @subpackage Http
 */
class BlobController implements HasFilters {
    /**
     * Database interface service.
     *
     * @var EntityManager
     */
    protected $em;

    /**
     * RepoController constructor.
     *
     * @param EntityManager $em
     */
    public function __construct( EntityManager $em ) {
        $this->em = $em;
    }

    /**
     * Creates a new blob with the provided data.
     *
     * @param WP_REST_Request $request
     *
     * @return WP_Error|WP_REST_Response
     */
    public function create( WP_REST_Request $request ) {
        $repo_id = $request->get_param( 'repo_id' );

        $repo = $this->em->find( Repo::class, $repo_id );

        if ( is_wp_error( $repo ) ) {
            $repo->add_data( array( 'status' => 404 ) );

            return $repo;
        }

        $blob = $this->em->create( Blob::class, [
            'filename' => $request->get_param( 'filename' ),
            'code'     => $request->get_param( 'code' ),
            'language' => [
                // @TODO(mAAdhaTTah) this is a bad API for the EntityManager.
                'slug' => $request->get_param( 'language' ),
            ],
            'repo_id'  => $repo->ID,
            'status'   => $repo->status,
        ], [ 'unguarded' => true ] );

        if ( is_wp_error( $blob ) ) {
            $blob->add_data( array( 'status' => 500 ) );

            return $blob;
        }

        $response = new WP_REST_Response( $blob->serialize(), 201 );
        $response->header( 'Location', $blob->rest_url );

        return $response;
    }

    /**
     * Creates a new blob with the provided data.
     *
     * @param WP_REST_Request $request
     *
     * @return WP_Error|WP_REST_Response
     */
    public function view( WP_REST_Request $request ) {
        $repo_id = $request->get_param( 'repo_id' );

        $repo = $this->em->find( Repo::class, $repo_id );

        if ( is_wp_error( $repo ) ) {
            $repo->add_data( array( 'status' => 404 ) );

            return $repo;
        }

        $blob_id = $request->get_param( 'blob_id' );

        $blob = $this->em->find( Blob::class, $blob_id, [
            'with' => [
                'language' => [],
            ],
        ] );

        if ( is_wp_error( $blob ) ) {
            $blob->add_data( array( 'status' => 404 ) );

            return $blob;
        }

        return new WP_REST_Response( $blob->serialize(), 200 );
    }

    /**
     * Creates a new blob with the provided data.
     *
     * @param WP_REST_Request $request
     *
     * @return WP_Error|WP_REST_Response
     */
    public function update( WP_REST_Request $request ) {
        $repo_id = $request->get_param( 'repo_id' );

        $repo = $this->em->find( Repo::class, $repo_id );

        if ( is_wp_error( $repo ) ) {
            $repo->add_data( array( 'status' => 404 ) );

            return $repo;
        }

        $blob_id = $request->get_param( 'blob_id' );

        $blob = $this->em->find( Blob::class, $blob_id, [
            'with' => [
                'language' => [],
            ],
        ] );

        if ( is_wp_error( $blob ) ) {
            $blob->add_data( array( 'status' => 404 ) );

            return $blob;
        }

        // @TODO(mAAdhaTTah) would benefit from cleanup
        $language_param = $request->get_param( 'language' );
        if ( $blob->language->slug === $language_param ) {
            $language = $blob->language;
        } else {
            $language = $this->em->find( Language::class, [ 'slug' => $request->get_param( 'language' ) ] );

            if ( is_wp_error( $language ) ) {
                if ( $language->get_error_code() !== 'not_found' ) {
                    $language->add_data( array( 'status' => 500 ) );

                    return $language;
                }

                $language = new Language( [ 'slug' => $language_param ] );
            }
        }

        $blob->merge( [
            'filename' => $request->get_param( 'filename' ),
            'code'     => $request->get_param( 'code' ),
            'language' => $language,
        ] );
        $blob = $this->em->persist( $blob );

        if ( is_wp_error( $blob ) ) {
            $blob->add_data( array( 'status' => 500 ) );

            return $blob;
        }

        $blob = $this->em->find( Blob::class, $blob_id, [
            'with' => [
                'language' => [],
            ],
        ] );

        if ( is_wp_error( $blob ) ) {
            $blob->add_data( array( 'status' => 500 ) );

            return $blob;
        }

        return new WP_REST_Response( $blob->serialize(), 200 );
    }

    /**
     * Fetches the requested Blob and sets up the appropriate response.
     *
     * @param WP_REST_Request $request
     *
     * @return WP_Error|WP_REST_Response
     */
    public function raw( WP_REST_Request $request ) {
        $blob = $this->em->find( Blob::class, $request->get_param( 'blob_id' ), array(
            'repo_id' => $request->get_param( 'repo_id' ),
        ) );

        if ( is_wp_error( $repo ) ) {
            $repo->add_data( array( 'status' => 404 ) );

            return $repo;
        }

        $response = new WP_REST_Response( $blob->code, 200 );
        $response->header( 'Content-Type', 'text/plain; charset=utf-8' );
        $response->header( 'Content-Security-Policy', "default-src 'none'; style-src 'unsafe-inline'" );

        return $response;
    }

    /**
     * Overwrites the default API server response when serving raw,
     * ensuring the response isn't JSON encoded and echoed directly.
     *
     * @param boolean          $served
     * @param WP_REST_Response $response
     * @param WP_REST_Request  $request
     *
     * @return bool
     */
    public function serve_raw( $served, WP_REST_Response $response, WP_REST_Request $request ) {
        if ( $served ||
            $request->get_method() !== 'GET' ||
            ! preg_match( '/\/intraxia\/v1\/gistpen\/repos\/\d+\/blobs\/\d+\/raw/', $request->get_route() )
        ) {
            return $served;
        }

        echo $response->get_data(); // @codingStandardsIgnoreLine

        return true;
    }

    /**
     * {@inheritDoc}
     *
     * @return array[]
     */
    public function filter_hooks() {
        return array(
            array(
                'hook'   => 'rest_pre_serve_request',
                'method' => 'serve_raw',
                'args'   => 3,
            ),
        );
    }
}