mirror of
https://github.com/vbatts/imgsrv.git
synced 2024-11-23 16:45:39 +00:00
client can upload now, but the response body is a whole page, not just
the url
This commit is contained in:
parent
5ed865d2e4
commit
96a4b57924
4 changed files with 65 additions and 19 deletions
|
@ -1,28 +1,63 @@
|
|||
package client
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"io/ioutil"
|
||||
"mime"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"mime/multipart"
|
||||
"bytes"
|
||||
"path"
|
||||
"io"
|
||||
)
|
||||
|
||||
func PutFileFromPath(host, filename string) (path string, err error) {
|
||||
ext := filepath.Ext(filename)
|
||||
file, err := os.Open(filename)
|
||||
func NewfileUploadRequest(uri, file_path string, params map[string]string) (*http.Request, error) {
|
||||
file, err := os.Open(file_path)
|
||||
if err != nil {
|
||||
return
|
||||
return nil, err
|
||||
}
|
||||
resp, err := http.Post(host, mime.TypeByExtension(ext), bufio.NewReader(file))
|
||||
defer file.Close()
|
||||
|
||||
body := &bytes.Buffer{}
|
||||
writer := multipart.NewWriter(body)
|
||||
part, err := writer.CreateFormFile("filename", path.Base(file_path))
|
||||
if err != nil {
|
||||
return
|
||||
return nil, err
|
||||
}
|
||||
_, err = io.Copy(part, file)
|
||||
|
||||
for key, val := range params {
|
||||
_ = writer.WriteField(key, val)
|
||||
}
|
||||
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
|
||||
}
|
||||
|
||||
func PutFileFromPath(uri, file_path string, params map[string]string) (path string, err error) {
|
||||
request, err := NewfileUploadRequest(uri, file_path, params)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
client := &http.Client{}
|
||||
resp, err := client.Do(request)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
log.Printf("%#v",resp)
|
||||
defer resp.Body.Close()
|
||||
bytes, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return
|
||||
return "", err
|
||||
}
|
||||
return string(bytes), nil
|
||||
}
|
||||
|
|
|
@ -64,7 +64,7 @@ func (c *Config) Merge(other *Config) error {
|
|||
if len(other.MongoPassword) > 0 {
|
||||
c.MongoPassword = other.MongoPassword
|
||||
}
|
||||
if len(other.RemoteHost) > 0 {
|
||||
if len(other.RemoteHost) > 0 && len(c.RemoteHost) == 0 {
|
||||
c.RemoteHost = other.RemoteHost
|
||||
}
|
||||
return nil
|
||||
|
|
14
imgsrv.go
14
imgsrv.go
|
@ -13,7 +13,6 @@ import (
|
|||
"log"
|
||||
"net/url"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/vbatts/imgsrv/client"
|
||||
"github.com/vbatts/imgsrv/config"
|
||||
|
@ -79,25 +78,24 @@ func main() {
|
|||
log.Println("Please provide files to be uploaded!")
|
||||
return
|
||||
}
|
||||
_, basename := filepath.Split(PutFile)
|
||||
queryParams := "?filename=" + basename
|
||||
params := map[string]string{}
|
||||
if len(FileKeywords) > 0 {
|
||||
queryParams = queryParams + "&keywords=" + FileKeywords
|
||||
params["keywords"] = FileKeywords
|
||||
} else {
|
||||
log.Println("WARN: you didn't provide any keywords :-(")
|
||||
}
|
||||
url, err := url.Parse(DefaultConfig.RemoteHost + "/f/" + queryParams)
|
||||
url, err := url.Parse(DefaultConfig.RemoteHost + "/f/")
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return
|
||||
}
|
||||
log.Printf("POSTing: %s\n", url.String())
|
||||
url_path, err := client.PutFileFromPath(url.String(), PutFile)
|
||||
//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("New Image!: %s%s\n", DefaultConfig.RemoteHost, url_path)
|
||||
log.Printf("uploaded: %s%s\n", DefaultConfig.RemoteHost, url_path)
|
||||
}
|
||||
}
|
||||
|
||||
|
|
13
server.go
13
server.go
|
@ -219,6 +219,18 @@ func routeFilesPOST(w http.ResponseWriter, r *http.Request) {
|
|||
TimeStamp: time.Now(),
|
||||
}
|
||||
|
||||
err := r.ParseMultipartForm(maxBytes)
|
||||
if err != nil {
|
||||
serverErr(w, r, err)
|
||||
return
|
||||
}
|
||||
|
||||
// Keep it DRY?
|
||||
if r.MultipartForm != nil {
|
||||
routeUpload(w, r)
|
||||
return
|
||||
}
|
||||
|
||||
filename = r.FormValue("filename")
|
||||
if len(filename) == 0 && len(uriChunks) == 2 && len(uriChunks[1]) != 0 {
|
||||
filename = strings.ToLower(uriChunks[1])
|
||||
|
@ -728,6 +740,7 @@ func routeUpload(w http.ResponseWriter, r *http.Request) {
|
|||
}
|
||||
}
|
||||
|
||||
log.Printf("%#v", r.MultipartForm.File)
|
||||
filehdr := r.MultipartForm.File["filename"][0]
|
||||
filename := filehdr.Filename
|
||||
exists, err := du.HasFileByFilename(filename)
|
||||
|
|
Loading…
Reference in a new issue