1bf130b25a
Sync CLI module imports across all command packages. Refactor daemon command with expanded functionality. Update go.mod and go.work dependencies. Co-Authored-By: Virgil <virgil@lethean.io>
64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package collect
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"forge.lthn.ai/core/go/pkg/cli"
|
|
"forge.lthn.ai/core/go-scm/collect"
|
|
"forge.lthn.ai/core/go/pkg/i18n"
|
|
)
|
|
|
|
// BitcoinTalk command flags
|
|
var bitcointalkPages int
|
|
|
|
// addBitcoinTalkCommand adds the 'bitcointalk' subcommand to the collect parent.
|
|
func addBitcoinTalkCommand(parent *cli.Command) {
|
|
btcCmd := &cli.Command{
|
|
Use: "bitcointalk <topic-id|url>",
|
|
Short: i18n.T("cmd.collect.bitcointalk.short"),
|
|
Long: i18n.T("cmd.collect.bitcointalk.long"),
|
|
Args: cli.ExactArgs(1),
|
|
RunE: func(cmd *cli.Command, args []string) error {
|
|
return runBitcoinTalk(args[0])
|
|
},
|
|
}
|
|
|
|
cli.IntFlag(btcCmd, &bitcointalkPages, "pages", "p", 0, i18n.T("cmd.collect.bitcointalk.flag.pages"))
|
|
|
|
parent.AddCommand(btcCmd)
|
|
}
|
|
|
|
func runBitcoinTalk(target string) error {
|
|
var topicID, url string
|
|
|
|
// Determine if argument is a URL or topic ID
|
|
if strings.HasPrefix(target, "http") {
|
|
url = target
|
|
} else {
|
|
topicID = target
|
|
}
|
|
|
|
cfg := newConfig()
|
|
setupVerboseLogging(cfg)
|
|
|
|
collector := &collect.BitcoinTalkCollector{
|
|
TopicID: topicID,
|
|
URL: url,
|
|
Pages: bitcointalkPages,
|
|
}
|
|
|
|
if cfg.DryRun {
|
|
cli.Info("Dry run: would collect from BitcoinTalk topic " + target)
|
|
return nil
|
|
}
|
|
|
|
ctx := context.Background()
|
|
result, err := collector.Collect(ctx, cfg)
|
|
if err != nil {
|
|
return cli.Wrap(err, "bitcointalk collection failed")
|
|
}
|
|
|
|
printResult(result)
|
|
return nil
|
|
}
|