Section.java

// Copyright (C) 2009 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.google.gerrit.pgm.init;

import com.google.gerrit.pgm.util.ConsoleUI;
import com.google.gerrit.server.config.ConfigUtil;
import com.google.gerrit.server.config.SitePaths;
import com.google.inject.Inject;
import com.google.inject.assistedinject.Assisted;

import java.io.File;
import java.util.ArrayList;
import java.util.Arrays;

/** Helper to edit a section of the configuration files. */
class Section {
  interface Factory {
    Section get(String name);
  }

  private final InitFlags flags;
  private final SitePaths site;
  private final ConsoleUI ui;
  private final String section;

  @Inject
  Section(final InitFlags flags, final SitePaths site, final ConsoleUI ui,
      @Assisted final String section) {
    this.flags = flags;
    this.site = site;
    this.ui = ui;
    this.section = section;
  }

  String get(String name) {
    return flags.cfg.getString(section, null, name);
  }

  void set(final String name, final String value) {
    final ArrayList<String> all = new ArrayList<String>();
    all.addAll(Arrays.asList(flags.cfg.getStringList(section, null, name)));

    if (value != null) {
      if (all.size() == 0 || all.size() == 1) {
        flags.cfg.setString(section, null, name, value);
      } else {
        all.set(0, value);
        flags.cfg.setStringList(section, null, name, all);
      }

    } else if (all.size() == 0) {
    } else if (all.size() == 1) {
      flags.cfg.unset(section, null, name);
    } else {
      all.remove(0);
      flags.cfg.setStringList(section, null, name, all);
    }
  }

  <T extends Enum<?>> void set(final String name, final T value) {
    if (value != null) {
      set(name, value.name());
    } else {
      unset(name);
    }
  }

  void unset(String name) {
    set(name, (String) null);
  }

  String string(final String title, final String name, final String dv) {
    return string(title, name, dv, false);
  }

  String string(final String title, final String name, final String dv,
      final boolean nullIfDefault) {
    final String ov = get(name);
    String nv = ui.readString(ov != null ? ov : dv, "%s", title);
    if (nullIfDefault && nv == dv) {
      nv = null;
    }
    if (!eq(ov, nv)) {
      set(name, nv);
    }
    return nv;
  }

  File path(final String title, final String name, final String defValue) {
    return site.resolve(string(title, name, defValue));
  }

  <T extends Enum<?>> T select(final String title, final String name,
      final T defValue) {
    return select(title, name, defValue, false);
  }

  <T extends Enum<?>> T select(final String title, final String name,
      final T defValue, final boolean nullIfDefault) {
    final boolean set = get(name) != null;
    T oldValue = ConfigUtil.getEnum(flags.cfg, section, null, name, defValue);
    T newValue = ui.readEnum(oldValue, "%s", title);
    if (nullIfDefault && newValue == defValue) {
      newValue = null;
    }
    if (!set || oldValue != newValue) {
      if (newValue != null) {
        set(name, newValue);
      } else {
        unset(name);
      }
    }
    return newValue;
  }

  String password(final String username, final String password) {
    final String ov = flags.sec.getString(section, null, password);

    String user = flags.sec.getString(section, null, username);
    if (user == null) {
      user = get(username);
    }

    if (user == null) {
      flags.sec.unset(section, null, password);
      return null;
    }

    if (ov != null) {
      // If the user already has a password stored, try to reuse it
      // rather than prompting for a whole new one.
      //
      if (ui.isBatch() || !ui.yesno(false, "Change %s's password", user)) {
        return ov;
      }
    }

    final String nv = ui.password("%s's password", user);
    if (!eq(ov, nv)) {
      if (nv != null) {
        flags.sec.setString(section, null, password, nv);
      } else {
        flags.sec.unset(section, null, password);
      }
    }
    return nv;
  }

  private static boolean eq(final String a, final String b) {
    if (a == null && b == null) {
      return true;
    }
    return a != null ? a.equals(b) : false;
  }
}