Add geoserver backup and geoserver restore commands

Author: jericks

README.rst

@@ -94,6 +94,10 @@ Commands
 
     * geoserver reset
 
+    * geoserver backup --directory backup1 --includedata false --includegwc false --includelog true
+
+    * geoserver restore --directory backup1
+
 * about
 
     * version list

src/main/java/org/geoserver/shell/GeoserverCommands.java

@@ -8,6 +8,8 @@
 import org.springframework.shell.core.annotation.CliOption;
 import org.springframework.stereotype.Component;
 
+import java.io.File;
+
 @Component
 public class GeoserverCommands implements CommandMarker {
 
@@ -18,7 +20,7 @@ public void setGeoserver(Geoserver gs) {
         this.geoserver = gs;
     }
 
-    @CliAvailabilityIndicator({"geoserver reset", "geoserver reload"})
+    @CliAvailabilityIndicator({"geoserver reset", "geoserver reload", "geoserver backup"})
     public boolean isCommandAvailable() {
         return geoserver.isSet();
     }
@@ -56,4 +58,25 @@ public boolean reload() {
         GeoServerRESTPublisher publisher = new GeoServerRESTPublisher(geoserver.getUrl(), geoserver.getUser(), geoserver.getPassword());
         return publisher.reload();
     }
+
+    @CliCommand(value = "geoserver backup", help = "Backup Geoserver's configuration.")
+    public boolean backup(
+            @CliOption(key = "directory", mandatory = true, help = "The backup ") String backupDir,
+            @CliOption(key = "includedata", mandatory = false, unspecifiedDefaultValue = "false", help = "The include data flag") boolean includeData,
+            @CliOption(key = "includegwc", mandatory = false, unspecifiedDefaultValue = "false", help = "The include GWC flag") boolean includeGwc,
+            @CliOption(key = "includelog", mandatory = false, unspecifiedDefaultValue = "false", help = "The include log files flag") boolean includeLog
+    ) {
+        GeoServerRESTPublisher publisher = new GeoServerRESTPublisher(geoserver.getUrl(), geoserver.getUser(), geoserver.getPassword());
+        String result = publisher.backup(backupDir, includeData, includeGwc, includeLog);
+        return result != null;
+    }
+
+    @CliCommand(value = "geoserver restore", help = "Restore Geoserver's configuration from a backup directory.")
+    public boolean restore(
+            @CliOption(key = "directory", mandatory = true, help = "The backup ") String backupDir
+    ) {
+        GeoServerRESTPublisher publisher = new GeoServerRESTPublisher(geoserver.getUrl(), geoserver.getUser(), geoserver.getPassword());
+        String result = publisher.restore(backupDir);
+        return result != null;
+    }
 }

src/test/java/org/geoserver/shell/GeoserverCommandsTest.java

@@ -77,4 +77,35 @@ public void reload() throws Exception {
         assertTrue(result);
         verifyHttp(server).once(method(Method.POST), uri(url));
     }
+
+    @Test
+    public void backup() throws Exception {
+        String url = "/geoserver/rest/bkprst/backup";
+        whenHttp(server).match(post(url)).then(stringContent("true"), status(HttpStatus.OK_200));
+        Geoserver geoserver = new Geoserver("http://00.0.0.0:8888/geoserver", "admin", "geoserver");
+        GeoserverCommands commands = new GeoserverCommands();
+        commands.setGeoserver(geoserver);
+        boolean result = commands.backup("backupdir1", true, true, false);
+        assertTrue(result);
+        String expected = "<task><path>backupdir1</path><includedata>true</includedata><includegwc>true</includegwc><includelog>false</includelog></task>";
+        String actual = server.getCalls().get(0).getPostBody();
+        assertEquals(expected, actual);
+        verifyHttp(server).once(method(Method.POST), uri(url));
+    }
+
+    @Test
+    public void restore() throws Exception {
+        String url = "/geoserver/rest/bkprst/restore";
+        whenHttp(server).match(post(url)).then(stringContent("true"), status(HttpStatus.OK_200));
+        Geoserver geoserver = new Geoserver("http://00.0.0.0:8888/geoserver", "admin", "geoserver");
+        GeoserverCommands commands = new GeoserverCommands();
+        commands.setGeoserver(geoserver);
+        boolean result = commands.restore("backupdir1");
+        assertTrue(result);
+        String expected = "<task><path>backupdir1</path></task>";
+        String actual = server.getCalls().get(0).getPostBody();
+        assertEquals(expected, actual);
+        verifyHttp(server).once(method(Method.POST), uri(url));
+    }
+
 }