shylie/serra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add --type

Browse Source
This commit is contained in:
Florian Baumann 2023-02-20 09:40:02 +01:00
parent 8d4bd7adbb
commit 096b76a350
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/serra/card.go
View File

@ -14,6 +14,7 @@ func init() {
cardCmd.Flags().StringVarP(&sort, "sort", "s", "name", "How to sort cards (value/number/name/added)") cardCmd.Flags().StringVarP(&sort, "sort", "s", "name", "How to sort cards (value/number/name/added)")
cardCmd.Flags().StringVarP(&name, "name", "n", "", "Name of the card (regex compatible)") cardCmd.Flags().StringVarP(&name, "name", "n", "", "Name of the card (regex compatible)")
cardCmd.Flags().StringVarP(&oracle, "oracle", "o", "", "Contains string in card text") cardCmd.Flags().StringVarP(&oracle, "oracle", "o", "", "Contains string in card text")
cardCmd.Flags().StringVarP(&cardType, "type", "t", "", "Contains string in card type line")
rootCmd.AddCommand(cardCmd) rootCmd.AddCommand(cardCmd)
} }
@ -27,7 +28,7 @@ otherwise you'll get a list of cards as a search result.`,
SilenceErrors: true, SilenceErrors: true,
RunE: func(cmd *cobra.Command, cards []string) error { RunE: func(cmd *cobra.Command, cards []string) error {
if len(cards) == 0 { if len(cards) == 0 {
Cards(rarity, set, sort, name, oracle) Cards(rarity, set, sort, name, oracle, cardType)
} else { } else {
ShowCard(cards) ShowCard(cards)
} }
@ -51,7 +52,7 @@ func ShowCard(cardids []string) {
} }
} }
func Cards(rarity, set, sort, name, oracle string) { func Cards(rarity, set, sort, name, oracle, cardType string) {
var total float64 var total float64
client := storage_connect() client := storage_connect()
@ -99,6 +100,10 @@ func Cards(rarity, set, sort, name, oracle string) {
filter = append(filter, bson.E{"oracletext", bson.D{{"$regex", ".*" + oracle + ".*"}, {"$options", "i"}}}) filter = append(filter, bson.E{"oracletext", bson.D{{"$regex", ".*" + oracle + ".*"}, {"$options", "i"}}})
} }
if len(cardType) > 0 {
filter = append(filter, bson.E{"typeline", bson.D{{"$regex", ".*" + cardType + ".*"}, {"$options", "i"}}})
}
cards, _ := coll.storage_find(filter, sortStage) cards, _ := coll.storage_find(filter, sortStage)
for _, card := range cards { for _, card := range cards {

6
src/serra/root.go
View File

@ -8,14 +8,16 @@ import (
) )
var Version = "unknown" var Version = "unknown"
var count int64 var count int64
var limit float64 var limit float64
var name string var name string
var since string var oracle string
var rarity string var rarity string
var set string var set string
var since string
var sort string var sort string
var oracle string var cardType string
var unique bool var unique bool
var rootCmd = &cobra.Command{ var rootCmd = &cobra.Command{