Initial commit

server/responses.go

@@ -0,0 +1,12 @@
+package server
+
+type ResponseIndex struct {
+	Name      string `json:"name"`
+	Version   string `json:"version"`
+	GoVersion string `json:"go_version"`
+	Commit    string `json:"commit"`
+}
+
+type ResponseAPIPost struct {
+	ID string `json:"id"`
+}

server/server.go

@@ -0,0 +1,147 @@
+package server
+
+import (
+	"encoding/json"
+	"io"
+	"log"
+	"mime"
+	"mime/multipart"
+	"net/http"
+	"runtime"
+	"runtime/debug"
+	"strings"
+
+	"git.t-juice.club/torjus/minipaste/store"
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
+)
+
+const Version = "v0.1.0"
+
+type Server struct {
+	store store.Store
+	http.Server
+}
+
+func NewServer(s store.Store) *Server {
+	srv := &Server{store: s}
+	r := chi.NewRouter()
+
+	r.Use(middleware.RealIP)
+	r.Use(middleware.Recoverer)
+	r.Use(middleware.Logger)
+
+	r.Get("/", srv.HandlerIndexGet)
+	r.Route("/api", func(r chi.Router) {
+		r.Get("/{id}", srv.HandlerAPIGet)
+		r.Post("/", srv.HandlerAPIPost)
+	})
+
+	srv.Handler = r
+
+	return srv
+}
+
+func (s *Server) HandlerIndexGet(w http.ResponseWriter, r *http.Request) {
+	bi, ok := debug.ReadBuildInfo()
+	if !ok {
+		panic("not ok")
+	}
+	var commitHash string
+	for i := range bi.Settings {
+		if bi.Settings[i].Key == "vcs.revision" {
+			commitHash = bi.Settings[i].Value
+		}
+	}
+	resp := &ResponseIndex{
+		Name:      "minipaste",
+		Version:   "v0.0.0",
+		Commit:    commitHash,
+		GoVersion: runtime.Version(),
+	}
+	encoder := json.NewEncoder(w)
+	if err := encoder.Encode(resp); err != nil {
+		log.Panicf("Error encoding response: %s", err)
+	}
+}
+
+func (s *Server) HandlerAPIPost(w http.ResponseWriter, r *http.Request) {
+	// Check if multipart
+	mediaType, params, err := mime.ParseMediaType(r.Header.Get("Content-Type"))
+	if err != nil {
+		log.Panicf("Error parsing media type: %s", err)
+	}
+	log.Printf("mt: %s", mediaType)
+
+	if strings.HasPrefix(mediaType, "multipart/") {
+		// Is multipart
+		mr := multipart.NewReader(r.Body, params["boundary"])
+		for {
+			p, err := mr.NextPart()
+			if err == io.EOF {
+				break
+			}
+			if err != nil {
+				log.Fatal(err)
+			}
+			defer p.Close()
+
+			id, err := s.store.Add(p)
+			if err != nil {
+				log.Panicf("Error storing file: %s", err)
+			}
+			log.Printf("Stored file with id %s", id)
+
+			resp := &ResponseAPIPost{ID: id}
+			encoder := json.NewEncoder(w)
+
+			if err := encoder.Encode(resp); err != nil {
+				log.Panicf("Error encoding response: %s", err)
+			}
+		}
+	} else {
+		// Is not multipart
+		if mediaType == "application/x-www-form-urlencoded" {
+			if err := r.ParseForm(); err != nil {
+				log.Printf("Error parsing form: %s", err)
+			}
+			for key := range r.Form {
+				file, _, err := r.FormFile(key)
+				if err != nil {
+					log.Panicf("Error parsing formfile: %s", err)
+				}
+				defer file.Close()
+
+				id, err := s.store.Add(file)
+				if err != nil {
+					log.Panicf("Error storing file: %s", err)
+				}
+				log.Printf("Stored file with id %s", id)
+
+				resp := &ResponseAPIPost{ID: id}
+				encoder := json.NewEncoder(w)
+
+				if err := encoder.Encode(resp); err != nil {
+					log.Panicf("Error encoding response: %s", err)
+				}
+			}
+		}
+	}
+}
+
+func (s *Server) HandlerAPIGet(w http.ResponseWriter, r *http.Request) {
+	id := chi.URLParam(r, "id")
+	if id == "" {
+		log.Panicf("Missing id")
+	}
+
+	reader, err := s.store.Get(id)
+	if err != nil {
+		log.Panicf("No such file")
+	}
+	defer reader.Close()
+
+	if _, err := io.Copy(w, reader); err != nil {
+		log.Panicf("Error writing to client: %s", err)
+	}
+}