Add plugin's InitStep to configure plugin during site init
Change-Id: I1db29e23faebb589271c554b71871fd5fd8c2478
diff --git a/BUCK b/BUCK
index 8232c2b..a4e7846 100644
--- a/BUCK
+++ b/BUCK
@@ -20,6 +20,7 @@
'Gerrit-ApiType: plugin',
'Gerrit-Module: com.ericsson.gerrit.plugins.highavailability.Module',
'Gerrit-HttpModule: com.ericsson.gerrit.plugins.highavailability.HttpModule',
+ 'Gerrit-InitStep: com.ericsson.gerrit.plugins.highavailability.Setup',
'Implementation-Title: high-availability plugin',
'Implementation-URL: https://gerrit-review.googlesource.com/#/admin/projects/plugins/high-availability',
'Implementation-Vendor: Ericsson',
diff --git a/src/main/java/com/ericsson/gerrit/plugins/highavailability/Setup.java b/src/main/java/com/ericsson/gerrit/plugins/highavailability/Setup.java
new file mode 100644
index 0000000..394f77a
--- /dev/null
+++ b/src/main/java/com/ericsson/gerrit/plugins/highavailability/Setup.java
@@ -0,0 +1,97 @@
+// Copyright (C) 2017 The Android Open Source Project
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package com.ericsson.gerrit.plugins.highavailability;
+
+import static com.ericsson.gerrit.plugins.highavailability.Configuration.*;
+
+import com.google.gerrit.common.FileUtil;
+import com.google.gerrit.extensions.annotations.PluginName;
+import com.google.gerrit.pgm.init.api.ConsoleUI;
+import com.google.gerrit.pgm.init.api.InitStep;
+import com.google.gerrit.pgm.init.api.Section;
+import com.google.gerrit.server.config.SitePaths;
+import com.google.inject.Inject;
+import java.nio.file.Path;
+
+public class Setup implements InitStep {
+
+ private final ConsoleUI ui;
+ private final Section mySection;
+ private final String pluginName;
+ private final SitePaths site;
+
+ @Inject
+ public Setup(
+ ConsoleUI ui, Section.Factory sections, @PluginName String pluginName, SitePaths site) {
+ this.ui = ui;
+ this.mySection = sections.get("plugin", pluginName);
+ this.pluginName = pluginName;
+ this.site = site;
+ }
+
+ @Override
+ public void run() throws Exception {
+ ui.message("\n");
+ ui.header("%s Plugin", pluginName);
+
+ if (ui.yesno(true, "Configure %s", pluginName)) {
+ ui.header("Configuring %s", pluginName);
+ configureSharedDir();
+ configurePeer();
+ configureTimeouts();
+ configureRetry();
+ configureThreadPools();
+ }
+ }
+
+ private void configureSharedDir() {
+ String sharedDir = mySection.string("Shared directory", SHARED_DIRECTORY, null);
+ if (sharedDir != null) {
+ Path shared = site.site_path.resolve(sharedDir);
+ FileUtil.mkdirsOrDie(shared, "cannot create " + shared);
+ }
+ }
+
+ private void configurePeer() {
+ mySection.string("Peer URL", URL_KEY, null);
+ mySection.string("User", USER_KEY, null);
+ mySection.string("Password", PASSWORD_KEY, null);
+ }
+
+ private void configureTimeouts() {
+ mySection.string("Connection timeout [ms]", CONNECTION_TIMEOUT_KEY, str(DEFAULT_TIMEOUT_MS));
+ mySection.string("Socket timeout [ms]", SOCKET_TIMEOUT_KEY, str(DEFAULT_TIMEOUT_MS));
+ }
+
+ private void configureRetry() {
+ mySection.string(
+ "Max number of tries to forward to remote peer", MAX_TRIES_KEY, str(DEFAULT_MAX_TRIES));
+ mySection.string("Retry interval [ms]", RETRY_INTERVAL_KEY, str(DEFAULT_RETRY_INTERVAL));
+ }
+
+ private void configureThreadPools() {
+ mySection.string(
+ "Index thread pool size", INDEX_THREAD_POOL_SIZE_KEY, str(DEFAULT_THREAD_POOL_SIZE));
+ mySection.string(
+ "Cache thread pool size", CACHE_THREAD_POOL_SIZE_KEY, str(DEFAULT_THREAD_POOL_SIZE));
+ }
+
+ private static String str(int n) {
+ return Integer.toString(n);
+ }
+
+ @Override
+ public void postRun() throws Exception {}
+}
