【課題１】画像変換コマンドを作ろう

kadai1/imgconv/.gitignore

@@ -0,0 +1 @@
+.idea

kadai1/imgconv/Makefile

@@ -0,0 +1,4 @@
+.PHONY: fmt
+fmt: ## Format source code
+	@which goimports > /dev/null || GO111MODULE=off go get -u golang.org/x/tools/cmd/goimports
+	@goimports -d -w $(shell find . -type f -name '*.go')

kadai1/imgconv/cmd/imgconv/main.go

@@ -0,0 +1,34 @@
+package main
+
+import (
+	"flag"
+	"log"
+	"os"
+
+	"github.com/gopherdojo/dojo7/kadai1/sabe/pkg/imgconv"
+)
+
+func main() {
+	if err := exec(); err != nil {
+		log.Println(err)
+		os.Exit(1)
+	}
+	os.Exit(0)
+}
+
+func exec() error {
+	var (
+		dir     string
+		fromExt string
+		toExt   string
+	)
+	flag.StringVar(&dir, "d", ".", "target directory flag")
+	flag.StringVar(&fromExt, "f", "jpg", "before convert ext flag")
+	flag.StringVar(&toExt, "t", "png", "after convert ext flag")
+	flag.Parse()
+	imageConfList, err := imgconv.NewImageConfList(dir, fromExt, toExt)
+	if err != nil {
+		return err
+	}
+	return imageConfList.Convert()
+}

kadai1/imgconv/go.mod

@@ -0,0 +1,3 @@
+module github.com/gopherdojo/dojo7/kadai1/sabe
+
+go 1.12

kadai1/imgconv/go.sum

@@ -0,0 +1 @@
+github.com/gopherdojo/dojo7 v0.0.0-20190903074013-69385d40c994 h1:UbdhtSFDdFiSYOb/TkEmlRIUFQRrBlKOzVDY3+SXaSw=

kadai1/imgconv/pkg/imgconv/convert.go

@@ -0,0 +1,32 @@
+package imgconv
+
+import (
+	"image/gif"
+	_ "image/gif"
+	"image/jpeg"
+	_ "image/jpeg"
+	"image/png"
+	_ "image/png"
+)
+
+func (l ImageConfList) Convert() error {
+	for _, v := range l {
+		err := v.Convert()
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func (imgCf ImageConf) Convert() error {
+	switch imgCf.ToExt {
+	case "png":
+		return png.Encode(imgCf.ToFile, imgCf.DecodedFile)
+	case "gif":
+		return gif.Encode(imgCf.ToFile, imgCf.DecodedFile, nil)
+	case "jpg", "jpeg":
+		return jpeg.Encode(imgCf.ToFile, imgCf.DecodedFile, nil)
+	}
+	return nil
+}

kadai1/imgconv/pkg/imgconv/main.go

@@ -0,0 +1,115 @@
+package imgconv
+
+import (
+	"fmt"
+	"image"
+	"io"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+type ImageConf struct {
+	FromExt     string
+	ToExt       string
+	DecodedFile image.Image
+	ToFile      io.Writer
+}
+
+type ImageConfList []ImageConf
+
+// create ImageConfList
+func NewImageConfList(dir, fromExt, toExt string) (ImageConfList, error) {
+	// read dir
+	paths, err := readDir(dir, strings.ToLower(fromExt))
+	if err != nil {
+		return nil, err
+	}
+	imgCfList := make(ImageConfList, len(paths))
+	for i, path := range paths {
+		imgCf := ImageConf{
+			FromExt: strings.ToLower(fromExt),
+			ToExt:   strings.ToLower(toExt),
+		}
+
+		// set decode file
+		err = imgCf.SetDecodedFile(path)
+		if err != nil {
+			return nil, err
+		}
+
+		// set output file
+		err = imgCf.SetToFile(getConvertedPath(path, strings.ToLower(toExt)))
+		if err != nil {
+			return nil, err
+		}
+
+		imgCfList[i] = imgCf
+	}
+	return imgCfList, nil
+}
+
+// set ImageConf.DecodedFile
+func (imgCf *ImageConf) SetDecodedFile(path string) (err error) {
+	// open file
+	file, err := os.Open(path)
+	if err != nil {
+		return
+	}
+	defer func() {
+		err = file.Close()
+	}()
+
+	// decode file
+	img, _, err := image.Decode(file)
+	if err != nil {
+		return
+	}
+	imgCf.DecodedFile = img
+	return
+}
+
+// set ImageConf.ToFile
+func (imgCf *ImageConf) SetToFile(path string) error {
+	// create output file
+	toFile, err := os.Create(path)
+	if err != nil {
+		return err
+	}
+	imgCf.ToFile = toFile
+	return nil
+}
+
+//  recursively read directory
+func readDir(dir, fromExt string) ([]string, error) {
+	files, err := ioutil.ReadDir(dir)
+	if err != nil {
+		return nil, err
+	}
+
+	var paths []string
+	for _, file := range files {
+		if file.IsDir() {
+			paths2, err := readDir(filepath.Join(dir, file.Name()), fromExt)
+			if err != nil {
+				return nil, err
+			}
+			paths = append(paths, paths2...)
+			continue
+		}
+		extStart := strings.LastIndex(file.Name(), ".")
+		if extStart > 0 && strings.ToLower(file.Name()[extStart+1:]) == fromExt {
+			paths = append(paths, filepath.Join(dir, file.Name()))
+		} else {
+			continue
+		}
+	}
+	return paths, nil
+}
+
+// get converted file path
+func getConvertedPath(path, toExt string) string {
+	extStart := strings.LastIndex(path, ".")
+	return fmt.Sprintf("%s.%s", path[:extStart], toExt)
+}