2019-03-30 04:31:31 +04:00
|
|
|
// Yandex Disk File Pusher
|
|
|
|
// Copyright (c) 2019 Vladimir "fat0troll" Hodakov
|
|
|
|
|
|
|
|
package yandexv1
|
|
|
|
|
2019-03-30 07:00:15 +04:00
|
|
|
import (
|
|
|
|
"github.com/schollz/progressbar/v2"
|
|
|
|
"io"
|
|
|
|
)
|
|
|
|
|
|
|
|
type progressReader struct {
|
|
|
|
r io.Reader
|
|
|
|
atEOF bool
|
|
|
|
progressbar *progressbar.ProgressBar
|
|
|
|
}
|
|
|
|
|
|
|
|
func (pr *progressReader) Read(p []byte) (int, error) {
|
|
|
|
n, err := pr.r.Read(p)
|
|
|
|
if err == io.EOF {
|
|
|
|
pr.atEOF = true
|
|
|
|
}
|
|
|
|
pr.report(int64(n))
|
|
|
|
return n, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (pr *progressReader) report(n int64) {
|
|
|
|
_ = pr.progressbar.Add64(n)
|
|
|
|
}
|
|
|
|
|
2019-03-30 04:31:31 +04:00
|
|
|
type TokenError struct {
|
|
|
|
Error string `json:"error"`
|
|
|
|
ErrorDescription string `json:"error_description"`
|
|
|
|
}
|
2019-03-30 07:00:15 +04:00
|
|
|
|
|
|
|
type UploadError struct {
|
|
|
|
Error string `json:"error"`
|
|
|
|
Description string `json:"description"`
|
|
|
|
}
|
|
|
|
|
|
|
|
type UploadInfo struct {
|
|
|
|
URL string `json:"href"`
|
|
|
|
Method string `json:"method"`
|
|
|
|
URLIsTemplated bool `json:"templated"`
|
|
|
|
}
|