all: export Render API

Author: changkun

Makefile

@@ -3,9 +3,9 @@
 # by a GNU GPL-3.0 license that can be found in the LICENSE file.
 
 all:
-	GOOS=linux go build -mod=vendor
+	GOOS=linux go build -mod=vendor ./cmd/code2img
 	docker build -t code2img .
-up: down
+up:
 	docker-compose -f docker-compose.yml up -d
 down:
 	docker-compose -f docker-compose.yml down

README.md

@@ -1,8 +1,27 @@
-# code2img
+# code2img [![PkgGoDev](https://pkg.go.dev/badge/golang.design/x/code2img)](https://pkg.go.dev/golang.design/x/code2img) ![](https://changkun.de/urlstat?mode=github&repo=golang-design/code2img)
 
 a carbon service wrapper
 
-## iOS Shortcut
+```go
+import "golang.design/x/code2img"
+```
+
+## API Usage
+
+Just one API `code2img.Render`, to use it:
+
+```go
+b, err := code2img.Render(`import "golang.design/x/code2img"`)
+if err != nil {
+    panic(err)
+}
+
+os.WriteFile("code.png", b, os.ModePerm)
+```
+
+## Service Usage
+
+### iOS Shortcut
 
 Basic usage notes:
 
@@ -15,7 +34,7 @@ Demo:
 
 ![](./demo.gif)
 
-## API
+### Server API
 
 ```
 POST golang.design/api/v1/code2img
@@ -36,6 +55,13 @@ You can also access the code text:
 https://golang.design/api/v1/code2img/data/code/06ad29c5-2989-4a8e-8cd2-1ce63e36167b.go
 ```
 
+### Deploy Instructions
+
+```sh
+make
+make up
+```
+
 ## License
 
 © 2020 The golang.design Initiative Authors.

main.go renamed to code2img.go

@@ -2,86 +2,25 @@
 // All rights reserved. Use of this source code is governed
 // by a GNU GPL-3.0 license that can be found in the LICENSE file.
 
-package main
+package code2img
 
 import (
 	"context"
 	"fmt"
-	"io/ioutil"
 	"math"
-	"net/http"
 	"net/url"
-	"os"
-	"os/signal"
-	"syscall"
 	"time"
 
 	"github.com/chromedp/cdproto/cdp"
 	"github.com/chromedp/cdproto/dom"
 	"github.com/chromedp/cdproto/emulation"
 	"github.com/chromedp/cdproto/page"
 	"github.com/chromedp/chromedp"
-	"github.com/gin-gonic/gin"
-	"github.com/google/uuid"
-	"github.com/sirupsen/logrus"
 )
 
-func main() {
-	router := gin.Default()
-	router.Static("/api/v1/code2img/data/code", "./data/code")
-	router.Static("/api/v1/code2img/data/images", "./data/images")
-	router.POST("/api/v1/code2img", code2img)
-	s := &http.Server{Addr: ":8080", Handler: router}
-
-	go func() {
-		if err := s.ListenAndServe(); err != nil && err != http.ErrServerClosed {
-			logrus.Fatalf("listen: %s\n", err)
-		}
-	}()
-
-	quit := make(chan os.Signal)
-	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
-	<-quit
-	logrus.Println("shutting down...")
-
-	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
-	defer cancel()
-	if err := s.Shutdown(ctx); err != nil {
-		logrus.Fatal("forced to shutdown: ", err)
-	}
-	logrus.Println("server exiting, good bye!")
-}
-
-type form struct {
-	Code string `json:"code"`
-}
-
-func code2img(c *gin.Context) {
-	b := form{}
-	if err := c.ShouldBindJSON(&b); err != nil {
-		c.String(http.StatusBadRequest, fmt.Sprintf("Error: %s", err))
-		return
-	}
-	id := uuid.New().String()
-	gofile := "./data/code/" + id + ".go"
-
-	err := ioutil.WriteFile(gofile, []byte(b.Code), os.ModePerm)
-	if err != nil {
-		logrus.Errorf("[%s]: write file error %v", gofile, err)
-		c.String(http.StatusBadRequest, fmt.Sprintf("Error: %s", err))
-		return
-	}
-
-	err = render("./data/images/"+id+".png", b.Code)
-	if err != nil {
-		c.String(http.StatusBadRequest, fmt.Sprintf("Error: %s", err))
-		return
-	}
-
-	c.String(http.StatusOK, "https://golang.design/api/v1/code2img/data/images/"+id+".png")
-}
-
-func render(imgfile, code string) error {
+// Render renders the given code string and returns a binary buffer
+// that contains a carbon-now based image.
+func Render(code string) ([]byte, error) {
 	// https://carbon.now.sh/?
 	// bg=rgba(74%2C144%2C226%2C1)&
 	// t=material&
@@ -147,15 +86,9 @@ func render(imgfile, code string) error {
 		screenshot(sel, &picbuf, chromedp.NodeReady, chromedp.ByID),
 	})
 	if err != nil {
-		return fmt.Errorf("render task error: %w", err)
-	}
-
-	err = ioutil.WriteFile(imgfile, picbuf, os.ModePerm)
-	if err != nil {
-		logrus.Errorf("[%s]: write screenshot error %v", imgfile, err)
-		return err
+		return nil, fmt.Errorf("code2img: render task failed: %w", err)
 	}
-	return nil
+	return picbuf, nil
 }
 
 func screenshot(sel interface{}, picbuf *[]byte, opts ...chromedp.QueryOption) chromedp.QueryAction {

go.sum

@@ -51,6 +51,7 @@ github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXf
 github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
 github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
 github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
+github.com/ugorji/go v1.1.7 h1:/68gy2h+1mWMrwZFeD1kQialdSzAb432dtpeJ42ovdo=
 github.com/ugorji/go v1.1.7/go.mod h1:kZn38zHttfInRq0xu/PH0az30d+z6vm202qpg1oXVMw=
 github.com/ugorji/go/codec v1.1.7 h1:2SvQaVZ1ouYrrKKwoSk2pzd4A9evlKJb9oTL+OaLUSs=
 github.com/ugorji/go/codec v1.1.7/go.mod h1:Ax+UKWsSmolVDwsd+7N3ZtXu+yMGCf907BLYF3GoBXY=