130 lines
3.0 KiB
Go
130 lines
3.0 KiB
Go
package groupmeext
|
|
|
|
import (
|
|
"bytes"
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
|
|
"gitea.watsonlabs.net/watsonb8/groupme-lib"
|
|
)
|
|
|
|
type Message struct{ groupme.Message }
|
|
|
|
func (m *Message) Scan(value interface{}) error {
|
|
bytes, ok := value.(string)
|
|
if !ok {
|
|
return errors.New(fmt.Sprint("Failed to unmarshal json value:", value))
|
|
}
|
|
|
|
message := Message{}
|
|
err := json.Unmarshal([]byte(bytes), &message)
|
|
|
|
*m = Message(message)
|
|
return err
|
|
}
|
|
|
|
func (m *Message) Value() (driver.Value, error) {
|
|
e, err := json.Marshal(m)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return e, nil
|
|
}
|
|
|
|
// DownloadImage helper function to download image from groupme;
|
|
// append .large/.preview/.avatar to get various sizes
|
|
func DownloadImage(URL string) (bytes *[]byte, mime string, err error) {
|
|
//TODO check its actually groupme?
|
|
response, err := http.Get(URL)
|
|
if err != nil {
|
|
return nil, "", errors.New("Failed to download avatar: " + err.Error())
|
|
}
|
|
defer response.Body.Close()
|
|
|
|
image, err := ioutil.ReadAll(response.Body)
|
|
bytes = &image
|
|
if err != nil {
|
|
return nil, "", errors.New("Failed to read downloaded image:" + err.Error())
|
|
}
|
|
|
|
mime = response.Header.Get("Content-Type")
|
|
if len(mime) == 0 {
|
|
mime = http.DetectContentType(image)
|
|
}
|
|
return
|
|
}
|
|
|
|
func DownloadFile(RoomJID groupme.ID, FileID string, token string) (contents []byte, fname, mime string) {
|
|
client := &http.Client{}
|
|
b, _ := json.Marshal(struct {
|
|
FileIDS []string `json:"file_ids"`
|
|
}{
|
|
FileIDS: []string{FileID},
|
|
})
|
|
|
|
req, _ := http.NewRequest("POST", fmt.Sprintf("https://file.groupme.com/v1/%s/fileData", RoomJID), bytes.NewReader(b))
|
|
req.Header.Add("X-Access-Token", token)
|
|
req.Header.Add("Content-Type", "application/json")
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
// TODO: FIX
|
|
panic(err)
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
data := []ImgData{}
|
|
json.NewDecoder(resp.Body).Decode(&data)
|
|
fmt.Println(data, RoomJID, FileID, token)
|
|
if len(data) < 1 {
|
|
return
|
|
}
|
|
|
|
req, _ = http.NewRequest("POST", fmt.Sprintf("https://file.groupme.com/v1/%s/files/%s", RoomJID, FileID), nil)
|
|
req.URL.Query().Add("token", token)
|
|
req.Header.Add("X-Access-Token", token)
|
|
resp, err = client.Do(req)
|
|
if err != nil {
|
|
// TODO: FIX
|
|
panic(err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
bytes, _ := ioutil.ReadAll(resp.Body)
|
|
return bytes, data[0].FileData.FileName, data[0].FileData.Mime
|
|
|
|
}
|
|
|
|
func DownloadVideo(previewURL, videoURL, token string) (vidContents []byte, mime string) {
|
|
//preview TODO
|
|
client := &http.Client{}
|
|
|
|
req, _ := http.NewRequest("GET", videoURL, nil)
|
|
req.AddCookie(&http.Cookie{Name: "token", Value: token})
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return nil, ""
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
bytes, _ := ioutil.ReadAll(resp.Body)
|
|
mime = resp.Header.Get("Content-Type")
|
|
if len(mime) == 0 {
|
|
mime = http.DetectContentType(bytes)
|
|
}
|
|
return bytes, mime
|
|
|
|
}
|
|
|
|
type ImgData struct {
|
|
FileData struct {
|
|
FileName string `json:"file_name"`
|
|
FileSize int `json:"file_size"`
|
|
Mime string `json:"mime_type"`
|
|
} `json:"file_data"`
|
|
}
|