132 lines
2.2 KiB
Go
132 lines
2.2 KiB
Go
package tg
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"io"
|
|
//"os"
|
|
//"path/filepath"
|
|
|
|
"github.com/go-telegram-bot-api/telegram-bot-api/v5"
|
|
)
|
|
|
|
type FileID string
|
|
type FileConfig = tgbotapi.FileConfig
|
|
type PhotoConfig = tgbotapi.PhotoConfig
|
|
type FileType int
|
|
|
|
const (
|
|
NoFileType FileType = iota
|
|
PhotoFileType
|
|
DocumentFileType
|
|
)
|
|
|
|
var (
|
|
UnknownFileTypeErr = errors.New("unknown file type")
|
|
)
|
|
|
|
// The type implements the structure to easily send
|
|
// files to the client.
|
|
type File struct {
|
|
MessageCompo
|
|
name string
|
|
reader io.Reader
|
|
upload bool
|
|
typ FileType
|
|
data, caption string
|
|
}
|
|
|
|
// Create the new file with the specified reader.
|
|
// By default it NeedsUpload is set to true.
|
|
func NewFile(reader io.Reader) *File {
|
|
ret := &File{}
|
|
|
|
ret.MessageCompo = *Messagef("")
|
|
ret.reader = reader
|
|
ret.upload = true
|
|
|
|
return ret
|
|
}
|
|
|
|
func (f *File) Name(name string) *File {
|
|
f.name = name
|
|
return f
|
|
}
|
|
|
|
func (f *File) withType(typ FileType) *File {
|
|
f.typ = typ
|
|
return f
|
|
}
|
|
|
|
// Get the file type.
|
|
func (f *File) Type() FileType {
|
|
return f.typ
|
|
}
|
|
|
|
// Set the file type to PhotoFileType.
|
|
func (f *File) Photo() *File {
|
|
return f.withType(PhotoFileType)
|
|
}
|
|
|
|
func (f *File) Document() *File {
|
|
return f.withType(DocumentFileType)
|
|
}
|
|
|
|
// Set the file caption.
|
|
func (f *File) Caption(caption string) *File {
|
|
f.caption = caption
|
|
return f
|
|
}
|
|
|
|
// Specifiy whether the file needs to be uploaded to Telegram.
|
|
func (f *File) Upload(upload bool) *File {
|
|
f.upload = upload
|
|
return f
|
|
}
|
|
|
|
// Set the data to return via SendData()
|
|
func (f *File) Data(data string) *File {
|
|
f.data = data
|
|
return f
|
|
}
|
|
|
|
func (f *File) NeedsUpload() bool {
|
|
return f.upload
|
|
}
|
|
|
|
func (f *File) UploadData() (string, io.Reader, error) {
|
|
// Bufferizing the reader
|
|
// to make it faster.
|
|
bufRd := bufio.NewReader(f.reader)
|
|
fileName := f.name
|
|
|
|
return fileName, bufRd, nil
|
|
}
|
|
|
|
func (f *File) SendData() string {
|
|
return f.data
|
|
}
|
|
|
|
func (f *File) SendConfig(
|
|
sid SessionID, bot *Bot,
|
|
) (SendConfig) {
|
|
var config SendConfig
|
|
cid := sid.ToAPI()
|
|
|
|
switch f.Type() {
|
|
case PhotoFileType:
|
|
photo := tgbotapi.NewPhoto(cid, f)
|
|
photo.Caption = f.caption
|
|
|
|
config.Chattable = photo
|
|
case DocumentFileType:
|
|
doc := tgbotapi.NewDocument(sid.ToAPI(), f)
|
|
doc.Caption = f.caption
|
|
config.Chattable = doc
|
|
default:
|
|
panic(UnknownFileTypeErr)
|
|
}
|
|
|
|
|
|
return config
|
|
}
|