96 lines
2.1 KiB
Go
96 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"github.com/nmoniz/any2anexoj/internal"
|
|
"github.com/nmoniz/any2anexoj/internal/trading212"
|
|
"github.com/spf13/pflag"
|
|
"golang.org/x/sync/errgroup"
|
|
"golang.org/x/text/language"
|
|
)
|
|
|
|
var (
|
|
// TODO: once we support more brokers or exchanges we should make this parameter required and
|
|
// remove/change default
|
|
platform = pflag.StringP("platform", "p", "trading212", "One of the supported platforms")
|
|
lang = pflag.StringP("language", "l", language.Portuguese.String(), "The 2 letter language code")
|
|
// TODO: improve documentation on selectors
|
|
selectors = pflag.StringSlice("selectors", nil, "Only process entries that conform to all the selectors:")
|
|
|
|
readerFactories = map[string]func() internal.RecordReader{
|
|
"trading212": func() internal.RecordReader {
|
|
return trading212.NewRecordReader(os.Stdin, internal.NewOpenFIGI(&http.Client{Timeout: 5 * time.Second}))
|
|
},
|
|
}
|
|
)
|
|
|
|
func main() {
|
|
pflag.Parse()
|
|
|
|
err := run(context.Background())
|
|
if err != nil {
|
|
slog.Error("found a fatal issue", slog.Any("err", err))
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func run(ctx context.Context) error {
|
|
if platform == nil || len(*platform) == 0 {
|
|
slog.Error("--platform flag is required")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if lang == nil || len(*lang) == 0 {
|
|
slog.Error("--language flag is required")
|
|
os.Exit(1)
|
|
}
|
|
|
|
ctx, cancel := signal.NotifyContext(ctx, os.Kill, os.Interrupt)
|
|
defer cancel()
|
|
|
|
eg, ctx := errgroup.WithContext(ctx)
|
|
|
|
slog.SetDefault(slog.New(slog.NewTextHandler(os.Stderr, nil)))
|
|
|
|
factory, ok := readerFactories[*platform]
|
|
if !ok {
|
|
return fmt.Errorf("unsupported platform: %s", *platform)
|
|
}
|
|
|
|
reader := factory()
|
|
|
|
writer := internal.NewAggregatorWriter()
|
|
|
|
selector, err := internal.ParseSelectors(*selectors)
|
|
if err != nil {
|
|
return fmt.Errorf("parsing selectors: %w", err)
|
|
}
|
|
|
|
eg.Go(func() error {
|
|
return internal.BuildReport(ctx, reader, writer, selector)
|
|
})
|
|
|
|
err = eg.Wait()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
loc, err := NewLocalizer(*lang)
|
|
if err != nil {
|
|
return fmt.Errorf("create localizer: %w", err)
|
|
}
|
|
|
|
printer := NewPrettyPrinter(os.Stdout, loc)
|
|
|
|
printer.Render(writer)
|
|
|
|
return nil
|
|
}
|