v0.1.0

cmd/main.go

@@ -0,0 +1,97 @@
+package main
+
+import (
+	"bytes"
+	"crypto/md5"
+	"flag"
+	"io"
+	"net/http"
+	"time"
+
+	"github.com/FloatTech/ttl"
+	"github.com/fumiama/imgsz"
+	"github.com/sirupsen/logrus"
+
+	"github.com/fumiama/imoto"
+)
+
+var imgcache *ttl.Cache[uint64, *imagebody]
+
+type imagebody struct {
+	key uint64
+	typ string
+	dat []byte
+}
+
+func main() {
+	cachetime := flag.Uint("t", 60, "cache time (s)")
+	endpoint := flag.String("e", "127.0.0.1:8000", "listening endpoint")
+	flag.Parse()
+	imgcache = ttl.NewCache[uint64, *imagebody](time.Second * time.Duration(*cachetime))
+	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		m, err := imoto.GetMD5(r.URL.Path)
+		defer r.Body.Close()
+		if err != nil {
+			http.Error(w, "400 Bad Request: "+err.Error(), http.StatusBadRequest)
+			return
+		}
+		p, k := imoto.SplitMD5(m)
+		logrus.Infoln("[handle]", r.Method, r.URL.Path)
+		switch r.Method {
+		case http.MethodHead:
+			if imgcache.Get(p) == nil {
+				w.WriteHeader(http.StatusNotFound)
+				return
+			}
+			w.WriteHeader(http.StatusOK)
+		case http.MethodGet:
+			img := imgcache.Get(p)
+			if img == nil {
+				w.WriteHeader(http.StatusNotFound)
+				return
+			}
+			w.WriteHeader(http.StatusOK)
+			w.Header().Set("Content-Type", "image/"+img.typ)
+			_, _ = w.Write(img.dat)
+		case http.MethodPut:
+			data, err := io.ReadAll(r.Body)
+			if err != nil {
+				http.Error(w, "500 Internal Server Error: "+err.Error(), http.StatusInternalServerError)
+				return
+			}
+			realm := md5.Sum(data)
+			if m != realm {
+				http.Error(w, "400 Bad Request: file md5 mismatch", http.StatusBadRequest)
+				return
+			}
+			_, typ, err := imgsz.DecodeSize(bytes.NewReader(data))
+			if err != nil {
+				http.Error(w, "400 Bad Request: "+err.Error(), http.StatusBadRequest)
+				return
+			}
+			imgcache.Set(p, &imagebody{
+				key: k,
+				typ: typ,
+				dat: data,
+			})
+			w.WriteHeader(http.StatusOK)
+		case http.MethodDelete:
+			img := imgcache.Get(p)
+			if img == nil {
+				w.WriteHeader(http.StatusNotFound)
+				return
+			}
+			if k != img.key {
+				http.Error(w, "403 Forbidden", http.StatusForbidden)
+				return
+			}
+			imgcache.Delete(p)
+			w.WriteHeader(http.StatusOK)
+			w.Header().Set("Content-Type", "image/"+img.typ)
+			_, _ = w.Write(img.dat)
+		default:
+			http.Error(w, "405 Method Not Allowed", http.StatusMethodNotAllowed)
+		}
+	})
+	logrus.Errorln(http.ListenAndServe(*endpoint, nil))
+}