webp encoder

go.mod

@@ -0,0 +1,8 @@
+module webp_server_go
+
+go 1.13
+
+require (
+	github.com/chai2010/webp v1.1.0
+	github.com/gofiber/fiber v1.4.0
+)

webp_server.go

@@ -2,13 +2,19 @@ package main
 
 import (
 	"bytes"
-	"github.com/gofiber/fiber"
 	"encoding/json"
 	"fmt"
-	"strings"
-	"path"
+	"github.com/chai2010/webp"
+	"github.com/gofiber/fiber"
+	"image"
+	"image/jpeg"
+	"image/png"
+	"io/ioutil"
+	"log"
 	"os"
-	"os/exec"
+	"path"
+	"strconv"
+	"strings"
 )
 
 type Config struct {
@@ -17,7 +23,26 @@ type Config struct {
 	IMG_PATH      string
 	QUALITY       string
 	ALLOWED_TYPES []string
+}
 
+func webpEncoder(p1, p2 string, quality float32) {
+	var buf bytes.Buffer
+	var img image.Image
+	data, _ := ioutil.ReadFile(p1)
+	if strings.Contains(p1, "jpg") || strings.Contains(p1, "jpeg") {
+		img, _ = jpeg.Decode(bytes.NewReader(data))
+	} else if strings.Contains(p1, "png") {
+		img, _ = png.Decode(bytes.NewReader(data))
+	}
+
+	if err := webp.Encode(&buf, img, &webp.Options{Lossless: true, Quality: quality}); err != nil {
+		log.Println(err)
+	}
+	if err := ioutil.WriteFile(p2, buf.Bytes(), 0666); err != nil {
+		log.Println(err)
+	}
+
+	fmt.Println("Save output.webp ok")
 }
 
 func main() {
@@ -98,17 +123,9 @@ func main() {
 			os.MkdirAll(DIR_ABSOLUTE_PATH, os.ModePerm)
 
 			// cwebp -q 60 Cute-Baby-Girl.png -o Cute-Baby-Girl.webp
-			OS_CMD := exec.Command("./webp/cwebp","-q",QUALITY,IMG_ABSOLUTE_PATH,"-o",WEBP_ABSOLUTE_PATH)
-			var out bytes.Buffer
-			var stderr bytes.Buffer
-			OS_CMD.Stdout = &out
-			OS_CMD.Stderr = &stderr
-			err := OS_CMD.Run()
-			if err != nil {
-				fmt.Println(stderr.String())
-				fmt.Println(err.Error())
-			}
 
+			q, _ := strconv.ParseFloat(QUALITY, 32)
+			webpEncoder(IMG_ABSOLUTE_PATH, WEBP_ABSOLUTE_PATH, float32(q))
 			if err != nil {
 				fmt.Println(err)
 			}