]> Sergey Matveev's repositories - tofuproxy.git/blob - rounds/transcodeJXL.go
Set BasicAuth before request
[tofuproxy.git] / rounds / transcodeJXL.go
1 /*
2 tofuproxy -- HTTP proxy with TLS certificates management
3 Copyright (C) 2021 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
18 package rounds
19
20 import (
21         "fmt"
22         "io"
23         "io/ioutil"
24         "log"
25         "net/http"
26         "os"
27         "os/exec"
28
29         "go.stargrave.org/tofuproxy/fifos"
30 )
31
32 func transcodeCmd2Png(
33         contentType, ext, cmdName string,
34         resp *http.Response,
35         w http.ResponseWriter,
36         req *http.Request,
37 ) (bool, error) {
38         if resp.Header.Get("Content-Type") != contentType {
39                 return true, nil
40         }
41         tmpFd, err := ioutil.TempFile("", "tofuproxy.*."+ext)
42         if err != nil {
43                 log.Fatalln(err)
44         }
45         defer os.Remove(tmpFd.Name())
46         defer tmpFd.Close()
47         defer resp.Body.Close()
48         if _, err = io.Copy(tmpFd, resp.Body); err != nil {
49                 log.Printf("Error during %s: %+v\n", req.URL, err)
50                 return false, err
51         }
52         tmpFd.Close()
53         dstFn := tmpFd.Name() + ".png"
54         cmd := exec.Command(cmdName, tmpFd.Name(), dstFn)
55         err = cmd.Run()
56         defer os.Remove(dstFn)
57         if err != nil {
58                 return false, err
59         }
60         data, err := ioutil.ReadFile(dstFn)
61         if err != nil {
62                 return false, err
63         }
64         w.Header().Add("Content-Type", "image/png")
65         w.WriteHeader(http.StatusOK)
66         w.Write(data)
67         fifos.SinkOther <- fmt.Sprintf(
68                 "%s %s\t%d\t%s transcoded to PNG",
69                 req.Method,
70                 req.URL.String(),
71                 http.StatusOK,
72                 contentType,
73         )
74         return false, nil
75 }
76
77 func RoundTranscodeJXL(
78         host string,
79         resp *http.Response,
80         w http.ResponseWriter,
81         req *http.Request,
82 ) (bool, error) {
83         return transcodeCmd2Png("image/jxl", "jxl", "djxl", resp, w, req)
84 }