fitnesse.authentication.PasswordFile Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fitnesse Show documentation

The fully integrated standalone wiki, and acceptance testing framework.

There is a newer version: 20241207

// Copyright (C) 2003-2009 by Object Mentor, Inc. All rights reserved.
// Released under the terms of the CPL Common Public License version 1.0.
package fitnesse.authentication;

import fitnesse.util.ClassUtils;
import util.FileUtil;

import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

public class PasswordFile {
  private final File passwordFile;
  private Map passwordMap = new HashMap<>();
  private PasswordCipher cipher = new TransparentCipher();

  public PasswordFile(String filename) throws IOException, ReflectiveOperationException {
    passwordFile = new File(filename);
    loadFile();
  }

  public PasswordFile(String filename, PasswordCipher cipher) throws IOException, ReflectiveOperationException {
    this(filename);
    this.cipher = cipher;
  }

  public Map getPasswordMap() {
    return passwordMap;
  }

  public String getName() {
    return passwordFile.getName();
  }

  public PasswordCipher getCipher() {
    return cipher;
  }

  public void savePassword(String user, String password) throws IOException {
    passwordMap.put(user, cipher.encrypt(password));
    savePasswords();
  }

  private void loadFile() throws IOException, ReflectiveOperationException {
    LinkedList lines = getPasswordFileLines();
    loadCipher(lines);
    loadPasswords(lines);
  }

  private void loadPasswords(LinkedList lines) {
    for (String line : lines) {
      if (!"".equals(line)) {
        String[] tokens = line.split(":");
        passwordMap.put(tokens[0], tokens[1]);
      }
    }
  }

  private void loadCipher(LinkedList lines) throws IllegalAccessException, ClassNotFoundException, InstantiationException {
    if (!lines.isEmpty()) {
      String firstLine = lines.getFirst();
      if (firstLine.startsWith("!")) {
        String cipherClassName = firstLine.substring(1);
        instantiateCipher(cipherClassName);
        lines.removeFirst();
      }
    }
  }

  public PasswordCipher instantiateCipher(String cipherClassName) throws InstantiationException, IllegalAccessException, ClassNotFoundException {
    cipher = (PasswordCipher) ClassUtils.forName(cipherClassName).newInstance();
    return cipher;
  }

  private void savePasswords() throws IOException {
    List lines = new LinkedList<>();
    lines.add("!" + cipher.getClass().getName());
    for (Map.Entry entry : passwordMap.entrySet()) {
      String user = entry.getKey();
      String password = entry.getValue();
      lines.add(user + ":" + password);
    }
    FileUtil.writeLinesToFile(passwordFile, lines);
  }

  private LinkedList getPasswordFileLines() throws IOException {
    LinkedList lines = new LinkedList<>();
    if (passwordFile.exists())
      lines.addAll(FileUtil.getFileLines(passwordFile));
    return lines;
  }
}