rounds/transcodeJXL.go

   1 // tofuproxy -- flexible HTTP/HTTPS proxy, TLS terminator, X.509 TOFU
   2 //              manager, WARC/geminispace browser
   3 // Copyright (C) 2021-2024 Sergey Matveev <stargrave@stargrave.org>
   4 //
   5 // This program is free software: you can redistribute it and/or modify
   6 // it under the terms of the GNU General Public License as published by
   7 // the Free Software Foundation, version 3 of the License.
   8 //
   9 // This program is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU General Public License
  15 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 package rounds
  18
  19 import (
  20         "fmt"
  21         "io"
  22         "log"
  23         "net/http"
  24         "os"
  25         "os/exec"
  26
  27         "go.stargrave.org/tofuproxy/fifos"
  28 )
  29
  30 func transcodeCmd2Png(
  31         contentType, ext, cmdName string,
  32         resp *http.Response,
  33         w http.ResponseWriter,
  34         req *http.Request,
  35 ) (bool, error) {
  36         if resp.Header.Get("Content-Type") != contentType {
  37                 return true, nil
  38         }
  39         tmpFd, err := os.CreateTemp("", "tofuproxy.*."+ext)
  40         if err != nil {
  41                 log.Fatalln(err)
  42         }
  43         defer os.Remove(tmpFd.Name())
  44         defer tmpFd.Close()
  45         defer resp.Body.Close()
  46         if _, err = io.Copy(tmpFd, resp.Body); err != nil {
  47                 log.Printf("Error during %s: %+v\n", req.URL, err)
  48                 return false, err
  49         }
  50         tmpFd.Close()
  51         dstFn := tmpFd.Name() + ".png"
  52         cmd := exec.Command(cmdName, tmpFd.Name(), dstFn)
  53         err = cmd.Run()
  54         defer os.Remove(dstFn)
  55         if err != nil {
  56                 return false, err
  57         }
  58         data, err := os.ReadFile(dstFn)
  59         if err != nil {
  60                 return false, err
  61         }
  62         w.Header().Add("Content-Type", "image/png")
  63         w.WriteHeader(http.StatusOK)
  64         w.Write(data)
  65         fifos.LogVarious <- fmt.Sprintf(
  66                 "%s %s\t%s transcoded to PNG", req.Method, req.URL, contentType,
  67         )
  68         return false, nil
  69 }
  70
  71 func RoundTranscodeJXL(
  72         host string,
  73         resp *http.Response,
  74         w http.ResponseWriter,
  75         req *http.Request,
  76 ) (bool, error) {
  77         return transcodeCmd2Png("image/jxl", "jxl", "djxl", resp, w, req)
  78 }