105 lines
2.4 KiB
Go
105 lines
2.4 KiB
Go
package output
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/michimani/gotwi"
|
|
"github.com/michimani/gotwi/media/upload"
|
|
uploadtypes "github.com/michimani/gotwi/media/upload/types"
|
|
"github.com/michimani/gotwi/tweet/managetweet"
|
|
"github.com/michimani/gotwi/tweet/managetweet/types"
|
|
)
|
|
|
|
type twitteroutput struct {
|
|
accessToken string
|
|
accessSecret string
|
|
client *gotwi.Client
|
|
}
|
|
|
|
func TwitterOutput(accessToken string, accessSecret string) *twitteroutput {
|
|
twitteroutput := &twitteroutput{
|
|
accessToken: accessToken,
|
|
accessSecret: accessSecret,
|
|
}
|
|
errOAuth := twitteroutput.newOAuth1Client(accessToken, accessSecret)
|
|
if errOAuth != nil {
|
|
fmt.Fprintln(os.Stderr, errOAuth)
|
|
os.Exit(1)
|
|
}
|
|
return twitteroutput
|
|
}
|
|
|
|
func (to *twitteroutput) newOAuth1Client(accessToken, accessSecret string) error {
|
|
in := &gotwi.NewClientInput{
|
|
AuthenticationMethod: gotwi.AuthenMethodOAuth1UserContext,
|
|
OAuthToken: accessToken,
|
|
OAuthTokenSecret: accessSecret,
|
|
}
|
|
|
|
client, err := gotwi.NewClient(in)
|
|
to.client = client
|
|
return err
|
|
}
|
|
|
|
func (to *twitteroutput) Write(str string, images []Image) error {
|
|
mediaIDs := make([]string, 0, len(images))
|
|
for _, img := range images {
|
|
mediaID, err := to.uploadImage(img)
|
|
if err != nil {
|
|
return fmt.Errorf("uploading %s: %w", img.Filename, err)
|
|
}
|
|
mediaIDs = append(mediaIDs, mediaID)
|
|
}
|
|
|
|
p := &types.CreateInput{
|
|
Text: gotwi.String(str),
|
|
}
|
|
if len(mediaIDs) > 0 {
|
|
p.Media = &types.CreateInputMedia{
|
|
MediaIDs: mediaIDs,
|
|
}
|
|
}
|
|
|
|
_, err := managetweet.Create(context.Background(), to.client, p)
|
|
return err
|
|
}
|
|
|
|
func (to *twitteroutput) uploadImage(img Image) (string, error) {
|
|
ctx := context.Background()
|
|
|
|
initialized, err := upload.Initialize(ctx, to.client, &uploadtypes.InitializeInput{
|
|
MediaCategory: uploadtypes.MediaCategoryTweetImage,
|
|
MediaType: uploadtypes.MediaType(img.ContentType),
|
|
TotalBytes: len(img.Data),
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
mediaID := initialized.Data.MediaID
|
|
|
|
_, err = upload.Append(ctx, to.client, &uploadtypes.AppendInput{
|
|
MediaID: mediaID,
|
|
Media: bytes.NewReader(img.Data),
|
|
SegmentIndex: 0,
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
_, err = upload.Finalize(ctx, to.client, &uploadtypes.FinalizeInput{
|
|
MediaID: mediaID,
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return mediaID, nil
|
|
}
|
|
|
|
func (to *twitteroutput) GetName() string {
|
|
return "twitter"
|
|
}
|