Kadai1 tomoya-ueno-2907

kadai1/cmd/cvt/Makefile

@@ -0,0 +1,8 @@
+SERVICE_NAME=cvt
+PROJECT_ROOT=$(shell git rev-parse --show-toplevel)
+
+build:
+	go build -o $(PROJECT_ROOT)/bin/$(SERVICE_NAME)
+
+exec:
+	$(PROJECT_ROOT)/bin/$(SERVICE_NAME) -i=pkg/testdata -o=pkg/png -be=jpg -ae=png

kadai1/cmd/cvt/main.go

@@ -0,0 +1,27 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"github.com/gopherdojo/gopherdojo-studyroom/kadai1/internal/cvt"
+	"log"
+)
+
+var inputDir, outputDir, beforeExt, afterExt string
+var removeSrc bool
+
+func init() {
+	flag.StringVar(&inputDir, "i", "", "input dir")
+	flag.StringVar(&outputDir, "o", "", "output dir")
+	flag.StringVar(&beforeExt, "be", ".jpg", "before ext")
+	flag.StringVar(&afterExt, "ae", ".png", "after ext")
+	flag.BoolVar(&removeSrc, "rm", false, "remove src")
+	flag.Parse()
+}
+
+func main() {
+	c := cvt.NewImageCvtGlue(inputDir, outputDir, beforeExt, afterExt, removeSrc)
+	if err := c.Exec(); err != nil {
+		log.Fatalf("Failed to execute image convert", fmt.Sprintf("%+v", err))
+	}
+}

kadai1/go.mod

@@ -0,0 +1,6 @@
+module "github.com/gopherdojo/gopherdojo-studyroom/kadai1"
+go 1.16
+
+require (
+  github.com/pkg/errors v0.9.1
+)

kadai1/go.sum

@@ -0,0 +1,2 @@
+github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
+github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=

kadai1/internal/cvt/cvt.go

@@ -0,0 +1,201 @@
+package cvt
+
+import (
+	"fmt"
+	"github.com/gopherdojo/gopherdojo-studyroom/kadai1/pkg/errof"
+	"github.com/pkg/errors"
+	"image"
+	"image/gif"
+	"image/jpeg"
+	"image/png"
+	"os"
+	"path/filepath"
+	"runtime"
+	"strings"
+)
+
+type ImageCvtGlue struct {
+	InputDir  string
+	OutputDir string
+	BeforeExt string
+	AfterExt  string
+	RemoveSrc bool
+}
+
+func NewImageCvtGlue(
+	inputDir,
+	outputDir,
+	beforeExt,
+	afterExt string,
+	removeSrc bool,
+) *ImageCvtGlue {
+	if strings.Index(beforeExt, ".") == -1 {
+		beforeExt = fmt.Sprintf(".%s", beforeExt)
+	}
+	if strings.Index(afterExt, ".") == -1 {
+		afterExt = fmt.Sprintf(".%s", afterExt)
+	}
+	return &ImageCvtGlue{
+		InputDir:  inputDir,
+		OutputDir: outputDir,
+		BeforeExt: beforeExt,
+		AfterExt:  afterExt,
+		RemoveSrc: removeSrc,
+	}
+}
+
+// Exec:
+func (c *ImageCvtGlue) Exec() (err error) {
+	var srcPaths []string
+	if srcPaths, err = c.pathWalk(); err != nil {
+		return err
+	}
+	if err = c.convert(srcPaths); err != nil {
+		return err
+	}
+	return nil
+}
+
+// convert:
+func (c *ImageCvtGlue) convert(srcPaths []string) (err error) {
+	for _, srcPath := range srcPaths {
+		var file *os.File
+		if file, err = os.Open(srcPath); err != nil {
+			return errors.Wrap(errof.ErrOpenSrcFile, err.Error())
+		}
+		// イメージのデコード
+		var img image.Image
+		if img, _, err = image.Decode(file); err != nil {
+			return errors.Wrap(errof.ErrDecodeImage, err.Error())
+		}
+		// イメージ出力先パス
+		var dstPath string
+		if dstPath, err = c.getDstPath(file.Name()); err != nil {
+			return err
+		}
+		// イメージファイルの作成
+		var dst *os.File
+		if dst, err = os.Create(dstPath); err != nil {
+			return errors.Wrap(errof.ErrCreateDstFile, err.Error())
+		}
+		//　作成したイメージファイルにデコードしたイメージをエンコード
+		if err = encodeImage(dstPath, dst, img); err != nil {
+			return err
+		}
+		// 変換前のファイルを削除(オプション)
+		if c.RemoveSrc {
+			if err = os.Remove(srcPath); err != nil {
+				return errors.Wrap(errof.ErrRemoveSrcFile, err.Error())
+			}
+		}
+		if err = file.Close(); err != nil {
+			return errors.Wrap(errof.ErrCloseSrcFile, err.Error())
+		}
+		if err = dst.Close(); err != nil {
+			return errors.Wrap(errof.ErrCloseSrcFile, err.Error())
+		}
+	}
+	return nil
+}
+
+// pathWalk:
+func (c *ImageCvtGlue) pathWalk() (srcPaths []string, err error) {
+	rootDir := getRootDir()
+	if err = filepath.Walk(filepath.Join(rootDir, c.InputDir), func(srcPath string, info os.FileInfo, err error) error {
+		if err != nil {
+			return errors.Wrap(errof.ErrWalkingSrcPath, err.Error())
+		}
+		// ファイルが存在しているパスかどうかを確認
+		var isFile bool
+		if isFile, err = isFileExist(srcPath); err != nil {
+			return err
+		}
+		// ファイルかつ指定した拡張子であれば配列に格納
+		if isFile && filepath.Ext(srcPath) == c.BeforeExt {
+			srcPaths = append(srcPaths, srcPath)
+		}
+		return nil
+	}); err != nil {
+		return srcPaths, err
+	}
+	return srcPaths, nil
+}
+
+// getDstPath:
+func (c *ImageCvtGlue) getDstPath(path string) (dstPath string, err error) {
+	fileNameWithoutExt := filepath.Base(path[:len(path)-len(filepath.Ext(path))])
+	if c.OutputDir == "" {
+		dirPath := filepath.Dir(path)
+		return fmt.Sprintf("%s%s", filepath.Join(dirPath, fileNameWithoutExt), c.AfterExt), nil
+	}
+	rootDir := getRootDir()
+	dirPath := filepath.Join(rootDir, c.OutputDir)
+	var isDir bool
+	if isDir, err = isDirExist(dirPath); err != nil {
+		return "", err
+	}
+	// 出力先ディレクトリが存在していなければ新規作成
+	if !isDir {
+		if err := os.MkdirAll(dirPath, 0777); err != nil {
+			return "", errors.Wrap(errof.ErrCreateDirectory, err.Error())
+		}
+	}
+	return fmt.Sprintf("%s%s", filepath.Join(dirPath, fileNameWithoutExt), c.AfterExt), nil
+}
+
+// encodeImage:
+func encodeImage(dstPath string, dst *os.File, img image.Image) (err error) {
+	switch filepath.Ext(dstPath) {
+	case ".png":
+		if err = png.Encode(dst, img); err != nil {
+			return errors.Wrap(errof.ErrEncodePNGImg, err.Error())
+		}
+	case ".jpg", ".jpeg":
+		if err = jpeg.Encode(dst, img, &jpeg.Options{Quality: jpeg.DefaultQuality}); err != nil {
+			return errors.Wrap(errof.ErrEncodeJPGImg, err.Error())
+		}
+	case ".gif":
+		if err = gif.Encode(dst, img, nil); err != nil {
+			return errors.Wrap(errof.ErrEncodeGIFImg, err.Error())
+		}
+	}
+	return nil
+}
+
+// getRootDir:
+func getRootDir() string {
+	_, b, _, _ := runtime.Caller(0)
+	cvtDir := filepath.Dir(b)
+	internalDir := filepath.Dir(cvtDir)
+	return filepath.Dir(internalDir)
+}
+
+// isFileExist:
+func isFileExist(filepath string) (isFile bool, err error) {
+	var fileInfo os.FileInfo
+	if fileInfo, err = os.Stat(filepath); err != nil {
+		if os.IsNotExist(err) {
+			return false, nil
+		}
+		return false, errors.Wrap(errof.ErrGetSrcFileInfo, err.Error())
+	}
+	if fileInfo.IsDir() {
+		return false, nil
+	}
+	return true, nil
+}
+
+// isDirExist:
+func isDirExist(filepath string) (isDir bool, err error) {
+	var fileInfo os.FileInfo
+	if fileInfo, err = os.Stat(filepath); err != nil {
+		if os.IsNotExist(err) {
+			return false, nil
+		}
+		return false, errors.Wrap(errof.ErrGetDirInfo, err.Error())
+	}
+	if fileInfo.IsDir() {
+		return true, nil
+	}
+	return false, nil
+}