package main

import (
	"github.com/banbox/banbot/entry"
	"github.com/banbox/banbot/opt"
	_ "github.com/banbox/banstagy/adv"
	_ "github.com/banbox/banstagy/arb"
	_ "github.com/banbox/banstagy/code"
	_ "github.com/banbox/banstagy/demo"
	_ "github.com/banbox/banstagy/demo/freqtrade"
	"github.com/banbox/banstagy/ent"
	_ "github.com/banbox/banstagy/fac"
	_ "github.com/banbox/banstagy/filters"
	_ "github.com/banbox/banstagy/flux/grid"
	_ "github.com/banbox/banstagy/llmjob"
	_ "github.com/banbox/banstagy/old"
	_ "github.com/banbox/banstagy/org/day"
	_ "github.com/banbox/banstagy/org/digo"
	_ "github.com/banbox/banstagy/org/fmz"
	_ "github.com/banbox/banstagy/org/ft"
	_ "github.com/banbox/banstagy/org/jesse"
	_ "github.com/banbox/banstagy/org/llm"
	_ "github.com/banbox/banstagy/org/sf"
	_ "github.com/banbox/banstagy/ptn/all"
	_ "github.com/banbox/banstagy/rpc_fea"
	_ "github.com/banbox/banstagy/surge"
	_ "github.com/banbox/banstagy/tmp"
	_ "github.com/banbox/banstagy/trend"
	_ "github.com/banbox/banstagy/trend/20250208"
	"os"
)

func main() {
	RegPairPickers()
	args := os.Args[1:]
	if len(args) >= 2 && args[0] == "my" {
		err := ent.RunMyEntry(args[1:])
		if err != nil {
			panic(err)
		}
		return
	}
	entry.RunCmd()
}

func RegPairPickers() {
	opt.PairPickers["profit"] = func(r *opt.BTResult) []string {
		var pairs = make([]string, 0, len(r.PairGrps))
		for _, p := range r.PairGrps {
			if p.ProfitSum > 0 {
				pairs = append(pairs, p.Title)
			}
		}
		return pairs
	}
	opt.PairPickers["sharpe1"] = func(r *opt.BTResult) []string {
		var pairs = make([]string, 0, len(r.PairGrps))
		for _, p := range r.PairGrps {
			if p.Sharpe > 1 {
				pairs = append(pairs, p.Title)
			}
		}
		return pairs
	}
}