shimmie2/ext/upload/theme.php

<?php

declare(strict_types=1);

namespace Shimmie2;

use MicroHTML\HTMLElement;

use function MicroHTML\TABLE;
use function MicroHTML\TR;
use function MicroHTML\TD;
use function MicroHTML\SMALL;
use function MicroHTML\rawHTML;
use function MicroHTML\INPUT;
use function MicroHTML\emptyHTML;
use function MicroHTML\NOSCRIPT;
use function MicroHTML\DIV;
use function MicroHTML\BR;
use function MicroHTML\A;

use function MicroHTML\P;

class UploadTheme extends Themelet
{
    protected bool $has_errors = false;

    public function display_block(Page $page): void
    {
        $b = new Block("Upload", (string)$this->build_upload_block(), "left", 20);
        $b->is_content = false;
        $page->add_block($b);
    }

    public function display_full(Page $page): void
    {
        $page->add_block(new Block("Upload", "Disk nearly full, uploads disabled", "left", 20));
    }

    public function display_page(Page $page): void
    {
        global $config, $page;

        $tl_enabled = ($config->get_string(UploadConfig::TRANSLOAD_ENGINE, "none") != "none");
        $max_size = $config->get_int(UploadConfig::SIZE);
        $max_kb = to_shorthand_int($max_size);
        $max_total_size = parse_shorthand_int(ini_get('post_max_size')) - 102400; //leave room for http request data
        $max_total_kb = to_shorthand_int($max_total_size);
        $upload_list = $this->h_upload_list_1();

        $form = SHM_FORM("upload", "POST", true, "file_upload");
        $form->appendChild(
            TABLE(
                ["id" => "large_upload_form", "class" => "vert"],
                TR(
                    TD(["width" => "20"], rawHTML("Common&nbsp;Tags")),
                    TD(["colspan" => "5"], INPUT(["name" => "tags", "type" => "text", "placeholder" => "tagme", "class" => "autocomplete_tags", "autocomplete" => "off"]))
                ),
                TR(
                    TD(["width" => "20"], rawHTML("Common&nbsp;Source")),
                    TD(["colspan" => "5"], INPUT(["name" => "source", "type" => "text"]))
                ),
                $upload_list,
                TR(
                    TD(["colspan" => $tl_enabled ? 2 : 4,"id" => "upload_size_tracker"], ""),
                    TD(["colspan" => 2], ""),
                ),
                TR(
                    TD(["colspan" => "6"], INPUT(["id" => "uploadbutton", "type" => "submit", "value" => "Post"]))
                ),
            )
        );
        $html = emptyHTML(
            $form,
            SMALL("(Max file size is $max_kb)"),
            SMALL(BR(), "(Max total size is $max_total_kb)"),
            rawHTML("<script>
                function fileSize(size){
                    var i = Math.floor(Math.log(size) / Math.log(1024));
                    return (size / Math.pow(1024, i)).toFixed(2) * 1 + ' ' + ['B', 'kB', 'MB', 'GB', 'TB'][i];
                }
                function updateTracker(){
                    var size = 0;
                    var upbtn = $(\"#uploadbutton\")[0];
                    var tracker = $(\"#upload_size_tracker\")[0];
                    var lockbtn = false;

                    $(\"input[name*='data'][id!='data[]']\").each(function(_,n){
                        var cancelbtn = $(\"div[name='cancel\"+n.name+\"']\")[0];
                        var toobig = false;
                        if (n.files.length){
                            cancelbtn.style.visibility = 'visible';
                            for (var i = 0; i<n.files.length; i++){
                                size += n.files[i].size;
                                if (n.files[i].size > $max_size){
                                    toobig = true;
                                }
                            }
                            if (toobig){
                                lockbtn = true;
                                n.style = 'color:red';
                            }else{
                                n.style = 'color:initial';
                            }
                        }else{
                            n.style = 'color:initial';
                            cancelbtn.style.visibility = 'hidden';
                        }
                    });

                    if (size){
                        tracker.innerText = 'Total: ' + fileSize(size);
                        if (size > $max_total_size){
                            lockbtn = true;
                            tracker.style = 'color:red';
                        }else{
                            tracker.style = 'color:initial';
                        }
                    }else{
                        tracker.innerText = '';
                    }
                    upbtn.disabled = lockbtn;
                }
                window.onload = function(){
                    $(\"input[name*='data'][id!='data[]']\").change(function(){
                        updateTracker();
                    });
                    updateTracker();
                }
            </script>")
        );

        $page->set_title("Upload");
        $page->set_heading("Upload");
        $page->add_block(new NavBlock());
        $page->add_block(new Block("Upload", $html, "main", 20));
        if ($tl_enabled) {
            $page->add_block(new Block("Bookmarklets", (string)$this->h_bookmarklets(), "left", 20));
        }
    }

    protected function h_upload_list_1(): HTMLElement
    {
        global $config;
        $upload_list = emptyHTML();
        $upload_count = $config->get_int(UploadConfig::COUNT);
        $tl_enabled = ($config->get_string(UploadConfig::TRANSLOAD_ENGINE, "none") != "none");
        $accept = $this->get_accept();

        $upload_list->appendChild(
            TR(
                TD(["colspan" => $tl_enabled ? 2 : 4], "Files"),
                $tl_enabled ? TD(["colspan" => "2"], "URLs") : emptyHTML(),
                TD(["colspan" => "2"], "Post-Specific Tags"),
            )
        );

        for ($i = 0; $i < $upload_count; $i++) {
            $upload_list->appendChild(
                TR(
                    TD(["colspan" => $tl_enabled ? 2 : 4], DIV(["name" => "canceldata{$i}[]","style" => "display:inline;margin-right:5px;font-size:15px;visibility:hidden;","onclick" => "$(\"input[name='data{$i}[]']\")[0].value='';updateTracker();"], "✖"), INPUT(["type" => "file", "name" => "data{$i}[]", "accept" => $accept, "multiple" => true])),
                    $tl_enabled ? TD(["colspan" => "2"], INPUT(["type" => "text", "name" => "url{$i}"])) : emptyHTML(),
                    TD(["colspan" => "2"], INPUT(["type" => "text", "name" => "tags{$i}", "class" => "autocomplete_tags", "autocomplete" => "off"])),
                )
            );
        }

        return $upload_list;
    }

    protected function h_bookmarklets(): HTMLElement
    {
        global $config;
        $link = make_http(make_link("upload"));
        $main_page = make_http(make_link());
        $title = $config->get_string(SetupConfig::TITLE);
        $max_size = $config->get_int(UploadConfig::SIZE);
        $max_kb = to_shorthand_int($max_size);
        $delimiter = $config->get_bool('nice_urls') ? '?' : '&amp;';

        $js = 'javascript:(
            function() {
                if(typeof window=="undefined" || !window.location || window.location.href=="about:blank") {
                    window.location = "'. $main_page .'";
                }
                else if(typeof document=="undefined" || !document.body) {
                    window.location = "'. $main_page .'?url="+encodeURIComponent(window.location.href);
                }
                else if(window.location.href.match("\/\/'. $_SERVER["HTTP_HOST"] .'.*")) {
                    alert("You are already at '. $title .'!");
                }
                else {
                    var tags = prompt("Please enter tags", "tagme");
                    if(tags !== "" && tags !== null) {
                        var link = "'. $link . $delimiter .'url="+location.href+"&tags="+tags;
                        var w = window.open(link, "_blank");
                    }
                }
            }
        )();';
        $html1 = P(
            A(["href" => $js], "Upload to $title"),
            rawHTML(' (Drag &amp; drop onto your bookmarks toolbar, then click when looking at a post)')
        );

        // Bookmarklet checks if shimmie supports ext. If not, won't upload to site/shows alert saying not supported.
        $supported_ext = join(" ", DataHandlerExtension::get_all_supported_exts());

        $title = "Booru to " . $config->get_string(SetupConfig::TITLE);
        // CA=0: Ask to use current or new tags | CA=1: Always use current tags | CA=2: Always use new tags
        $js = '
            javascript:
            var ste=&quot;'. $link . $delimiter .'url=&quot;;
            var supext=&quot;'.$supported_ext.'&quot;;
            var maxsize=&quot;'.$max_kb.'&quot;;
            var CA=0;
            void(document.body.appendChild(document.createElement(&quot;script&quot;)).src=&quot;'.make_http(get_base_href())."/ext/upload/bookmarklet.js".'&quot;)
        ';
        $html2 = P(
            A(["href" => $js], $title),
            rawHTML("(Click when looking at a post page. Works on sites running Shimmie / Danbooru / Gelbooru. (This also grabs the tags / rating / source!))"),
        );

        return emptyHTML($html1, $html2);
    }

    /**
     * Only allows 1 file to be uploaded - for replacing another image file.
     */
    public function display_replace_page(Page $page, int $image_id)
    {
        global $config, $page;
        $tl_enabled = ($config->get_string(UploadConfig::TRANSLOAD_ENGINE, "none") != "none");
        $accept = $this->get_accept();

        $upload_list = emptyHTML(
            TR(
                TD("File"),
                TD(INPUT(["name" => "data[]", "type" => "file", "accept" => $accept]))
            )
        );
        if ($tl_enabled) {
            $upload_list->appendChild(
                TR(
                    TD("or URL"),
                    TD(INPUT(["name" => "url", "type" => "text"]))
                )
            );
        }

        $max_size = $config->get_int(UploadConfig::SIZE);
        $max_kb = to_shorthand_int($max_size);

        $image = Image::by_id($image_id);
        $thumbnail = $this->build_thumb_html($image);

        $form = SHM_FORM("upload/replace/".$image_id, "POST", true);
        $form->appendChild(emptyHTML(
            INPUT(["type" => "hidden", "name" => "image_id", "value" => $image_id]),
            TABLE(
                ["id" => "large_upload_form", "class" => "vert"],
                $upload_list,
                TR(TD("Source"), TD(["colspan" => 3], INPUT(["name" => "source", "type" => "text"]))),
                TR(TD(["colspan" => 4], INPUT(["id" => "uploadbutton", "type" => "submit", "value" => "Post"]))),
            )
        ));

        $html = emptyHTML(
            P(
                "Replacing Post ID $image_id",
                BR(),
                "Please note: You will have to refresh the post page, or empty your browser cache."
            ),
            $thumbnail,
            BR(),
            $form,
            SMALL("(Max file size is $max_kb)"),
        );

        $page->set_title("Replace Post");
        $page->set_heading("Replace Post");
        $page->add_block(new NavBlock());
        $page->add_block(new Block("Upload Replacement Post", $html, "main", 20));
    }

    public function display_upload_status(Page $page, array $image_ids): void
    {
        global $user;

        if ($this->has_errors) {
            $page->set_title("Upload Status");
            $page->set_heading("Upload Status");
            $page->add_block(new NavBlock());
        } else {
            if (count($image_ids) < 1) {
                $page->set_title("No images uploaded");
                $page->set_heading("No images uploaded");
                $page->add_block(new NavBlock());
            } elseif (count($image_ids) == 1) {
                $page->set_mode(PageMode::REDIRECT);
                $page->set_redirect(make_link("post/view/{$image_ids[0]}"));
            } else {
                $page->set_mode(PageMode::REDIRECT);
                $page->set_redirect(search_link(["poster={$user->name}"]));
            }
        }
    }

    public function display_upload_error(Page $page, string $title, string $message): void
    {
        // this message has intentional HTML in it...
        $message = str_contains($message, "already has hash") ? $message : html_escape($message);
        $page->add_block(new Block($title, $message));
        $this->has_errors = true;
    }

    protected function build_upload_block(): HTMLElement
    {
        global $config;

        $accept = $this->get_accept();

        $max_size = $config->get_int(UploadConfig::SIZE);
        $max_kb = to_shorthand_int($max_size);
        $max_total_size = parse_shorthand_int(ini_get('post_max_size')) - 102400; //leave room for http request data
        $max_total_kb = to_shorthand_int($max_total_size);

        // <input type='hidden' name='max_file_size' value='$max_size' />
        $form = SHM_FORM("upload", "POST", true);
        $form->appendChild(
            emptyHTML(
                INPUT(["id" => "data[]", "name" => "data[]", "size" => "16", "type" => "file", "accept" => $accept, "multiple" => true]),
                INPUT(["name" => "tags", "type" => "text", "placeholder" => "tagme", "class" => "autocomplete_tags", "required" => true, "autocomplete" => "off"]),
                INPUT(["type" => "submit", "value" => "Post"]),
            )
        );

        return DIV(
            ["class" => 'mini_upload'],
            $form,
            SMALL("(Max file size is $max_kb)"),
            SMALL(BR(), "(Max total size is $max_total_kb)"),
            NOSCRIPT(BR(), A(["href" => make_link("upload")], "Larger Form"))
        );
    }

    protected function get_accept(): string
    {
        return ".".join(",.", DataHandlerExtension::get_all_supported_exts());
    }
}