shimmie2/ext/upload/theme.php

<?php declare(strict_types=1);

class UploadTheme extends Themelet
{
    public function display_block(Page $page)
    {
        $page->add_block(new Block("Upload", $this->build_upload_block(), "left", 20));
    }

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

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

        $tl_enabled = ($config->get_string("transload_engine", "none") != "none");
        $max_size = $config->get_int('upload_size');
        $max_kb = to_shorthand_int($max_size);
        $upload_list = $this->h_upload_list_1();
        $html = "
			".make_form(make_link("upload"), "POST", $multipart=true, 'file_upload')."
				<table id='large_upload_form' class='vert'>
					<tr><td width='20'>Common&nbsp;Tags<td colspan='5'><input name='tags' type='text' placeholder='tagme' class='autocomplete_tags' autocomplete='off'></td></tr>
					<tr><td>Common&nbsp;Source</td><td colspan='5'><input name='source' type='text'></td></tr>
					$upload_list
					<tr><td colspan='6'><input id='uploadbutton' type='submit' value='Post'></td></tr>
				</table>
			</form>
			<small>(Max file size is $max_kb)</small>
		";

        $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", $this->h_bookmarklets(), "left", 20));
        }
    }

    protected function h_upload_list_1(): string
    {
        global $config;
        $upload_list = "";
        $upload_count = $config->get_int('upload_count');
        $tl_enabled = ($config->get_string("transload_engine", "none") != "none");
        $accept = $this->get_accept();

        if ($tl_enabled) {
            $upload_list .= "
				<tr>
					<td colspan='2'>Files</td>
					<td colspan='2'>URLs</td>
					<td colspan='2'>Image-Specific Tags</td>
				</tr>
			";

            for ($i=0; $i<$upload_count; $i++) {
                $upload_list .= "
					<tr>
						<td colspan='2'><input type='file' name='data$i' accept='$accept'></td>
						<td colspan='2'><input type='text' name='url$i'></td>
						<td colspan='2'><input type='text' name='tags$i' class='autocomplete_tags' autocomplete='off'></td>
					</tr>
				";
            }
        } else {
            $upload_list .= "
				<tr>
					<td colspan='4'>Files</td>
					<td colspan='2'>Image-Specific Tags</td>
				</tr>
			";

            for ($i=0; $i<$upload_count; $i++) {
                $upload_list .= "
					<tr>
						<td colspan='4'><input type='file' name='data$i' accept='$accept'></td>
						<td colspan='2'><input type='text' name='tags$i' class='autocomplete_tags' autocomplete='off'></td>
					</tr>
				";
            }
        }

        return $upload_list;
    }

    protected function h_bookmarklets(): string
    {
        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('upload_size');
        $max_kb = to_shorthand_int($max_size);
        $delimiter = $config->get_bool('nice_urls') ? '?' : '&amp;';
        $html = '';

        $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");
					}
				}
			}
		)();';
        $html .= '<a href=\''.$js.'\'>Upload to '.$title.'</a>';
        $html .= ' (Drag &amp; drop onto your bookmarks toolbar, then click when looking at an image)';

        // Bookmarklet checks if shimmie supports ext. If not, won't upload to site/shows alert saying not supported.
        $supported_ext = "jpg jpeg gif png webp";
        if (class_exists("FlashFileHandler")) {
            $supported_ext .= " swf";
        }
        if (class_exists("ICOFileHandler")) {
            $supported_ext .= " ico ani cur";
        }
        if (class_exists("MP3FileHandler")) {
            $supported_ext .= " mp3";
        }
        if (class_exists("SVGFileHandler")) {
            $supported_ext .= " svg";
        }
        if (class_exists("VideoFileHandler")) {
            $supported_ext .= " flv mp4 ogv webm m4v";
        }
        $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
        $html .= '<p><a href="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;)
		">'. $title . '</a> (Click when looking at an image page. Works on sites running Shimmie / Danbooru / Gelbooru. (This also grabs the tags / rating / source!))';

        return $html;
    }

    /**
     * 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("transload_engine", "none") != "none");
        $accept = $this->get_accept();

        $upload_list = "
			<tr>
				<td>File</td>
				<td><input name='data' type='file' accept='$accept'></td>
			</tr>
		";
        if ($tl_enabled) {
            $upload_list .="
			<tr>
				<td>or URL</td>
				<td><input name='url' type='text'></td>
			</tr>
			";
        }

        $max_size = $config->get_int('upload_size');
        $max_kb = to_shorthand_int($max_size);

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

        $html = "
				<p>Replacing Image ID ".$image_id."<br>Please note: You will have to refresh the image page, or empty your browser cache.</p>"
                .$thumbnail."<br>"
                .make_form(make_link("upload/replace/".$image_id), "POST", $multipart=true)."
				<input type='hidden' name='image_id' value='$image_id'>
				<table id='large_upload_form' class='vert'>
					$upload_list
					<tr><td>Source</td><td colspan='3'><input name='source' type='text'></td></tr>
					<tr><td colspan='4'><input id='uploadbutton' type='submit' value='Post'></td></tr>
				</table>
			</form>
			<small>(Max file size is $max_kb)</small>
		";

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

    public function display_upload_status(Page $page, bool $ok)
    {
        if ($ok) {
            $page->set_mode(PageMode::REDIRECT);
            $page->set_redirect(make_link());
        } else {
            $page->set_title("Upload Status");
            $page->set_heading("Upload Status");
            $page->add_block(new NavBlock());
        }
    }

    public function display_upload_error(Page $page, string $title, string $message)
    {
        $page->add_block(new Block($title, html_escape($message)));
    }

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

        $upload_list = "";
        $upload_count = $config->get_int('upload_count');
        $accept = $this->get_accept();

        for ($i=0; $i<$upload_count; $i++) {
            if ($i == 0) {
                $style = "";
            } // "style='display:visible'";
            else {
                $style = "style='display:none'";
            }
            $upload_list .= "<input id='data$i' name='data$i' $style onchange=\"$('#data".($i+1)."').show()\" size='16' type='file' accept='$accept'>\n";
        }
        $max_size = $config->get_int('upload_size');
        $max_kb = to_shorthand_int($max_size);
        // <input type='hidden' name='max_file_size' value='$max_size' />
        return "
			<div class='mini_upload'>
			".make_form(make_link("upload"), "POST", $multipart=true)."
				$upload_list
				<input name='tags' type='text' placeholder='tagme' class='autocomplete_tags' required='required' autocomplete='off'>
				<input type='submit' value='Post'>
			</form>
			<small>(Max file size is $max_kb)</small>
			<noscript><br><a href='".make_link("upload")."'>Larger Form</a></noscript>
			</div>
		";
    }

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