<?php
/*
 * Name: Amazon S3 Mirror
 * Author: Shish <webmaster@shishnet.org>
 * License: GPLv2
 * Description: Copy uploaded files to S3
 * Documentation:
 */

require_once "ext/amazon_s3/lib/S3.php";

class UploadS3 extends Extension
{
    public function onInitExt(InitExtEvent $event)
    {
        global $config;
        $config->set_default_string("amazon_s3_access", "");
        $config->set_default_string("amazon_s3_secret", "");
        $config->set_default_string("amazon_s3_bucket", "");
    }

    public function onSetupBuilding(SetupBuildingEvent $event)
    {
        $sb = new SetupBlock("Amazon S3");
        $sb->add_text_option("amazon_s3_access", "Access key: ");
        $sb->add_text_option("amazon_s3_secret", "<br>Secret key: ");
        $sb->add_text_option("amazon_s3_bucket", "<br>Bucket: ");
        $event->panel->add_block($sb);
    }

    public function onImageAddition(ImageAdditionEvent $event)
    {
        global $config;
        $access = $config->get_string("amazon_s3_access");
        $secret = $config->get_string("amazon_s3_secret");
        $bucket = $config->get_string("amazon_s3_bucket");
        if (!empty($bucket)) {
            log_debug("amazon_s3", "Mirroring Image #".$event->image->id." to S3 #$bucket");
            $s3 = new S3($access, $secret);
            $s3->putBucket($bucket, S3::ACL_PUBLIC_READ);
            $s3->putObjectFile(
                warehouse_path("thumbs", $event->image->hash),
                $bucket,
                'thumbs/'.$event->image->hash,
                S3::ACL_PUBLIC_READ,
                [],
                [
                    "Content-Type" => "image/jpeg",
                    "Content-Disposition" => "inline; filename=image-" . $event->image->id . ".jpg",
                ]
            );
            $s3->putObjectFile(
                warehouse_path("images", $event->image->hash),
                $bucket,
                'images/'.$event->image->hash,
                S3::ACL_PUBLIC_READ,
                [],
                [
                    "Content-Type" => $event->image->get_mime_type(),
                    "Content-Disposition" => "inline; filename=image-" . $event->image->id . "." . $event->image->ext,
                ]
            );
        }
    }

    public function onImageDeletion(ImageDeletionEvent $event)
    {
        global $config;
        $access = $config->get_string("amazon_s3_access");
        $secret = $config->get_string("amazon_s3_secret");
        $bucket = $config->get_string("amazon_s3_bucket");
        if (!empty($bucket)) {
            log_debug("amazon_s3", "Deleting Image #".$event->image->id." from S3");
            $s3 = new S3($access, $secret);
            $s3->deleteObject($bucket, "images/" . $event->image->hash);
            $s3->deleteObject($bucket, "thumbs/" . $event->image->hash);
        }
    }
}