write output as jpeg as option

dhogborg · Jul 17, 2015 · bfcc7c7 · bfcc7c7
1 parent da11b94
commit bfcc7c7
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/gopow.go b/gopow.go
@@ -66,7 +66,7 @@ func main() {
 		cli.StringFlag{
 			Name:  "format,f",
 			Value: "png",
-			Usage: "Output file format, default png",
+			Usage: "Output file format, default png [png,jpeg]",
 		},
 		cli.BoolFlag{
 			Name:  "verbose",

diff --git a/internal/gopow/gopow.go b/internal/gopow/gopow.go
@@ -3,6 +3,7 @@ package gopow
 import (
 	"fmt"
 	"image"
+	"image/jpeg"
 	"image/png"
 	"os"
 	"time"
@@ -108,7 +109,22 @@ func (g *GoPow) Write() error {
 		return err
 	}
 
-	err = png.Encode(out, g.image)
+	switch g.config.Format {
+	case "png":
+		err = png.Encode(out, g.image)
+		break
+
+	case "jpeg", "jpg":
+		opt := &jpeg.Options{
+			Quality: 98,
+		}
+		err = jpeg.Encode(out, g.image, opt)
+		break
+
+	default:
+		return fmt.Errorf("unsupported format: %s", g.config.Format)
+	}
+
 	if err != nil {
 		return err
 	}