gmessages/libgm/media_processor.go

159 lines
3.8 KiB
Go
Raw Normal View History

2023-06-30 11:05:33 +00:00
package libgm
2023-06-30 09:54:08 +00:00
import (
"bytes"
"errors"
"io"
"net/http"
"strconv"
"go.mau.fi/mautrix-gmessages/libgm/binary"
"go.mau.fi/mautrix-gmessages/libgm/crypto"
"go.mau.fi/mautrix-gmessages/libgm/util"
)
type StartGoogleUpload struct {
2023-06-30 13:26:46 +00:00
UploadID string
UploadURL string
2023-06-30 09:54:08 +00:00
UploadStatus string
ChunkGranularity int64
2023-06-30 13:26:46 +00:00
ControlURL string
2023-06-30 09:54:08 +00:00
Image *Image
EncryptedMediaBytes []byte
}
type MediaUpload struct {
2023-06-30 13:26:46 +00:00
MediaID string
2023-06-30 09:54:08 +00:00
MediaNumber int64
Image *Image
}
var (
errStartUploadMedia = errors.New("failed to start uploading media")
errFinalizeUploadMedia = errors.New("failed to finalize uploading media")
)
func (c *Client) FinalizeUploadMedia(upload *StartGoogleUpload) (*MediaUpload, error) {
imageType := upload.Image.GetImageType()
encryptedImageSize := strconv.Itoa(len(upload.EncryptedMediaBytes))
finalizeUploadHeaders := util.NewMediaUploadHeaders(encryptedImageSize, "upload, finalize", "0", imageType.Format, "")
2023-06-30 13:26:46 +00:00
req, reqErr := http.NewRequest("POST", upload.UploadURL, bytes.NewBuffer(upload.EncryptedMediaBytes))
2023-06-30 09:54:08 +00:00
if reqErr != nil {
return nil, reqErr
}
req.Header = *finalizeUploadHeaders
res, resErr := c.http.Do(req)
if resErr != nil {
2023-06-30 11:48:50 +00:00
panic(resErr)
2023-06-30 09:54:08 +00:00
}
statusCode := res.StatusCode
if statusCode != 200 {
return nil, errFinalizeUploadMedia
}
defer res.Body.Close()
rHeaders := res.Header
googleResponse, err3 := io.ReadAll(res.Body)
if err3 != nil {
return nil, err3
}
uploadStatus := rHeaders.Get("x-goog-upload-status")
2023-06-30 11:48:50 +00:00
c.Logger.Debug().Str("upload_status", uploadStatus).Msg("Upload status")
2023-06-30 09:54:08 +00:00
2023-06-30 13:26:46 +00:00
mediaIDs := &binary.UploadMediaResponse{}
err3 = crypto.DecodeAndEncodeB64(string(googleResponse), mediaIDs)
2023-06-30 09:54:08 +00:00
if err3 != nil {
return nil, err3
}
return &MediaUpload{
2023-06-30 13:26:46 +00:00
MediaID: mediaIDs.Media.MediaID,
MediaNumber: mediaIDs.Media.MediaNumber,
2023-06-30 09:54:08 +00:00
Image: upload.Image,
}, nil
}
func (c *Client) StartUploadMedia(image *Image) (*StartGoogleUpload, error) {
imageType := image.GetImageType()
encryptedImageBytes, encryptErr := image.GetEncryptedBytes()
if encryptErr != nil {
return nil, encryptErr
}
encryptedImageSize := strconv.Itoa(len(encryptedImageBytes))
startUploadHeaders := util.NewMediaUploadHeaders(encryptedImageSize, "start", "", imageType.Format, "resumable")
startUploadPayload, buildPayloadErr := c.buildStartUploadPayload()
if buildPayloadErr != nil {
return nil, buildPayloadErr
}
req, reqErr := http.NewRequest("POST", util.UPLOAD_MEDIA, bytes.NewBuffer([]byte(startUploadPayload)))
if reqErr != nil {
return nil, reqErr
}
req.Header = *startUploadHeaders
res, resErr := c.http.Do(req)
if resErr != nil {
2023-06-30 11:48:50 +00:00
panic(resErr)
2023-06-30 09:54:08 +00:00
}
statusCode := res.StatusCode
if statusCode != 200 {
return nil, errStartUploadMedia
}
rHeaders := res.Header
chunkGranularity, convertErr := strconv.Atoi(rHeaders.Get("x-goog-upload-chunk-granularity"))
if convertErr != nil {
return nil, convertErr
}
uploadResponse := &StartGoogleUpload{
2023-06-30 13:26:46 +00:00
UploadID: rHeaders.Get("x-guploader-uploadid"),
UploadURL: rHeaders.Get("x-goog-upload-url"),
2023-06-30 09:54:08 +00:00
UploadStatus: rHeaders.Get("x-goog-upload-status"),
ChunkGranularity: int64(chunkGranularity),
2023-06-30 13:26:46 +00:00
ControlURL: rHeaders.Get("x-goog-upload-control-url"),
2023-06-30 09:54:08 +00:00
Image: image,
EncryptedMediaBytes: encryptedImageBytes,
}
return uploadResponse, nil
}
func (c *Client) buildStartUploadPayload() (string, error) {
requestId := util.RandomUUIDv4()
protoData := &binary.StartMediaUploadPayload{
ImageType: 1,
AuthData: &binary.AuthMessage{
2023-06-30 13:26:46 +00:00
RequestID: requestId,
RpcKey: c.rpcKey,
2023-06-30 09:54:08 +00:00
Date: &binary.Date{
Year: 2023,
Seq1: 6,
2023-07-02 14:19:00 +00:00
Seq2: 22,
2023-06-30 09:54:08 +00:00
Seq3: 4,
Seq4: 6,
},
},
Mobile: c.devicePair.Mobile,
}
protoDataEncoded, protoEncodeErr := crypto.EncodeProtoB64(protoData)
if protoEncodeErr != nil {
return "", protoEncodeErr
}
return protoDataEncoded, nil
}