audio.go 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. package metadata
  2. import (
  3. "bytes"
  4. "image"
  5. "image/jpeg"
  6. "os"
  7. "path/filepath"
  8. "github.com/dhowden/tag"
  9. "github.com/nfnt/resize"
  10. "github.com/oliamb/cutter"
  11. )
  12. func generateThumbnailForAudio(cacheFolder string, file string, generateOnly bool) (string, error) {
  13. //This extension is supported by id4. Call to library
  14. f, err := os.Open(file)
  15. defer f.Close()
  16. if err != nil {
  17. return "", err
  18. }
  19. m, err := tag.ReadFrom(f)
  20. if err != nil {
  21. return "", err
  22. }
  23. if m.Picture() != nil {
  24. //Convert the picture bytecode to image object
  25. img, _, err := image.Decode(bytes.NewReader(m.Picture().Data))
  26. if err != nil {
  27. //Fail to convert this image. Continue next one
  28. return "", err
  29. }
  30. //Create an empty file
  31. out, _ := os.Create(cacheFolder + filepath.Base(file) + ".jpg")
  32. defer out.Close()
  33. b := img.Bounds()
  34. imgWidth := b.Max.X
  35. imgHeight := b.Max.Y
  36. //Resize the albumn image
  37. var m image.Image
  38. if imgWidth > imgHeight {
  39. m = resize.Resize(0, 480, img, resize.Lanczos3)
  40. } else {
  41. m = resize.Resize(480, 0, img, resize.Lanczos3)
  42. }
  43. //Crop out the center
  44. croppedImg, err := cutter.Crop(m, cutter.Config{
  45. Width: 480,
  46. Height: 480,
  47. Mode: cutter.Centered,
  48. })
  49. //Write the cache image to disk
  50. jpeg.Encode(out, croppedImg, nil)
  51. if !generateOnly {
  52. //return the image as well
  53. ctx, err := getImageAsBase64(cacheFolder + filepath.Base(file) + ".jpg")
  54. return ctx, err
  55. }
  56. }
  57. return "", nil
  58. }