add CommandModel.toBuilder

Author: h908714124

jbock/src/main/java/net/jbock/model/CommandModel.java

@@ -58,6 +58,24 @@ public static final class Builder {
         private Builder() {
         }
 
+        public Builder withOptions(List<Option> newOptions) {
+            options.clear();
+            options.addAll(newOptions);
+            return this;
+        }
+
+        public Builder withParameters(List<Parameter> newParameters) {
+            parameters.clear();
+            parameters.addAll(newParameters);
+            return this;
+        }
+
+        public Builder withDescriptionLines(List<String> newDescriptionLines) {
+            descriptionLines.clear();
+            descriptionLines.addAll(newDescriptionLines);
+            return this;
+        }
+
         /**
          * Sets the description key.
          *
@@ -140,6 +158,16 @@ public CommandModel build() {
         }
     }
 
+    public Builder toBuilder() {
+        return builder()
+                .withDescriptionKey(descriptionKey)
+                .withDescriptionLines(descriptionLines)
+                .withProgramName(programName)
+                .withSuperCommand(superCommand)
+                .withOptions(options)
+                .withParameters(parameters);
+    }
+
     /**
      * Returns the description key from the {@link Command#descriptionKey()} attribute,
      * possibly an empty string.

jbock/src/test/java/net/jbock/model/CommandModelTest.java

@@ -0,0 +1,23 @@
+package net.jbock.model;
+
+import org.junit.jupiter.api.Test;
+
+import java.util.List;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+class CommandModelTest {
+
+    @Test
+    void testToBuilder() {
+        Option option1 = Option.nullary().withParamLabel("v").withNames(List.of("-v")).build();
+        Option option2 = Option.nullary().withParamLabel("p").withNames(List.of("-p")).build();
+        CommandModel model1 = CommandModel.builder()
+                .addOption(option1)
+                .addOption(option2)
+                .build();
+        assertEquals(2, model1.options().size());
+        CommandModel model2 = model1.toBuilder().withOptions(List.of(option1)).build();
+        assertEquals(1, model2.options().size());
+    }
+}