52 lines
871 B
Go
52 lines
871 B
Go
package web
|
|
|
|
import (
|
|
"image"
|
|
"image/jpeg"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"time"
|
|
|
|
"github.com/nfnt/resize"
|
|
)
|
|
|
|
// Download used for downloading file
|
|
var Download = func(srcURL, dest string, scale bool) error {
|
|
if err := createDirectory(dest); err != nil {
|
|
return err
|
|
}
|
|
|
|
var httpClient = &http.Client{
|
|
Timeout: time.Second * 10,
|
|
}
|
|
resp, err := httpClient.Get(srcURL)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
image, _, err := image.Decode(resp.Body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if scale {
|
|
image = resize.Resize(300, 0, image, resize.Lanczos3)
|
|
}
|
|
|
|
// Create the file
|
|
file, err := os.Create(dest)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
|
|
return jpeg.Encode(file, image, nil)
|
|
}
|
|
|
|
// createDirectory creates the destination directory
|
|
func createDirectory(dest string) error {
|
|
return os.MkdirAll(path.Dir(dest), os.ModePerm)
|
|
}
|