83 lines
1.2 KiB
Go
83 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/sha1"
|
|
"encoding/hex"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
)
|
|
|
|
var (
|
|
token = flag.String("token", "", "Upload token")
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
args := flag.Args()
|
|
if len(args) != 2 {
|
|
fmt.Print("source file and target URL are required")
|
|
os.Exit(1)
|
|
}
|
|
|
|
inPath := args[0]
|
|
outURL := args[1]
|
|
|
|
in, err := os.Open(inPath)
|
|
fail(err)
|
|
|
|
defer in.Close()
|
|
|
|
// hash the file
|
|
log.Print("hashing...")
|
|
h := sha1.New()
|
|
_, err = io.Copy(h, in)
|
|
fail(err)
|
|
|
|
sha1Hex := hex.EncodeToString(h.Sum(nil))
|
|
|
|
log.Print("SHA1 of source: ", sha1Hex)
|
|
|
|
// check remote
|
|
head, err := http.Head(outURL)
|
|
fail(err)
|
|
|
|
if head.StatusCode == 200 {
|
|
remoteSHA1 := head.Header.Get("X-Content-SHA1")
|
|
if sha1Hex == remoteSHA1 {
|
|
log.Print("remote is already up-to-date")
|
|
return
|
|
}
|
|
}
|
|
|
|
// rewind
|
|
_, err = in.Seek(0, os.SEEK_SET)
|
|
fail(err)
|
|
|
|
// upload
|
|
req, err := http.NewRequest("POST", outURL, in)
|
|
fail(err)
|
|
|
|
req.Header.Set("X-Content-SHA1", sha1Hex)
|
|
|
|
log.Print("uploading...")
|
|
resp, err := http.DefaultClient.Do(req)
|
|
fail(err)
|
|
|
|
if resp.StatusCode != http.StatusCreated {
|
|
log.Fatalf("unexpected HTTP status: %s", resp.Status)
|
|
}
|
|
|
|
log.Print("uploaded successfully")
|
|
}
|
|
|
|
func fail(err error) {
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|