natercioandnatercio
1ce8561782
Badges / coveralls (push) Successful in 1m4s
Co-authored-by: Natercio Moniz <[email protected]> Co-committed-by: Natercio Moniz <[email protected]>
105 lines
2.8 KiB
Go
105 lines
2.8 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")
|
|
debug = pflag.BoolP("debug", "d", false, "Activate to log debug messages")
|
|
format = pflag.StringP("format", "f", "table", "Output format: table or csv")
|
|
ofAPIKey = pflag.String("open-figi-api-key", "", "An OpenFIGI API key for faster report generation (better rate api rate limits)")
|
|
// TODO: improve documentation on selectors
|
|
selectors = pflag.StringSlice("selectors", nil, "Only process entries that conform to all the selectors: code, assetCountry")
|
|
)
|
|
|
|
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 {
|
|
ctx, cancel := signal.NotifyContext(ctx, os.Kill, os.Interrupt)
|
|
defer cancel()
|
|
|
|
eg, ctx := errgroup.WithContext(ctx)
|
|
|
|
logLevel := slog.LevelInfo
|
|
if *debug {
|
|
logLevel = slog.LevelDebug
|
|
}
|
|
slog.SetDefault(slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{Level: logLevel})))
|
|
|
|
if platform == nil || len(*platform) == 0 {
|
|
return fmt.Errorf("--platform flag is required")
|
|
}
|
|
|
|
if lang == nil || len(*lang) == 0 {
|
|
return fmt.Errorf("--language flag is required")
|
|
}
|
|
|
|
reader, err := getReader(*platform, *ofAPIKey)
|
|
if err != nil {
|
|
return fmt.Errorf("getting reader: %w", err)
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
switch *format {
|
|
case "csv":
|
|
return NewCSVWriter(os.Stdout).Render(writer)
|
|
case "table":
|
|
loc, err := NewLocalizer(*lang)
|
|
if err != nil {
|
|
return fmt.Errorf("create localizer: %w", err)
|
|
}
|
|
NewPrettyPrinter(os.Stdout, loc).Render(writer)
|
|
return nil
|
|
default:
|
|
return fmt.Errorf("unsupported format %q: must be table or csv", *format)
|
|
}
|
|
}
|
|
|
|
func getReader(platform string, ofAPIKey string) (internal.RecordReader, error) {
|
|
switch platform {
|
|
case "trading212":
|
|
return trading212.NewRecordReader(os.Stdin, internal.NewOpenFIGI(&http.Client{Timeout: 5 * time.Second}, ofAPIKey)), nil
|
|
default:
|
|
return nil, fmt.Errorf("unsupported platform: %s", platform)
|
|
}
|
|
}
|