2023-05-17 09:50:17 +09:00
|
|
|
|
package main
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"fmt"
|
|
|
|
|
"io"
|
|
|
|
|
"log"
|
2023-11-09 13:39:46 +09:00
|
|
|
|
"net/http"
|
|
|
|
|
"strings"
|
|
|
|
|
"time"
|
2023-05-17 09:50:17 +09:00
|
|
|
|
)
|
|
|
|
|
|
2023-11-09 13:39:46 +09:00
|
|
|
|
var sofname = "imgproxy"
|
|
|
|
|
var version = "1.1.0"
|
|
|
|
|
var client = http.Client{
|
|
|
|
|
Timeout: time.Second * 30,
|
|
|
|
|
}
|
2023-05-17 09:50:17 +09:00
|
|
|
|
|
2023-11-09 13:39:46 +09:00
|
|
|
|
func fetchImage(curl string, nurl string) (*http.Response, error) {
|
|
|
|
|
req, err := http.NewRequest("GET", curl, nil)
|
2023-05-20 20:12:32 +09:00
|
|
|
|
if err != nil {
|
2023-11-09 13:39:46 +09:00
|
|
|
|
return nil, err
|
2023-05-20 20:12:32 +09:00
|
|
|
|
}
|
|
|
|
|
|
2023-11-09 13:39:46 +09:00
|
|
|
|
resp, err := client.Do(req)
|
2023-05-20 20:12:32 +09:00
|
|
|
|
if err != nil {
|
2023-11-09 13:39:46 +09:00
|
|
|
|
return nil, err
|
2023-05-20 20:12:32 +09:00
|
|
|
|
}
|
|
|
|
|
|
2023-11-09 13:39:46 +09:00
|
|
|
|
// 画像だけをプロクシーする様に
|
|
|
|
|
ctype := resp.Header.Get("Content-Type")
|
|
|
|
|
if ctype != "image/jpeg" && ctype != "image/png" && ctype != "image/gif" && ctype != "image/tga" && ctype != "image/targa" && ctype != "image/webp" {
|
|
|
|
|
return nil, fmt.Errorf("画像ではありません: %s", ctype)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return resp, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func serveImage(w http.ResponseWriter, img *http.Response) {
|
|
|
|
|
w.Header().Set("Content-Type", img.Header.Get("Content-Type"))
|
|
|
|
|
w.Header().Set("Content-Length", fmt.Sprint(img.ContentLength))
|
|
|
|
|
if _, err := io.Copy(w, img.Body); err != nil {
|
|
|
|
|
log.Printf("画像を出力に失敗: %v", err)
|
2023-05-20 20:12:32 +09:00
|
|
|
|
}
|
2023-11-09 13:39:46 +09:00
|
|
|
|
}
|
2023-05-20 20:12:32 +09:00
|
|
|
|
|
2023-11-09 13:39:46 +09:00
|
|
|
|
func pixivImg(w http.ResponseWriter, req *http.Request) {
|
|
|
|
|
resp, err := client.Do(req)
|
2023-05-17 09:50:17 +09:00
|
|
|
|
if err != nil {
|
2023-11-09 13:39:46 +09:00
|
|
|
|
log.Printf("Pixivから画像を受取に失敗: %v", err)
|
|
|
|
|
http.Error(w, "Pixivから画像を受取に失敗。", http.StatusInternalServerError)
|
2023-05-17 09:50:17 +09:00
|
|
|
|
return
|
|
|
|
|
}
|
2023-11-09 13:39:46 +09:00
|
|
|
|
defer resp.Body.Close()
|
|
|
|
|
serveImage(w, resp)
|
|
|
|
|
}
|
2023-05-17 09:50:17 +09:00
|
|
|
|
|
2023-11-09 13:39:46 +09:00
|
|
|
|
func imgproxy(w http.ResponseWriter, r *http.Request) {
|
|
|
|
|
// URLを修正
|
|
|
|
|
nurl := r.URL.Path[1:]
|
|
|
|
|
if nurl == "" {
|
|
|
|
|
fmt.Fprintf(
|
|
|
|
|
w,
|
|
|
|
|
`<style>body { background: #1b0326; color: #f545f5; } a { color: #ff88ff; }</style><p style="text-align: center;"><a href="https://technicalsuwako.moe/blog/%s-%s">%s-%s</a> | <a href="https://gitler.moe/suwako/%s">Git</a> | <a href="https://076.moe/">076</a></p>`,
|
|
|
|
|
sofname,
|
|
|
|
|
strings.ReplaceAll(version, ".", ""),
|
|
|
|
|
sofname,
|
|
|
|
|
version,
|
|
|
|
|
sofname,
|
|
|
|
|
)
|
2023-05-17 09:50:17 +09:00
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
2023-11-09 13:39:46 +09:00
|
|
|
|
// Pixivの場合
|
|
|
|
|
if strings.Contains(nurl, "s.pximg.net") || strings.Contains(nurl, "i.pximg.net") {
|
|
|
|
|
req, _ := http.NewRequest("GET", "https://" + nurl, nil)
|
|
|
|
|
req.Header.Set("Referer", "https://www.pixiv.net/")
|
|
|
|
|
pixivImg(w, req)
|
|
|
|
|
return
|
|
|
|
|
}
|
2023-05-17 09:50:17 +09:00
|
|
|
|
|
2023-11-09 13:39:46 +09:00
|
|
|
|
var img *http.Response
|
|
|
|
|
var err error
|
|
|
|
|
|
|
|
|
|
// HTTPリク
|
|
|
|
|
img, err = fetchImage("https://" + nurl, nurl)
|
|
|
|
|
if err != nil {
|
|
|
|
|
img, err = fetchImage("http://" + nurl, nurl)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Printf("画像を受け取りに失敗: %v", err)
|
|
|
|
|
http.Error(w, "画像を受け取りに失敗。", http.StatusInternalServerError)
|
|
|
|
|
return
|
2023-05-17 09:50:17 +09:00
|
|
|
|
}
|
|
|
|
|
|
2023-11-09 13:39:46 +09:00
|
|
|
|
defer img.Body.Close()
|
|
|
|
|
serveImage(w, img)
|
2023-05-17 09:50:17 +09:00
|
|
|
|
}
|
|
|
|
|
|
2023-11-09 13:39:46 +09:00
|
|
|
|
func main() {
|
2023-05-17 09:50:17 +09:00
|
|
|
|
http.HandleFunc("/", imgproxy)
|
2023-11-09 13:39:46 +09:00
|
|
|
|
fmt.Println("http://0.0.0.0:9810 でサーバーを実行中。終了するには、CTRL+Cを押して下さい。")
|
|
|
|
|
http.ListenAndServe("0.0.0.0:9810", nil)
|
2023-05-17 09:50:17 +09:00
|
|
|
|
}
|