catbase/plugins/emojy/web.go

package emojy

import (
	"embed"
	"encoding/json"
	"fmt"
	"github.com/ggicci/httpin"
	"io"
	"io/ioutil"
	"net/http"
	"os"
	"path"
	"path/filepath"
	"strings"

	"github.com/go-chi/chi/v5"
	"github.com/rs/zerolog/log"
)

//go:embed *.html
var embeddedFS embed.FS

func (p *EmojyPlugin) registerWeb() {
	r := chi.NewRouter()
	r.HandleFunc("/all", p.handleAll)
	r.HandleFunc("/allFiles", p.handleAllFiles)
	r.With(httpin.NewInput(UploadReq{})).
		Post("/upload", p.handleUpload)
	r.With(httpin.NewInput(EmojyReq{})).
		HandleFunc("/file/{name}", p.handleEmojy)
	r.HandleFunc("/stats", p.handleStats)
	r.HandleFunc("/list", p.handleList)
	r.HandleFunc("/new", p.handleUploadForm)
	r.HandleFunc("/", p.handleStats)
	p.b.GetWeb().RegisterWebName(r, "/emojy", "Emojys")
}

type emojyMap map[string][]EmojyCount

func (p *EmojyPlugin) handleUploadForm(w http.ResponseWriter, r *http.Request) {
	p.b.GetWeb().Index("Emojy", p.uploadIndex()).Render(r.Context(), w)
}

func (p *EmojyPlugin) handleList(w http.ResponseWriter, r *http.Request) {
	threshold := p.c.GetInt("emojy.statthreshold", 1)
	emojy, err := p.allCountsFlat(threshold)
	if err != nil {
		fmt.Fprintf(w, "Error: %s", err)
		return
	}
	p.b.GetWeb().Index("Emojy", p.listTempl(emojy)).Render(r.Context(), w)
}

func (p *EmojyPlugin) handleStats(w http.ResponseWriter, r *http.Request) {
	threshold := p.c.GetInt("emojy.statthreshold", 1)
	emojy, err := p.allCounts(threshold)
	if err != nil {
		w.WriteHeader(500)
		log.Error().Err(err).Msgf("handleAll")
		out, _ := json.Marshal(struct{ err error }{err})
		w.Write(out)
		return
	}
	p.b.GetWeb().Index("Emojy", p.statsIndex(emojy)).Render(r.Context(), w)
}

func (p *EmojyPlugin) handlePage(file string) func(w http.ResponseWriter, r *http.Request) {
	return func(w http.ResponseWriter, r *http.Request) {
		index, _ := embeddedFS.ReadFile(file)
		w.Write(index)
	}
}

func (p *EmojyPlugin) handleAll(w http.ResponseWriter, r *http.Request) {
	threshold := p.c.GetInt("emojy.statthreshold", 1)
	emojy, err := p.allCounts(threshold)
	if err != nil {
		w.WriteHeader(500)
		log.Error().Err(err).Msgf("handleAll")
		out, _ := json.Marshal(struct{ err error }{err})
		w.Write(out)
	}
	out, _ := json.Marshal(emojy)
	w.Write(out)
}

func (p *EmojyPlugin) handleAllFiles(w http.ResponseWriter, r *http.Request) {
	_, urlMap, err := AllFiles(p.emojyPath, p.baseURL)
	if err != nil {
		w.WriteHeader(500)
		out, _ := json.Marshal(struct{ err error }{err})
		w.Write(out)
		return
	}
	enc := json.NewEncoder(w)
	err = enc.Encode(urlMap)
	if err != nil {
		w.WriteHeader(500)
		out, _ := json.Marshal(struct{ err error }{err})
		w.Write(out)
	}
}

type UploadReq struct {
	Password   string       `in:"form=password"`
	Attachment *httpin.File `in:"form=attachment"`
}

func (p *EmojyPlugin) handleUpload(w http.ResponseWriter, r *http.Request) {
	input := r.Context().Value(httpin.Input).(*UploadReq)
	log.Printf("handleUpload: %#v", input)
	newFilePath, err := p.FileSave(input)
	if err != nil {
		log.Error().Err(err).Msgf("could not upload file")
		w.WriteHeader(500)
		fmt.Fprintf(w, "Error with file upload")
		return
	}
	log.Debug().Msgf("uploaded file to %s", newFilePath)
	w.WriteHeader(200)
	fmt.Fprintf(w, "success")
}

func (p *EmojyPlugin) FileSave(input *UploadReq) (string, error) {
	if !p.b.CheckPassword("", input.Password) {
		return "", fmt.Errorf("incorrect password")
	}

	file := input.Attachment
	emojyFileName := file.Filename()
	emojyName := strings.TrimSuffix(emojyFileName, filepath.Ext(emojyFileName))
	if ok, _, _, _ := p.isKnownEmojy(emojyName); ok {
		return "", fmt.Errorf("emojy already exists")
	}
	contentType := file.MIMEHeader().Get("Content-Type")
	if !strings.HasPrefix(contentType, "image") {
		return "", fmt.Errorf("incorrect mime type - given: %s", contentType)
	}
	fullPath := filepath.Clean(filepath.Join(p.emojyPath, emojyFileName))
	_ = os.MkdirAll(p.emojyPath, os.ModePerm)
	log.Debug().Msgf("trying to create/open file: %s", fullPath)
	outFile, err := os.OpenFile(fullPath, os.O_WRONLY|os.O_CREATE, os.ModePerm)
	if err != nil {
		return "", err
	}
	defer outFile.Close()
	inFile, _ := file.ReadAll()
	stream, _ := file.OpenReceiveStream()
	_, err = outFile.Write(inFile)
	_, err = io.Copy(outFile, stream)
	if err != nil {
		return "", err
	}
	return emojyFileName, nil
}

type EmojyReq struct {
	Name string `in:"path=name"`
}

func (p *EmojyPlugin) handleEmojy(w http.ResponseWriter, r *http.Request) {
	input := r.Context().Value(httpin.Input).(*EmojyReq)
	contents, err := ioutil.ReadFile(path.Join(p.emojyPath, input.Name))
	if err != nil {
		w.WriteHeader(404)
		out, _ := json.Marshal(struct{ err error }{err})
		w.Write(out)
	}
	w.Write(contents)
}