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
}