Merge pull request #5 from trickest/output-param

Add output file parameter
trickest · Feb 7, 2022 · 7321ed7 · 7321ed7
2 parents 4a34d2e + ddeb40a
commit 7321ed7
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/main.go b/main.go
@@ -143,7 +143,8 @@ func getRepos(query string) {
 
 func main() {
 	token := flag.String("token", "", "Github token")
-	query := flag.String("query", "", "Query")
+	query := flag.String("query", "", "GraphQL search query")
+	outputFile := flag.String("o", "", "Output file name")
 	flag.Parse()
 
 	src := oauth2.StaticTokenSource(
@@ -189,10 +190,13 @@ func main() {
 
 		data, _ := json.MarshalIndent(reposResults, "", "   ")
 
-		fileName := strings.Trim(*query, "-*")
-		for _, char := range illegalFileNameChars {
-			if strings.Contains(fileName, char) {
-				fileName = strings.ReplaceAll(fileName, char, "")
+		fileName := *outputFile
+		if fileName == "" {
+			fileName = strings.Trim(*query, "-*")
+			for _, char := range illegalFileNameChars {
+				if strings.Contains(fileName, char) {
+					fileName = strings.ReplaceAll(fileName, char, "")
+				}
 			}
 		}