mirror of
https://github.com/usememos/memos.git
synced 2024-11-17 04:07:30 +08:00
45 lines
1 KiB
Go
45 lines
1 KiB
Go
|
package telegram
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"fmt"
|
||
|
"io"
|
||
|
"net/http"
|
||
|
)
|
||
|
|
||
|
// downloadFileId download file with fileID, return the filepath and blob.
|
||
|
func (r *Robot) downloadFileID(ctx context.Context, fileID string) (string, []byte, error) {
|
||
|
file, err := r.GetFile(ctx, fileID)
|
||
|
if err != nil {
|
||
|
return "", nil, err
|
||
|
}
|
||
|
blob, err := r.downloadFilepath(ctx, file.FilePath)
|
||
|
if err != nil {
|
||
|
return "", nil, err
|
||
|
}
|
||
|
|
||
|
return file.FilePath, blob, nil
|
||
|
}
|
||
|
|
||
|
// downloadFilepath download file with filepath, you can get filepath by calling GetFile.
|
||
|
func (r *Robot) downloadFilepath(ctx context.Context, filePath string) ([]byte, error) {
|
||
|
token := r.handler.RobotToken(ctx)
|
||
|
if token == "" {
|
||
|
return nil, ErrNoToken
|
||
|
}
|
||
|
|
||
|
uri := "https://api.telegram.org/file/bot" + token + "/" + filePath
|
||
|
resp, err := http.Get(uri)
|
||
|
if err != nil {
|
||
|
return nil, fmt.Errorf("fail to http.Get: %s", err)
|
||
|
}
|
||
|
defer resp.Body.Close()
|
||
|
|
||
|
body, err := io.ReadAll(resp.Body)
|
||
|
if err != nil {
|
||
|
return nil, fmt.Errorf("fail to io.ReadAll: %s", err)
|
||
|
}
|
||
|
|
||
|
return body, nil
|
||
|
}
|