mirror of
https://github.com/idanoo/GoScrobble.git
synced 2024-11-24 17:35:16 +00:00
128 lines
2.9 KiB
Go
128 lines
2.9 KiB
Go
package goscrobble
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"log"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/disintegration/imaging"
|
|
)
|
|
|
|
func importUploadedImage(file multipart.File, uuid string, recordType string) error {
|
|
// Create image
|
|
out, err := os.Create(DataDirectory + string(os.PathSeparator) + "img" + string(os.PathSeparator) + uuid + "_full.jpg")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer out.Close()
|
|
|
|
fileBytes, err := ioutil.ReadAll(file)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Write image
|
|
_, err = out.Write(fileBytes)
|
|
if err == nil {
|
|
// Make sure we queue it to process!
|
|
_, err = db.Exec("UPDATE `"+recordType+"` SET `img` = 'pending' WHERE `uuid` = UUID_TO_BIN(?,true)", uuid)
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
func importImage(uuid string, url string) error {
|
|
// Create image
|
|
out, err := os.Create(DataDirectory + string(os.PathSeparator) + "img" + string(os.PathSeparator) + uuid + "_full.jpg")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer out.Close()
|
|
|
|
// Get the data
|
|
resp, err := http.Get(url)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
// Check server response
|
|
if resp.StatusCode != http.StatusOK {
|
|
return fmt.Errorf("Bad response status: %s", resp.Status)
|
|
}
|
|
|
|
// Writer the body to file
|
|
_, err = io.Copy(out, resp.Body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func resizeImages() {
|
|
resizeBulk("artists")
|
|
resizeBulk("albums")
|
|
}
|
|
|
|
func resizeBulk(recordType string) {
|
|
// Fetch pending 500 at a time cause we do it every minute anyway
|
|
rows, err := db.Query("SELECT BIN_TO_UUID(`uuid`, true) FROM `" + recordType + "` WHERE `img` = 'pending' LIMIT 500")
|
|
if err != nil {
|
|
log.Printf("Failed to get pending images: %+v", err)
|
|
return
|
|
}
|
|
|
|
// Fetch pending 100 at a time
|
|
for rows.Next() {
|
|
var uuid string
|
|
err := rows.Scan(&uuid)
|
|
if err != nil {
|
|
log.Printf("Failed to fetch record image resize: %+v", err)
|
|
rows.Close()
|
|
return
|
|
}
|
|
|
|
// Run the resize to 300px
|
|
success := resizeImage(uuid, 300)
|
|
if !success {
|
|
// If we get an error.. lets just remove the image link for now so it can reimport
|
|
_, err = db.Exec("UPDATE `"+recordType+"` SET `img` = NULL WHERE `uuid` = UUID_TO_BIN(?,true)", uuid)
|
|
} else {
|
|
// Update DB to reflect complete
|
|
_, err = db.Exec("UPDATE `"+recordType+"` SET `img` = 'complete' WHERE `uuid` = UUID_TO_BIN(?,true)", uuid)
|
|
}
|
|
}
|
|
|
|
rows.Close()
|
|
}
|
|
|
|
func resizeAlbums() {
|
|
|
|
}
|
|
|
|
func resizeImage(uuid string, size int) bool {
|
|
// Open source image
|
|
src, err := imaging.Open(DataDirectory + string(os.PathSeparator) + "img" + string(os.PathSeparator) + uuid + "_full.jpg")
|
|
if err != nil {
|
|
log.Printf("Failed to open image: %+v", err)
|
|
return false
|
|
}
|
|
|
|
// Resize image to specified size
|
|
resizedImage := imaging.Resize(src, size, 0, imaging.Lanczos)
|
|
|
|
// Save resized image
|
|
err = imaging.Save(resizedImage, DataDirectory+string(os.PathSeparator)+"img"+string(os.PathSeparator)+uuid+"_300px.jpg")
|
|
if err != nil {
|
|
log.Printf("failed to save image: %v", err)
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|