Move to batch/hotcli.php

batch/cli/cli_upload.php

@@ -0,0 +1,145 @@
+<?php
+// upload_cli.php -- HotCRP script for interacting with site APIs
+// Copyright (c) 2006-2025 Eddie Kohler; see LICENSE.
+
+class Upload_CLIBatch implements CLIBatchCommand {
+    /** @var resource */
+    public $stream;
+    /** @var ?string */
+    public $mimetype;
+    /** @var ?string */
+    public $filename;
+    /** @var ?string */
+    public $input_filename;
+    /** @var ?int */
+    public $size;
+    /** @var int */
+    private $offset;
+    /** @var bool */
+    private $retry;
+
+    function __construct($stream) {
+        $this->stream = $stream;
+        $stat = fstat($stream);
+        if ($stat && $stat["size"] > 0) {
+            $this->size = $stat["size"];
+        }
+    }
+
+    /** @return bool */
+    function chunk_retry(HotCLI_Batch $clib) {
+        if ($clib->status_code > 399
+            && $clib->content_json
+            && is_int($clib->content_json->maxblob ?? null)
+            && $clib->content_json->maxblob < $clib->chunk
+            && $this->offset === 0
+            && !$this->retry) {
+            $clib->set_chunk(max(1, $clib->content_json->maxblob));
+            $this->retry = true;
+            return true;
+        }
+    }
+
+    /** @return ?string */
+    function execute(HotCLI_Batch $clib) {
+        $this->offset = 0;
+        $this->retry = false;
+        curl_setopt($clib->curlh, CURLOPT_CUSTOMREQUEST, "POST");
+        $token = null;
+        $startargs = "start=1";
+        if ($this->filename) {
+            $startargs .= "&filename=" . urlencode($this->filename);
+        }
+        if ($this->mimetype) {
+            $startargs .= "&mimetype=" . urlencode($this->mimetype);
+        }
+        if (isset($this->size)) {
+            $startargs .= "&size={$this->size}";
+        }
+        $buf = "";
+        while (true) {
+            if (strlen($buf) < $clib->chunk) {
+                $x = fread($this->stream, $clib->chunk - strlen($buf));
+                if ($x === false) {
+                    $mi = $clib->error_at(null, "<0>Error reading file");
+                    $mi->landmark = $this->input_filename;
+                    return null;
+                }
+                $buf .= $x;
+            }
+            $breakpos = min($clib->chunk, strlen($buf));
+            $s = substr($buf, 0, $breakpos);
+            if ($s === "" && $this->offset === 0) {
+                $mi = $clib->error_at(null, "<0>Empty file");
+                $mi->landmark = $this->input_filename;
+                return null;
+            }
+            $eof = $buf === "" && feof($this->stream);
+            curl_setopt($clib->curlh, CURLOPT_URL, $clib->site
+                . "/upload?"
+                . ($this->offset === 0 ? $startargs : "token={$token}")
+                . ($s === "" ? "" : "&offset={$this->offset}")
+                . ($eof ? "&finish=1" : ""));
+            curl_setopt($clib->curlh, CURLOPT_POSTFIELDS, $s === "" ? [] : ["blob" => $s]);
+            if (!$clib->exec_api([$this, "chunk_retry"])) {
+                return null;
+            }
+            if ($this->retry) {
+                continue;
+            }
+            if ($token === null) {
+                if (!is_string($clib->content_json->token ?? null)
+                    || !preg_match('/\Ahcup[A-Za-z0-9]++\z/', $clib->content_json->token)) {
+                    if (!$clib->has_error()) {
+                        $clib->error_at(null, "<0>Upload token missing from server response");
+                    }
+                    return null;
+                }
+                $token = $clib->content_json->token;
+            }
+            if ($clib->verbose && $eof) {
+                fwrite(STDERR, $clib->content_string);
+            }
+            if ($eof) {
+                return $token;
+            }
+            $this->offset += $breakpos;
+            $buf = (string) substr($buf, $breakpos);
+        }
+    }
+
+    /** @return int */
+    function run(HotCLI_Batch $clib) {
+        $token = $this->execute($clib);
+        if ($token === null) {
+            return 1;
+        }
+        $clib->set_output("{$token}\n");
+        return 0;
+    }
+
+    /** @return Upload_CLIBatch */
+    static function make_arg(HotCLI_Batch $clib, $arg) {
+        if (empty($arg["_"])) {
+            $f = STDIN;
+        } else {
+            $f = @fopen($arg["_"][0], "rb");
+            if (!$f) {
+                throw CommandLineException::make_file_error($arg["_"][0]);
+            }
+            if (!isset($arg["filename"])) {
+                $arg["filename"] = preg_replace('/\A.*\/(?=[^\/]+\z)/', "", $arg["_"][0]);
+            }
+        }
+        $ucb = new Upload_CLIBatch($f);
+        $ucb->mimetype = $arg["mimetype"] ?? null;
+        $ucb->input_filename = $arg["_"][0] ?? "<stdin>";
+        if (isset($arg["filename"]) && !isset($arg["no-filename"])) {
+            $ucb->filename = $arg["filename"];
+        }
+        if (isset($arg["no-filename"])) {
+            unset($arg["filename"]);
+        }
+        return $ucb;
+    }
+}