Merge pull request #13 from wcatz/duckx

feat(twitter posts): long live duckBot

Author: wcatz

go.mod

@@ -10,6 +10,7 @@ require (
 	github.com/cardano-community/koios-go-client/v3 v3.1.3
 	github.com/cenkalti/backoff/v4 v4.3.0
 	github.com/gorilla/websocket v1.5.3
+	github.com/michimani/gotwi v0.18.1
 	github.com/spf13/viper v1.21.0
 	gopkg.in/tucnak/telebot.v2 v2.5.0
 )

go.sum

@@ -120,6 +120,8 @@ github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
 github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
 github.com/leanovate/gopter v0.2.11 h1:vRjThO1EKPb/1NsDXuDrzldR28RLkBflWYcU9CvzWu4=
 github.com/leanovate/gopter v0.2.11/go.mod h1:aK3tzZP/C+p1m3SPRE4SYZFGP7jjkuSI4f7Xvpt0S9c=
+github.com/michimani/gotwi v0.18.1 h1:Tp7uia9qby8I0AXk9oDZRqaCPg31qyQ7NkeyiGDCDaE=
+github.com/michimani/gotwi v0.18.1/go.mod h1:yz1cyV/30Uy/KGQyN8BVfXFPt/63Imzonykny8/SMi0=
 github.com/minio/sha256-simd v1.0.1 h1:6kaan5IFmwTNynnKKpDHe6FWHohJOHhCPchzK49dzMM=
 github.com/minio/sha256-simd v1.0.1/go.mod h1:Pz6AKMiUdngCLpeTL/RJY1M9rUuPMYujV5xJjtbRSN8=
 github.com/nsf/jsondiff v0.0.0-20210926074059-1e845ec5d249 h1:NHrXEjTNQY7P0Zfx1aMrNhpgxHmow66XQtm0aQLY0AE=

main.go

@@ -6,6 +6,7 @@ import (
         "encoding/json"
         "errors"
         "fmt"
+        "io"
         "log"
         "net/http"
         "os"
@@ -22,6 +23,9 @@ import (
         koios "github.com/cardano-community/koios-go-client/v3"
         "github.com/cenkalti/backoff/v4"
         "github.com/gorilla/websocket"
+        "github.com/michimani/gotwi"
+        "github.com/michimani/gotwi/tweet/managetweet"
+        "github.com/michimani/gotwi/tweet/managetweet/types"
         "github.com/spf13/viper"
         telebot "gopkg.in/tucnak/telebot.v2"
 )
@@ -69,6 +73,9 @@ type Indexer struct {
         epoch           int
         networkMagic    int
         wg              sync.WaitGroup
+        // Twitter fields
+        twitterClient  *gotwi.Client
+        twitterEnabled bool
 }
 
 type BlockEvent struct {
@@ -138,6 +145,32 @@ func (i *Indexer) Start() error {
                 log.Fatalf("failed to start bot: %s", err)
         }
 
+        // Initialize Twitter client if credentials are provided
+        twitterAPIKey := os.Getenv("TWITTER_API_KEY")
+        twitterAPISecret := os.Getenv("TWITTER_API_KEY_SECRET")
+        twitterAccessToken := os.Getenv("TWITTER_ACCESS_TOKEN")
+        twitterAccessTokenSecret := os.Getenv("TWITTER_ACCESS_TOKEN_SECRET")
+
+        if twitterAPIKey != "" && twitterAPISecret != "" &&
+                twitterAccessToken != "" && twitterAccessTokenSecret != "" {
+                in := &gotwi.NewClientInput{
+                        AuthenticationMethod: gotwi.AuthenMethodOAuth1UserContext,
+                        OAuthToken:           twitterAccessToken,
+                        OAuthTokenSecret:     twitterAccessTokenSecret,
+                        APIKey:               twitterAPIKey,
+                        APIKeySecret:         twitterAPISecret,
+                }
+                i.twitterClient, err = gotwi.NewClient(in)
+                if err != nil {
+                        log.Printf("failed to initialize Twitter client: %s", err)
+                } else {
+                        i.twitterEnabled = true
+                        log.Println("Twitter client initialized successfully")
+                }
+        } else {
+                log.Println("Twitter credentials not provided, Twitter notifications disabled")
+        }
+
         /// Initialize the kois client based on networkMagic number
         if i.networkMagic == 1 {
                 i.koios, e = koios.New(
@@ -357,16 +390,37 @@ func (i *Indexer) handleEvent(event event.Event) error {
                         timeDiffString, i.epochBlocks, i.totalBlocks,
                         blockEvent.Context.BlockNumber, blockEvent.Payload.BlockHash)
 
+                // Get duck image URL for both Telegram and Twitter
+                duckImageURL := getDuckImage()
+
                 // Send the message to the appropriate channel
                 channelID, err := strconv.ParseInt(i.telegramChannel, 10, 64)
                 if err != nil {
                         log.Fatalf("failed to parse telegram channel ID: %s", err)
                 }
-                photo := &telebot.Photo{File: telebot.FromURL(getDuckImage()), Caption: msg}
+                photo := &telebot.Photo{File: telebot.FromURL(duckImageURL), Caption: msg}
                 _, err = i.bot.Send(&telebot.Chat{ID: channelID}, photo)
                 if err != nil {
                         log.Printf("failed to send Telegram message: %s", err)
                 }
+
+                // Send tweet if Twitter is enabled (shorter format for 280 char limit)
+                if i.twitterEnabled {
+                        tweetMsg := fmt.Sprintf(
+                                "🦆 New Block!\n\n"+
+                                        "Pool: %s\n"+
+                                        "Tx: %d | Size: %.2fKB\n"+
+                                        "Epoch: %d | Lifetime: %d\n\n"+
+                                        "pooltool.io/realtime/%d",
+                                i.poolName, blockEvent.Payload.TransactionCount, blockSizeKB,
+                                i.epochBlocks, i.totalBlocks,
+                                blockEvent.Context.BlockNumber)
+
+                        err = i.sendTweet(tweetMsg, duckImageURL)
+                        if err != nil {
+                                log.Printf("failed to send tweet: %s", err)
+                        }
+                }
         }
 
         // Print the received event information
@@ -432,6 +486,47 @@ func getDuckImage() string {
         return imageURL
 }
 
+// Download image from URL to bytes
+func downloadImage(imageURL string) ([]byte, error) {
+        resp, err := http.Get(imageURL)
+        if err != nil {
+                return nil, fmt.Errorf("failed to download image: %w", err)
+        }
+        defer resp.Body.Close()
+
+        if resp.StatusCode != http.StatusOK {
+                return nil, fmt.Errorf("failed to download image: status %d", resp.StatusCode)
+        }
+
+        imageData, err := io.ReadAll(resp.Body)
+        if err != nil {
+                return nil, fmt.Errorf("failed to read image data: %w", err)
+        }
+
+        return imageData, nil
+}
+
+// Send tweet with optional image
+func (i *Indexer) sendTweet(text string, imageURL string) error {
+        if !i.twitterEnabled || i.twitterClient == nil {
+                return nil
+        }
+
+        // Create tweet request
+        input := &types.CreateInput{
+                Text: gotwi.String(text),
+        }
+
+        // Post the tweet
+        _, err := managetweet.Create(context.Background(), i.twitterClient, input)
+        if err != nil {
+                return fmt.Errorf("failed to post tweet: %w", err)
+        }
+
+        log.Println("Tweet posted successfully")
+        return nil
+}
+
 // Convert to bech32 poolID
 func convertToBech32(hash string) (string, error) {
         bytes, err := hex.DecodeString(hash)