From 6c3cb6066d5b16a2844c11faf2839f9f24771836 Mon Sep 17 00:00:00 2001 From: Justin Hawkins Date: Mon, 20 Feb 2017 21:54:16 +1030 Subject: [PATCH] Upload. Lacks all but the most basic of error checking --- dau.go | 80 +++++++++++++++++++++++++++++++++++++++++++++++++++------- 1 file changed, 71 insertions(+), 9 deletions(-) diff --git a/dau.go b/dau.go index e5d917f..c51e00f 100644 --- a/dau.go +++ b/dau.go @@ -8,44 +8,52 @@ import ( "os" "time" "net/http" + "log" + "io" + "bytes" + "mime/multipart" + // "json" ) var last_check = time.Now() var new_last_check = time.Now() +var webhook_url string + +type webhook_response struct { + Test string +} func main() { - webhook_url, path := parse_options() - fmt.Println(webhook_url, path) + webhook, path, watch := parse_options() + webhook_url = webhook // wander the path, forever for { err := filepath.Walk(path, check_file) - if err != nil { fmt.Printf("SHIT ERROR") } + if err != nil { log.Fatal("oh dear") } //fmt.Printf("filepath.Walk() returned %v\n", err) last_check = new_last_check - time.Sleep(500 * time.Millisecond) + time.Sleep(time.Duration(watch)*time.Second) } } - -func parse_options() (webhook_url string, path string) { +func parse_options() (webhook_url string, path string, watch int) { // Declare the flags to be used // helpFlag := getopt.Bool('h', "display help") webhookFlag := getopt.StringLong("webhook", 'w', "", "webhook URL") pathFlag := getopt.StringLong("directory", 'd', "", "directory") + watchFlag := getopt.Int16Long("watch", 's', 10, "time between scans") getopt.Parse() - return *webhookFlag, *pathFlag - + return *webhookFlag, *pathFlag, int(*watchFlag) } func check_file(path string, f os.FileInfo, err error) error { // fmt.Println("Comparing", f.ModTime(), "to", last_check, "for", path) if f.ModTime().After(last_check) && f.Mode().IsRegular() { - fmt.Println("YES", path, "is new") if file_eligible(path) { // process file @@ -70,4 +78,58 @@ func file_eligible(file string) (bool) { func process_file(file string) { fmt.Println("Uploading", file) + // resp, err := http.Post("http://example.com/upload", "image/jpeg", &buf) + + extraParams := map[string]string{ + // "username": "Some username", + } + + request, err := newfileUploadRequest(webhook_url, extraParams, "file", file) + if err != nil { + log.Fatal(err) + } + client := &http.Client{} + resp, err := client.Do(request) + if err != nil { + log.Fatal(err) + } else { + body := &bytes.Buffer{} + _, err := body.ReadFrom(resp.Body) + if err != nil { + log.Fatal(err) + } + resp.Body.Close() + fmt.Println(resp.StatusCode) + // fmt.Println(resp.Header) + // fmt.Println(body) + + } +} + +func newfileUploadRequest(uri string, params map[string]string, paramName, path string) (*http.Request, error) { + file, err := os.Open(path) + if err != nil { + return nil, err + } + defer file.Close() + + body := &bytes.Buffer{} + writer := multipart.NewWriter(body) + part, err := writer.CreateFormFile(paramName, filepath.Base(path)) + if err != nil { + return nil, err + } + _, err = io.Copy(part, file) + + for key, val := range params { + _ = writer.WriteField(key, val) + } + err = writer.Close() + if err != nil { + return nil, err + } + + req, err := http.NewRequest("POST", uri, body) + req.Header.Set("Content-Type", writer.FormDataContentType()) + return req, err }