refactor: rewriteSchema

Author: miparnisari

internal/cmd/schema.go

@@ -319,23 +319,15 @@ func schemaWriteCmdImpl(cmd *cobra.Command, args []string, client v1.SchemaServi
 
 // rewriteSchema rewrites the given existing schema to include the specified prefix on all definitions and caveats.
 func rewriteSchema(ctx context.Context, existingSchemaText string, definitionPrefix string, sourceFolder string) (string, error) {
-	if definitionPrefix == "" && sourceFolder == "" {
+	if definitionPrefix == "" {
 		return existingSchemaText, nil
 	}
 
-	var prefixOpt compiler.ObjectPrefixOption
-	if definitionPrefix != "" {
-		prefixOpt = compiler.ObjectTypePrefix(definitionPrefix)
-	} else {
-		prefixOpt = compiler.AllowUnprefixedObjectType()
-	}
-
-	opts := []compiler.Option{compiler.SkipValidation(), compiler.SourceFolder(sourceFolder)}
-
 	compiled, err := compiler.Compile(
 		compiler.InputSchema{Source: input.Source("schema"), SchemaString: existingSchemaText},
-		prefixOpt,
-		opts...,
+		compiler.ObjectTypePrefix(definitionPrefix),
+		compiler.SkipValidation(),
+		compiler.SourceFolder(sourceFolder),
 	)
 	if err != nil {
 		return "", err