Compare commits

..

No commits in common. "20783738890600090b8ce41daad15e6f6719e6fd" and "54d2714dec267ead4e1c11cd5969fe5a9903db8c" have entirely different histories.

2 changed files with 10 additions and 68 deletions

View File

@ -2,7 +2,6 @@ package main
import (
"crypto/sha1"
"crypto/sha256"
"encoding/hex"
"errors"
"fmt"
@ -32,13 +31,6 @@ type UploadCreate struct {
UploadId string `json:"upload_id"`
}
type UploadPartResult struct {
UploadId string `json:"upload_id"`
Part uint `json:"part"`
Size uint `json:"size"`
PartSha256 string `json:"sha256"`
}
type UploadFileRequest struct {
Name string
Checksum string
@ -191,70 +183,19 @@ func (c *UploadCommand) FileUpload(sum string) error {
func (c *UploadCommand) Execute(args []string) error {
cli := c.Cli()
resp, err := cli.R().SetError(&UploadError{}).SetResult(&UploadCreate{}).Post("/upload")
resp, err := cli.R().SetError(&UploadError{}).SetResult(&UploadCreate{}).Post("/upload/create")
if err != nil {
return err
}
if err, ok := resp.Error().(*UploadError); ok {
logger.Println(string(resp.Body()))
logger.Println(resp.Error(), resp.StatusCode(), resp.Header())
return errors.New(err.Error)
}
uploadId := resp.Result().(*UploadCreate).UploadId
f, err := os.Open(c.File)
if err != nil {
return err
}
defer f.Close()
st, err := f.Stat()
if err != nil {
return err
}
progress := progressbar.DefaultBytes(st.Size(), fmt.Sprintf("Uploading %s", filepath.Base(c.File)))
defer progress.Close()
tee := io.TeeReader(f, progress)
b := make([]byte, photosapi.MaxUploadPartSize)
part := 0
for {
n, err := tee.Read(b)
if err != nil {
if err == io.EOF {
break
} else {
return err
}
}
part++
partsha256 := sha256.New()
partsha256.Write(b[:n])
resp, err := cli.
R().
SetError(&UploadError{}).
SetResult(&UploadPartResult{}).
SetQueryParam("part", fmt.Sprint(part)).
SetQueryParam("sha256", hex.EncodeToString(partsha256.Sum(nil))).
SetBody(b[:n]).
SetPathParam("id", uploadId).
Put("/upload/{id}")
if err != nil {
return err
}
if err, ok := resp.Error().(*UploadError); ok {
return errors.New(err.Error)
}
}
fmt.Printf(
"Upload: %s\nParts: %d\n",
uploadId,
part,
)
result := resp.Result().(*UploadCreate)
fmt.Printf("Upload create: %s", result.UploadId)
return nil
}

View File

@ -145,12 +145,13 @@ func (s *Service) UploadComplete(c *gin.Context) {
func (s *Service) UploadInit() {
upload := s.Gin.Group("/upload")
// start
// upload.GET("/create", s.UploadCreate)
// upload.POST("/part/:upload_id", s.UploadPart)
// upload.GET("/cancel/:upload_id", s.UploadCancel)
// upload.POST("/complete/:upload_id", s.UploadComplete)
upload.POST("", s.UploadCreate)
// Cancel
upload.DELETE("/:upload_id", s.UploadCancel)
// Add part
upload.PUT("/:upload_id", s.UploadPart)
// Complete
upload.POST("/:upload_id", s.UploadComplete)
}