gomp/imageUtils.go

package main

import (
	"image"
	"os"
	"strings"

	"github.com/bogem/id3v2"
	"github.com/mewkiz/flac"
	"github.com/mewkiz/flac/meta"
	"github.com/nfnt/resize"
	"github.com/spf13/viper"
)

func GetMp3Image(songPath, imagePath string) string {
	tag, err := id3v2.Open(songPath, id3v2.Options{Parse: true})
	if err != nil {
		return ""
	}
	defer tag.Close()
	if err != nil {
		return ""
	}
	// Read tags.
	Frames := tag.GetFrames(tag.CommonID("Attached picture"))
	var ImageData []byte
	for _, er := range Frames {
		pic, ok := er.(id3v2.PictureFrame)
		if ok {
			for _, i := range pic.Picture {
				ImageData = append(ImageData, byte(i))
			}
			imageHandler, err := os.Create(imagePath)
			if err != nil {
				return ""
			} else {
				imageHandler.Write(ImageData)
				return imagePath
			}
		}
	}
	return ""
}

func GetFlacImage(songPath, imagePath string) string {
	stream, err := flac.ParseFile(songPath)
	if err != nil {
		return ""
	}
	defer stream.Close()
	for _, block := range stream.Blocks {
		if block.Type == meta.TypePicture {
			pic := block.Body.(*meta.Picture)
			if pic.Type == 3 {
				imageHandler, err := os.Create(imagePath)
				if err != nil {
					return ""
				}
				imageHandler.Write(pic.Data)
				return imagePath
			}
		}
	}
	return ""
}

func extractImageFromFile(uri string, imagePath string) string {
	if strings.HasSuffix(uri, ".mp3") {
		imagePath := GetMp3Image(uri, imagePath)
		if imagePath == "" {
			return viper.GetString("DEFAULT_IMAGE_PATH")
		} else {
			return imagePath
		}
	} else if strings.HasSuffix(uri, ".flac") {
		imagePath := GetFlacImage(uri, imagePath)
		if imagePath == "" {
			return viper.GetString("DEFAULT_IMAGE_PATH")
		} else {
			return imagePath
		}
	} else {
		return viper.GetString("DEFAULT_IMAGE_PATH")
	}
}

func getImg(uri string) (image.Image, error) {
	f, err := os.Open(uri)
	if err != nil {
		return nil, err
	}
	defer f.Close()
	img, _, err := image.Decode(f)
	if err != nil {
		return nil, err
	}
	fw, fh := getFontWidth()
	img = resize.Resize(
		uint(float32(IMG_W)*(fw+float32(viper.GetFloat64("IMAGE_WIDTH_EXTRA_X")))), uint(float32(IMG_H)*(fh+float32(viper.GetFloat64("IMAGE_WIDTH_EXTRA_Y")))),
		img,
		resize.Bilinear,
	)
	return img, nil
}