From ac8a79db87095504bcb643c58a9cba19e1059cdf Mon Sep 17 00:00:00 2001
From: elee <eddy@gfxlabs.io>
Date: Fri, 25 Mar 2022 17:20:50 -0500
Subject: [PATCH] added cory

---
 Makefile             |   2 +-
 cmd/cory-bin/main.go | 108 +++++++++++++++++++++++++++++++++++++++++++
 readme.md            |   6 +++
 3 files changed, 115 insertions(+), 1 deletion(-)
 create mode 100644 cmd/cory-bin/main.go

diff --git a/Makefile b/Makefile
index 31463f1..318ce2f 100644
--- a/Makefile
+++ b/Makefile
@@ -3,7 +3,7 @@ img_name = cory
 img_tag = local
 
 cory: clean
-	go build  -o ./cory ./cmd/cory
+	go build  -o ./cory ./cmd/cory-bin
 
 docker:
 	DOCKER_BUILDKIT=1 docker build -f server.Dockerfile -t $(img_user)/$(img_name):$(img_tag) .
diff --git a/cmd/cory-bin/main.go b/cmd/cory-bin/main.go
new file mode 100644
index 0000000..a28ebed
--- /dev/null
+++ b/cmd/cory-bin/main.go
@@ -0,0 +1,108 @@
+package main
+
+import (
+	"bytes"
+	"flag"
+	"fmt"
+	"net/http"
+	"os"
+	"time"
+
+	"git.tuxpa.in/a/gotagonist/common/nori"
+	"git.tuxpa.in/a/gotagonist/common/utils"
+	"git.tuxpa.in/a/zlog/log"
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
+)
+
+var (
+	servePtr  = flag.Bool("serve", false, "launch web server")
+	dumpPtr   = flag.Bool("d", false, "dump nri file to zip")
+	silentPtr = flag.Bool("s", false, "silent")
+	filePtr   = flag.String("f", "example.nri", "file to load")
+	outputPtr = flag.String("o", "output.zip", "output zip file")
+)
+
+func main() {
+	flag.Parse()
+	if *servePtr {
+		serve()
+		return
+	}
+	if *dumpPtr {
+		filename := *filePtr
+		output := *outputPtr
+		if _, err := os.Stat(filename); err != nil {
+			printIf("could not find file %s\n", filename)
+			return
+		}
+		dump(filename, output)
+		return
+	}
+	flag.PrintDefaults()
+}
+
+func printIf(s string, args ...interface{}) {
+	if !*silentPtr {
+		fmt.Printf(s, args...)
+	}
+}
+
+func dump(filename string, output string) {
+	start := time.Now()
+	printIf("reading file '%s'\n", filename)
+	n, err := nori.FromFile(filename)
+	if err != nil {
+		log.Panicln("decode: %s", err)
+	}
+	out := new(bytes.Buffer)
+	printIf("rendering %d animation(s)\n", n.AnimationCount)
+	frames, err := nori.RenderAnimations(n)
+	if err != nil {
+		log.Panicln("animation: %s", err)
+	}
+	printIf("saving to %s \n", output)
+	if err := utils.ZipApngs(out, frames); err != nil {
+		log.Panicln("zipping: %s", err)
+	}
+	printIf("done in %v \n", time.Now().Sub(start))
+	os.WriteFile(output, out.Bytes(), 0740)
+}
+
+func serve() {
+	r := chi.NewRouter()
+
+	r.Use(middleware.RequestID)
+	r.Use(middleware.RealIP)
+	r.Use(middleware.Logger)
+	r.Use(middleware.Recoverer)
+
+	r.Use(middleware.Timeout(60 * time.Second))
+
+	r.Post("/animations", func(w http.ResponseWriter, r *http.Request) {
+		output_type := chi.URLParam(r, "format")
+		switch output_type {
+		case "":
+			output_type = "apng"
+		case "apng":
+		default:
+			http.Error(w, "output type not supported", 500)
+			return
+		}
+		n := nori.New()
+		if err := n.Decode(r.Body); err != nil {
+			http.Error(w, err.Error(), 500)
+		}
+		frames, err := nori.RenderAnimations(n)
+		if err != nil {
+			http.Error(w, err.Error(), 500)
+		}
+		if err := utils.ZipApngs(w, frames); err != nil {
+			http.Error(w, err.Error(), 500)
+		}
+		w.WriteHeader(200)
+	})
+
+	log.Println("listening on 3333")
+	http.ListenAndServe(":3333", r)
+}
diff --git a/readme.md b/readme.md
index 648dc77..d08c500 100644
--- a/readme.md
+++ b/readme.md
@@ -32,3 +32,9 @@ oh yeah there's also a docker image for cory lol
 ```
 make docker
 ```
+
+and to run server
+
+```
+make server-run
+```