client upload works again

This commit is contained in:
Vincent Batts 2013-10-10 10:31:27 -04:00
parent 96a4b57924
commit adbdffbfb4
3 changed files with 27 additions and 29 deletions

View File

@ -1,17 +1,18 @@
package client
import (
"bytes"
"errors"
"io"
"io/ioutil"
"mime/multipart"
"net/http"
"net/url"
"log"
"os"
"mime/multipart"
"bytes"
"path"
"io"
"path"
)
var ErrorNotOK = errors.New("HTTP Response was not 200 OK")
func NewfileUploadRequest(uri, file_path string, params map[string]string) (*http.Request, error) {
file, err := os.Open(file_path)
if err != nil {
@ -30,46 +31,37 @@ func NewfileUploadRequest(uri, file_path string, params map[string]string) (*htt
for key, val := range params {
_ = writer.WriteField(key, val)
}
contentType := writer.FormDataContentType()
_ = writer.WriteField("returnUrl", "true")
contentType := writer.FormDataContentType()
err = writer.Close()
if err != nil {
return nil, err
}
log.Println(uri)
req, err := http.NewRequest("POST", uri, body)
req.Header.Add("Content-Type", contentType)
return req, err
req, err := http.NewRequest("POST", uri, body)
req.Header.Add("Content-Type", contentType)
return req, err
}
func PutFileFromPath(uri, file_path string, params map[string]string) (path string, err error) {
request, err := NewfileUploadRequest(uri, file_path, params)
request, err := NewfileUploadRequest(uri, file_path, params)
if err != nil {
return "", err
}
client := &http.Client{}
resp, err := client.Do(request)
resp, err := client.Do(request)
if err != nil {
return "", err
}
log.Printf("%#v",resp)
defer resp.Body.Close()
defer resp.Body.Close()
bytes, err := ioutil.ReadAll(resp.Body)
if err != nil {
return "", err
}
if resp.StatusCode != 200 {
return string(bytes), ErrorNotOK
}
return string(bytes), nil
}
func hurr() {
values := make(url.Values)
values.Set("email", "anything@email.com")
values.Set("name", "bob")
values.Set("count", "1")
r, err := http.PostForm("http://example.com/form", values)
if err != nil {
return
}
_ = r
}

View File

@ -89,13 +89,12 @@ func main() {
log.Println(err)
return
}
//log.Printf("POSTing: %s\n", url.String())
url_path, err := client.PutFileFromPath(url.String(), PutFile, params)
if err != nil {
log.Println(err)
return
}
log.Printf("uploaded: %s%s\n", DefaultConfig.RemoteHost, url_path)
fmt.Printf("%s%s\n", DefaultConfig.RemoteHost, url_path)
}
}

View File

@ -729,12 +729,15 @@ func routeUpload(w http.ResponseWriter, r *http.Request) {
return
}
useRandName := false
returnUrl := false
log.Printf("%q", r.MultipartForm.Value)
for k, v := range r.MultipartForm.Value {
if k == "keywords" {
info.Keywords = append(info.Keywords, strings.Split(v[0], ",")...)
} else if k == "rand" {
useRandName = true
} else if k == "returnUrl" {
returnUrl = true
} else {
log.Printf("WARN: not sure what to do with param [%s = %s]", k, v)
}
@ -781,6 +784,10 @@ func routeUpload(w http.ResponseWriter, r *http.Request) {
n)
}
if returnUrl {
fmt.Fprintf(w, "/v/%s", filename)
return
}
http.Redirect(w, r, fmt.Sprintf("/v/%s", filename), 302)
} else {
httplog.LogRequest(r, 404)