WIP: Connecter /admin/documents/upload à handleV1DocumentCreate

This commit is contained in:
Victor Lacasse-Beaudoin 2023-04-28 15:57:09 -04:00
parent fd3eebb68c
commit 07005c8753
3 changed files with 188 additions and 8 deletions

View file

@ -1,9 +1,13 @@
package api package api
import ( import (
"bytes"
"encoding/json"
"errors" "errors"
"fmt" "fmt"
"io/ioutil" "io/ioutil"
"log"
"mime/multipart"
"net/http" "net/http"
) )
@ -17,6 +21,9 @@ type API struct {
type APIOptions struct { type APIOptions struct {
KeyAuth bool KeyAuth bool
Key string Key string
BasicAuth bool
Username string
Password string
} }
func New(protocol, host string, port int, opts APIOptions) (*API, error) { func New(protocol, host string, port int, opts APIOptions) (*API, error) {
@ -97,5 +104,133 @@ func (a *API) Call(method, route string) ([]byte, error) {
} }
return respBody, nil return respBody, nil
} }
//return nil, errors.New(fmt.Sprintf("method must be 'GET' or 'DELETE', got '%s'", method))
return nil, errors.New(fmt.Sprintf("method must be 'GET' or 'DELETE', got '%s'", method)) return nil, errors.New(fmt.Sprintf("method must be 'GET' or 'DELETE', got '%s'", method))
} }
func (a *API) UploadDocument(bucket string, file_header *multipart.FileHeader) (string, error) {
endpoint := fmt.Sprintf("%s://%s:%d",
a.Protocol,
a.Host,
a.Port,
)
route := fmt.Sprintf("/v1/bucket/%s", bucket)
request := fmt.Sprintf("%s%s", endpoint, route)
client := &http.Client{}
// set the HTTP method, url, and request body
req, err := http.NewRequest(http.MethodPost, request, nil)
if err != nil {
return "", err
}
if a.Opts.KeyAuth {
req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", a.Opts.Key))
}
req.Header.Add("Content-Type", "multipart/form-data")
// CreateFormFile with multipart.NewWriter
//TODO
/*
buf := new(bytes.Buffer)
form := multipart.NewWriter(buf)
form.CreateFormFile("document", )
*/
//TODO
log.Println("req: ", req)
file, file_header, err := req.FormFile("document")
if err != nil {
log.Println("Error during http#Request.Formfile")
return "", err
}
log.Println("file: ", file, "file_header: ", file_header)
log.Println("file_header: ", file_header)
resp, err := client.Do(req)
if err != nil {
log.Println("Error during http#Client.Do")
return "", err
}
var res map[string]interface{}
json.NewDecoder(resp.Body).Decode(&res)
return fmt.Sprintf("%s", res), nil
}
// CallWithData takes data and returns a string representing a response body.
// Can be used for POST or PUT methods
func (a *API) CallWithData(method, route string, data []byte) (string, error) {
endpoint := fmt.Sprintf("%s://%s:%d",
a.Protocol,
a.Host,
a.Port,
)
request := fmt.Sprintf("%s%s", endpoint, route)
switch method {
case http.MethodPost:
// initialize http client
client := &http.Client{}
// set the HTTP method, url, and request body
req, err := http.NewRequest(http.MethodPost, request, bytes.NewBuffer(data))
if err != nil {
return "", err
}
if a.Opts.KeyAuth {
req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", a.Opts.Key))
}
// set the request header Content-Type for json
req.Header.Set("Content-Type", "application/json; charset=utf-8")
resp, err := client.Do(req)
if err != nil {
return "", err
}
var res map[string]interface{}
json.NewDecoder(resp.Body).Decode(&res)
return fmt.Sprintf("%s\n", res["message"]), nil
/*
case http.MethodPut:
// initialize http client
client := &http.Client{}
// set the HTTP method, url, and request body
req, err := http.NewRequest(http.MethodPut, request, bytes.NewBuffer(data))
if err != nil {
return "", err
}
if a.Opts.KeyAuth {
req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", a.Opts.Key))
}
// set the request header Content-Type for json
//req.Header.Set("Content-Type", "application/json; charset=utf-8")
resp, err := client.Do(req)
if err != nil {
return "", err
}
var res map[string]interface{}
json.NewDecoder(resp.Body).Decode(&res)
return fmt.Sprintf("%s\n", res["message"]), nil
*/
}
//return "", errors.New(fmt.Sprintf("method must be 'POST' or 'PUT', got '%s'", method))
return "", errors.New(fmt.Sprintf("method must be 'POST', got '%s'", method))
}

View file

@ -388,7 +388,11 @@ func handleV1DocumentCreate(c echo.Context) error {
form_file, err := c.FormFile("file") form_file, err := c.FormFile("file")
if err != nil { if err != nil {
return err log.Println(err)
return c.JSON(http.StatusBadRequest, map[string]interface{}{
"message": "Error during handleV1DocumentCreate's echo#Context.FormFile",
"error": err,
})
} }
allowed := false allowed := false
@ -729,13 +733,41 @@ func handleAdminDocumentsUpload(c echo.Context) error {
} }
func handleAdminDocumentsUploadPOST(c echo.Context) error { func handleAdminDocumentsUploadPOST(c echo.Context) error {
data := struct { client, err := api.New("http", "localhost", viper.GetInt("server.port"), api.APIOptions{
Message string KeyAuth: viper.GetBool("server.api.auth"),
}{ Key: viper.GetString("server.api.key"),
Message: "Pas implémenté", BasicAuth: viper.GetBool("server.admin.auth"),
Username: viper.GetString("server.admin.username"),
Password: viper.GetString("server.admin.password"),
})
if err != nil {
return c.JSON(http.StatusNotFound, map[string]string{"message": "Not Found"})
} }
return c.Render(http.StatusOK, "admin-upload-html", data) bucket := c.FormValue("bucket")
document, err := c.FormFile("document")
if err != nil {
//return c.JSON(http.StatusBadRequest, map[string]string{"message": "Error during file parse"})
return c.JSON(http.StatusNotFound, map[string]string{"message": "Error during echo#Context.FormFile", "error": err.Error()})
}
/*
file, err := document.Open()
if err != nil {
//return c.JSON(http.StatusNotFound, map[string]string{"message": "Not Found"})
return c.JSON(http.StatusNotFound, map[string]string{"message": "Error during file.Open()", "error": err.Error()})
}
defer file.Close()
*/
response, err := client.UploadDocument(bucket, document)
if err != nil {
return c.JSON(http.StatusNotFound, map[string]string{"message": "Error duing api#client.UploadDocument", "error": err.Error()})
//return c.JSON(http.StatusNotFound, map[string]string{"message": "Not Found"})
}
return c.Render(http.StatusOK, "admin-upload-html", struct{ Message string }{Message: response})
} }
// CSS Handlers // CSS Handlers

View file

@ -9,6 +9,19 @@
<body> <body>
{{ template "header-html" }} {{ template "header-html" }}
<h1>Upload</h1> <h1>Upload</h1>
<form action="/admin/documents/upload" method="post" enctype="multipart/form-data">
<label for="bucket">Type de document:</label>
<select name="bucket" id="bucket">
<option value="proces-verbaux">Procès verbaux</option>
<option value="politiques-et-reglements">Politiques et Règlements</option>
</select>
<br>
Document: <input type="file" name="document">
<br>
<br>
<input type="submit" value="Submit">
</form>
<p>{{ .Message }}</p> <p>{{ .Message }}</p>
</body> </body>
</html> </html>