mirror of https://github.com/vbatts/imgsrv.git
76 lines
1.5 KiB
Go
76 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"crypto/tls"
|
|
"io/ioutil"
|
|
"log"
|
|
"mime"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
)
|
|
|
|
func PutFileFromPath(host, filename string) (path string, err error) {
|
|
ext := filepath.Ext(filename)
|
|
file, err := os.Open(filename)
|
|
if (err != nil) {
|
|
return
|
|
}
|
|
resp, err := http.Post(host, mime.TypeByExtension(ext) , bufio.NewReader(file))
|
|
if (err != nil) {
|
|
return
|
|
}
|
|
bytes, err := ioutil.ReadAll(resp.Body)
|
|
if (err != nil) {
|
|
return
|
|
}
|
|
return string(bytes), nil
|
|
}
|
|
|
|
func FetchFileFromURL(url string) (filename string, err error) {
|
|
var t time.Time
|
|
|
|
tr := &http.Transport{
|
|
TLSClientConfig: &tls.Config{ InsecureSkipVerify: true },
|
|
}
|
|
client := &http.Client{
|
|
//CheckRedirect: redirectPolicyFunc,
|
|
Transport: tr,
|
|
}
|
|
resp, err := client.Get(url)
|
|
defer resp.Body.Close()
|
|
if (err != nil) {
|
|
return
|
|
}
|
|
|
|
mtime := resp.Header.Get("last-modified")
|
|
if (len(mtime) > 0) {
|
|
t, err = time.Parse(http.TimeFormat, mtime)
|
|
if (err != nil) {
|
|
return
|
|
}
|
|
} else {
|
|
log.Println("Last-Modified not present. Using current time")
|
|
t = time.Now()
|
|
}
|
|
_, url_filename := filepath.Split(url)
|
|
|
|
log.Println(resp)
|
|
|
|
bytes, err := ioutil.ReadAll(resp.Body)
|
|
if (err != nil) {
|
|
return
|
|
}
|
|
err = ioutil.WriteFile(filepath.Join(os.TempDir(), url_filename), bytes, 0644)
|
|
if (err != nil) {
|
|
return
|
|
}
|
|
err = os.Chtimes(filepath.Join(os.TempDir(), url_filename), t, t)
|
|
|
|
// lastly, return
|
|
return filepath.Join(os.TempDir(), url_filename), nil
|
|
}
|
|
|