fix(update-ref): permit source to be passed via environment (#57)

nexthink-oss · Oct 7, 2024 · b97d0ec · b97d0ec
1 parent 2d92186
commit b97d0ec
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cmd/updateref.go b/cmd/updateref.go
@@ -52,7 +52,6 @@ var updateRefCmd = &cobra.Command{
 
 func init() {
 	updateRefCmd.Flags().StringP("source", "s", "", "source `ref-or-commit`")
-	updateRefCmd.MarkFlagRequired("source")
 	viper.BindPFlag("source", updateRefCmd.Flags().Lookup("source"))
 
 	viper.BindEnv("targets", "GHUP_TARGETS")
@@ -83,6 +82,10 @@ func runUpdateRefCmd(cmd *cobra.Command, args []string) (err error) {
 	}
 
 	sourceRefName := viper.GetString("source")
+	if sourceRefName == "" {
+		return errors.New("no source ref specified")
+	}
+
 	var sourceObject string
 
 	if util.IsCommitHash(sourceRefName) {