add filter and filterValue flags for more advanced and flexible filte…

…ring

README.md

@@ -13,6 +13,10 @@ Usage of ./allthefirmwares:
   -d string
     	the location to save/check IPSW files.
     		Can include templates e.g. {{.Identifier}} or {{.BuildID}} (default "./")
+  -filter string
+    	filter by a specific struct field
+  -filterValue string
+    	the value to filter by (used with -filter)
   -i string
     	only download for the specified device
   -l	only download the latest firmware for the specified devices

allthefirmwares.go

@@ -14,6 +14,7 @@ import (
 	"os"
 	"os/signal"
 	"path/filepath"
+	"reflect"
 	"sort"
 	"text/template"
 
@@ -25,6 +26,8 @@ import (
 var (
 	ipswClient = api.NewIPSWClient("https://api.ipsw.me/v4", nil)
 
+	filter, filterValue string
+
 	// flags
 	verifyIntegrity, reDownloadOnVerificationFailed, downloadSigned, downloadLatest bool
 	downloadDirectoryTemplate, specifiedDevice                                      string
@@ -41,6 +44,8 @@ func init() {
 	flag.BoolVar(&downloadSigned, "s", false, "only download signed firmwares")
 	flag.StringVar(&downloadDirectoryTemplate, "d", "./", "the location to save/check IPSW files.\n\tCan include templates e.g. {{.Identifier}} or {{.BuildID}}")
 	flag.StringVar(&specifiedDevice, "i", "", "only download for the specified device")
+	flag.StringVar(&filter, "filter", "", "filter by a specific struct field")
+	flag.StringVar(&filterValue, "filterValue", "", "the value to filter by (used with -filter)")
 	flag.Parse()
 }
 
@@ -91,6 +96,10 @@ func main() {
 				continue
 			}
 
+			if filter != "" && filterValue != "" && !passesFilter(ipsw, filter, filterValue) {
+				continue
+			}
+
 			directory, err := parseDownloadDirectory(&ipsw, device.Identifier)
 
 			if err != nil {
@@ -302,3 +311,32 @@ func download(url string, location string, writer io.Writer, callback func(n, do
 
 	return hex.EncodeToString(h.Sum(nil)), err
 }
+
+func passesFilter(firmware api.Firmware, filterName, filterValue string) bool {
+	field := reflect.Indirect(reflect.ValueOf(firmware)).FieldByName(filterName)
+
+	str := ""
+
+	switch t := field.Interface().(type) {
+	case uint, uint8, uint16, uint32, uint64, int, int8, int16, int32, int64:
+		str = fmt.Sprintf("%d", t)
+
+	case string:
+		str = t
+
+	case fmt.Stringer:
+		str = t.String()
+
+	case bool:
+		if t {
+			str = "true"
+		} else {
+			str = "false"
+		}
+
+	default:
+		return false
+	}
+
+	return filterValue == str
+}