pcgamedb/cmd/organize.go

48 lines
1.2 KiB
Go
Raw Normal View History

2024-09-24 06:17:11 -04:00
package cmd
import (
2024-11-15 02:02:45 -05:00
"github.com/nitezs/pcgamedb/crawler"
"github.com/nitezs/pcgamedb/db"
"github.com/nitezs/pcgamedb/log"
2024-09-24 06:17:11 -04:00
"github.com/spf13/cobra"
"go.uber.org/zap"
)
var organizeCmd = &cobra.Command{
Use: "organize",
Long: "Organize game info by repack game's name",
Short: "Organize game info by repack game's name",
Run: organizeRun,
}
type organizeCommandConfig struct {
Num int
}
var organizeCmdCfg organizeCommandConfig
func init() {
organizeCmd.Flags().IntVarP(&organizeCmdCfg.Num, "num", "n", -1, "number of items to process")
RootCmd.AddCommand(organizeCmd)
}
func organizeRun(cmd *cobra.Command, args []string) {
games, err := db.GetUnorganizedGameDownloads(organizeCmdCfg.Num)
if err != nil {
log.Logger.Error("Failed to get games", zap.Error(err))
}
for _, game := range games {
gameInfo, err := crawler.OrganizeGameDownload(game)
if err == nil {
err = db.SaveGameInfo(gameInfo)
if err != nil {
log.Logger.Error("Failed to save game info", zap.Error(err))
continue
}
log.Logger.Info("Organized game", zap.String("name", game.Name))
}
log.Logger.Error("Failed to organize game", zap.String("name", game.Name))
}
}