Leon Mika
5d1f4c78f4
Started working on a new tool to poll and browse an SQS queue. This is built using a TUI framework
61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"github.com/aws/aws-sdk-go-v2/config"
|
|
"github.com/aws/aws-sdk-go-v2/service/sqs"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/lmika/awstools/internal/sqs-browse/models"
|
|
"github.com/lmika/awstools/internal/sqs-browse/providers/memstore"
|
|
sqsprovider "github.com/lmika/awstools/internal/sqs-browse/providers/sqs"
|
|
"github.com/lmika/awstools/internal/sqs-browse/services/pollmessage"
|
|
"github.com/lmika/awstools/internal/sqs-browse/ui"
|
|
"github.com/lmika/events"
|
|
"github.com/lmika/gopkgs/cli"
|
|
"log"
|
|
"os"
|
|
)
|
|
|
|
func main() {
|
|
var flagQueue = flag.String("q", "", "queue to poll")
|
|
flag.Parse()
|
|
|
|
ctx := context.Background()
|
|
cfg, err := config.LoadDefaultConfig(ctx)
|
|
if err != nil {
|
|
cli.Fatalf("cannot load AWS config: %v", err)
|
|
}
|
|
sqsClient := sqs.NewFromConfig(cfg)
|
|
|
|
bus := events.New()
|
|
|
|
msgStore := memstore.NewStore()
|
|
msgPoller := sqsprovider.NewProvider(sqsClient)
|
|
|
|
pollService := pollmessage.NewService(msgStore, msgPoller, *flagQueue, bus)
|
|
|
|
uiModel := ui.NewModel()
|
|
p := tea.NewProgram(uiModel, tea.WithAltScreen())
|
|
|
|
bus.On("new-messages", func(m []*models.Message) { p.Send(ui.NewMessagesEvent(m)) })
|
|
|
|
f, err := tea.LogToFile("debug.log", "debug")
|
|
if err != nil {
|
|
fmt.Println("fatal:", err)
|
|
os.Exit(1)
|
|
}
|
|
defer f.Close()
|
|
|
|
go func() {
|
|
if err := pollService.Poll(context.Background()); err != nil {
|
|
log.Printf("cannot start poller: %v", err)
|
|
}
|
|
}()
|
|
|
|
if err := p.Start(); err != nil {
|
|
fmt.Printf("Alas, there's been an error: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
} |