pterotype/inc/objects.php

<?php
namespace objects;

// TODO for 'post' objects, store a post id instead of the full post text,
// then hydrate the text on read

function create_local_object( $object ) {
    global $wpdb;
    if ( !array_key_exists( 'type', $object) ) {
        return new \WP_Error(
            'invalid_object',
            __( 'Object must have a "type" field', 'pterotype' ),
            array( 'status' => 400 )
        );
    }
    $res = $wpdb->insert( 'pterotype_objects', array(
        'object' => wp_json_encode( $object )
    ) );
    if ( !$res ) {
        return new \WP_Error(
            'db_error', __( 'Failed to insert object row', 'pterotype' )
        );
    }
    $object_id = $wpdb->insert_id;
    $type = $object['type'];
    $object_url = get_rest_url( null, sprintf( '/pterotype/v1/object/%d', $object_id ) );
    $object_likes = get_rest_url( null, sprintf( '/pterotype/v1/object/%d/likes', $object_id ) );
    $object_shares = get_rest_url(
        null, sprintf( '/pterotype/v1/object/%d/shares', $object_id )
    );
    $object['id'] = $object_url;
    $object['likes'] = $object_likes;
    $object['shares'] = $object_shares;
    $res = $wpdb->replace(
        'pterotype_objects',
        array (
            'id' => $object_id,
            'activitypub_id' => $object_url,
            'type' => $type,
            'object' => wp_json_encode( $object )
        ),
        array( '%d', '%s', '%s', '%s' )
    );
    if ( !$res ) {
        return new \WP_Error(
            'db_error', __( 'Failed to hydrate object id', 'pterotype' )
        );
    }
    return $object;
}

function upsert_object( $object ) {
    global $wpdb;
    if ( !array_key_exists( 'id', $object ) ) {
        return new \WP_Error(
            'invalid_object',
            __( 'Objects must have an "id" field', 'pterotype' ),
            array( 'status' => 400 )
        );
    }
    if ( !array_key_exists( 'type', $object) ) {
        return new \WP_Error(
            'invalid_object',
            __( 'Object must have a "type" field', 'pterotype' ),
            array( 'status' => 400 )
        );
    }
    $row = $wpdb->get_row( $wpdb->prepare(
        'SELECT * FROM pterotype_objects WHERE activitypub_url = %s', $object['id']
    ) );
    $res = true;
    if ( $row === null ) {
        $res = $wpdb->insert(
            'pterotype_objects',
            array(
                'activitypub_id' => $object['id'],
                'type' => $object['type'],
                'object' => wp_json_encode( $object )
            )
        );
    } else {
        $res = $wpdb->replace(
            'pterotype_objects',
            array(
                'id' => $row->id,
                'activitypub_id' => $object['id'],
                'type' => $object['type'],
                'object' => wp_json_encode( $object )
            ),
            array( '%d', '%s', '%s', '%s' )
        );
        $row = new stdClass();
        $row->id = $wpdb->insert_id;
        $activites_res = $wpdb->query( $wpdb->prepare(
            '
            UPDATE pterotype_activities
            SET activity = JSON_SET(activity, "$.object", %s)
            WHERE activity->"$.object.id" = %s;
            ',
            wp_json_encode( $object ), $object['id']
        ) );
        if ( $activities_res === false ) {
            return new \WP_Error(
                'db_error', __( 'Failed to update associated activities', 'pterotype' )
            );
        }
    }
    if ( !$res ) {
        return new \WP_Error(
            'db_error', __( 'Failed to upsert object row', 'pterotype' )
        );
    }
    $row->object = $object;
    return $row;
}

function update_object( $object ) {
    global $wpdb;
    if ( !array_key_exists( 'id', $object ) ) {
        return new \WP_Error(
            'invalid_object',
            __( 'Object must have an "id" field', 'pterotype' ),
            array( 'status' => 400 )
        );
    }
    $object_json = wp_json_encode( $object );
    $res = $wpdb->update(
        'pterotype_objects',
        array( 'object' => $object_json ),
        array( 'id' => $id ),
        '%s', '%d' );
    if ( !$res ) {
        return new \WP_Error(
            'db_error', __( 'Failed to update object row', 'pterotype' )
        );
    }
    $activites_res = $wpdb->query( $wpdb->prepare(
        '
        UPDATE pterotype_activities
        SET activity = JSON_SET(activity, "$.object", %s)
        WHERE activity->"$.object.id" = %s;
        ',
        $object_json, $object['id']
    ) );
    if ( $activities_res === false ) {
         return new \WP_Error(
            'db_error', __( 'Failed to update associated activities', 'pterotype' )
        );
    }
    return $object;
}

function get_object( $id ) {
    global $wpdb;
    $object_json = $wpdb->get_var( $wpdb->prepare(
        'SELECT object FROM pterotype_objects WHERE id = %d', $id
    ) ); 
    if ( is_null( $object_json ) ) {
        return new \WP_Error(
            'not_found', __( 'Object not found', 'pterotype' ), array( 'status' => 404 )
        );
    }
    return json_decode( $object_json, true );
}

function get_object_by_activitypub_id( $activitypub_id ) {
    global $wpdb;
    $object_json = $wpdb->get_var( $wpdb->prepare(
        'SELECT object FROM pterotype_objects WHERE activitypub_id = %s', $activitypub_id
    ) );
    if ( is_null( $object_json ) ) {
        return new \WP_Error(
            'not_found', __( 'Object not found', 'pterotype' ), array( 'status' => 404 )
        );
    }
    return json_decode( $object_json, true );
}

function get_object_id( $activitypub_id ) {
    global $wpdb;
    return $wpdb->get_var( $wpdb->prepare(
        'SELECT id FROM pterotype_objects WHERE activitypub_id = %s', $activitypub_id
    ) );
}

function delete_object( $object ) {
    global $wpdb;
    if ( !array_key_exists( 'id', $object ) ) {
        return new \WP_Error(
            'invalid_object',
            __( 'Object must have an "id" field', 'pterotype' ),
            array( 'status' => 400 )
        );
    }
    if ( !array_key_exists( 'type', $object ) ) {
        return new \WP_Error(
            'invalid_object',
            __( 'Object must have a "type" field', 'pterotype' ),
            array( 'status' => 400 )
        );
    }
    $activitypub_id = $object['id'];
    $tombstone = make_tombstone( $object );
    $res = $wpdb->replace(
        'pterotype_objects',
        array(
            'activitypub_id' => $activitypub_id,
            'type' => $tombstone['type'],
            'object' => wp_json_encode( $tombstone ),
        ),
        array( '%s', '%s', '%s' )
    );
    if ( !$res ) {
        return new \WP_Error( 'db_error', __( 'Error deleting object', 'pterotype' ) );
    }
    $res = $wpdb->query( $wpdb->prepare(
        '
        UPDATE pterotype_activities
        SET activity = JSON_SET(activity, "$.object", %s)
        WHERE activity->"$.object.id" = %s;
        ',
        wp_json_encode( $tombstone ), $object['id']
    ) );
    if ( $res === false ) {
        return new \WP_Error(
            'db_error', __( 'Failed to update associated activities', 'pterotype' )
        );
    }
    return $tombstone;
}

function make_tombstone( $object ) {
    $tombstone = array(
        'type' => 'Tombstone',
        'formerType' => $object['type'],
        'id' => $object['id'],
        'deleted' => date( \DateTime::ISO8601, time() ),
    );
    return $tombstone;
}

function is_local_object( $object ) {
    if ( is_array( $object ) ) {
        if ( array_key_exists( 'id', $object ) ) {
            $parsed = parse_url( $object['id'] );
            if ( $parsed ) {
                $site_host = parse_url( get_site_url() )['host'];
                return $parsed['host'] === $site_host;
            }
        } else {
            return false;
        }
    } else {
        $parsed = parse_url( $object );
        if ( $parsed ) {
            $site_host = parse_url( get_site_url() )['host'];
            return $parsed['host'] === $site_host;
        }
    }
    return false;
}
?>