Fix: get hq command

internetarchive · Jul 22, 2024 · f5c4625 · f5c4625
1 parent ed7728d
commit f5c4625
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/cmd/get.go b/cmd/get.go
@@ -16,6 +16,7 @@ func getCMDs() *cobra.Command {
 	}
 
 	getCMDsFlags(getCmd)
+	getHQCmdFlags(getHQCmd)
 
 	getCmd.AddCommand(getURLCmd)
 	getCmd.AddCommand(getHQCmd)

diff --git a/cmd/get_hq.go b/cmd/get_hq.go
@@ -54,4 +54,9 @@ func getHQCmdFlags(getHQCmd *cobra.Command) {
 	getHQCmd.PersistentFlags().Bool("hq-continuous-pull", false, "If turned on, the crawler will pull URLs from Crawl HQ continuously.")
 	getHQCmd.PersistentFlags().String("hq-strategy", "lifo", "Crawl HQ feeding strategy.")
 	getHQCmd.PersistentFlags().Bool("hq-rate-limiting-send-back", false, "If turned on, the crawler will send back URLs that hit a rate limit to crawl HQ.")
+
+	getHQCmd.MarkPersistentFlagRequired("hq-address")
+	getHQCmd.MarkPersistentFlagRequired("hq-key")
+	getHQCmd.MarkPersistentFlagRequired("hq-secret")
+	getHQCmd.MarkPersistentFlagRequired("hq-project")
 }