imgsrv/client.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
}