no-java/src/nodash/core/NoDefaultAdapter.java

package nodash.core;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.security.PublicKey;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

import javax.crypto.BadPaddingException;
import javax.crypto.IllegalBlockSizeException;

import org.apache.commons.codec.binary.Base64;

import nodash.core.exceptions.NoAdapterException;
import nodash.exceptions.NoDashFatalException;
import nodash.exceptions.NoUserAlreadyOnlineException;
import nodash.exceptions.NoUserNotValidException;
import nodash.models.NoByteSet;
import nodash.models.NoSession;

public class NoDefaultAdapter implements NoAdapter {
  private static Map<PublicKey, Collection<NoByteSet>> byteSets =
      new ConcurrentHashMap<PublicKey, Collection<NoByteSet>>();
  private static Map<String, NoSession> sessions = new ConcurrentHashMap<String, NoSession>();
  private static Set<String> online = Collections
      .newSetFromMap(new ConcurrentHashMap<String, Boolean>());

  private static final String HASH_FILE = "nosystem.hash";

  private static byte[] getFile() throws NoAdapterException {
    try {
      return alterFile(null, null);
    } catch (IOException e) {
      throw new NoAdapterException("Could not get hash file.", e);
    }
  }

  private static synchronized byte[] alterFile(byte[] hashToAdd, byte[] hashToRemove)
      throws IOException {
    if (!(hashToAdd == null ^ hashToRemove == null) && Arrays.equals(hashToAdd, hashToRemove)) {
      throw new IllegalArgumentException("Hashes to add and remove cannot be the same.");
    }

    File file = new File(HASH_FILE);
    byte[] originalFileBytes = Files.readAllBytes(file.toPath());
    if (hashToAdd == null && hashToRemove == null) {
      return originalFileBytes;
    }

    int hashes = originalFileBytes.length / 64;
    List<Byte> newFile = new ArrayList<Byte>();

    for (int x = 0; x < hashes; x++) {
      byte[] hash = Arrays.copyOfRange(originalFileBytes, x * 64, x * 64 + 64);

      if (hashToRemove == null || !Arrays.equals(hash, hashToRemove)) {
        for (byte hashByte : hash) {
          newFile.add(hashByte);
        }
      }

      if (hashToAdd != null || Arrays.equals(hash, hashToAdd)) {
        hashToAdd = null;
      }
    }

    if (hashToAdd != null) {
      for (byte hashByte : hashToAdd) {
        newFile.add(hashByte);
      }
    }

    byte[] newFileAsPrimitive = new byte[newFile.size()];
    for (int x = 0; x < newFile.size(); x++) {
      newFileAsPrimitive[x] = newFile.get(x).byteValue();
    }

    Files.write(file.toPath(), newFileAsPrimitive);

    return newFileAsPrimitive;
  }

  public NoDefaultAdapter() {

  }

  @Override
  public void insertHash(byte[] hash) throws NoAdapterException {
    try {
      alterFile(hash, null);
    } catch (IOException e) {
      throw new NoAdapterException("Trouble while inserting hash.", e);
    }
  }

  @Override
  public void removeHash(byte[] hash) throws NoAdapterException {
    try {
      alterFile(null, hash);
    } catch (IOException e) {
      throw new NoAdapterException("Trouble removing hash.", e);
    }
  }

  @Override
  public void checkHash(byte[] hash) throws NoAdapterException, NoUserNotValidException {
    byte[] hashFile = getFile();
    int hashes = hashFile.length / 64;
    for (int x = 0; x < hashes; x++) {
      byte[] examine = Arrays.copyOfRange(hashFile, x * 64, x * 64 + 64);
      if (Arrays.equals(hash, examine)) {
        return;
      }
    }
    throw new NoUserNotValidException();
  }

  @Override
  public byte[][] exportHashes() throws NoAdapterException {
    byte[] hashFile = getFile();
    int hashes = hashFile.length / 64;
    byte[][] export = new byte[hashes][64];
    for (int x = 0; x < hashes; x++) {
      byte[] hash = Arrays.copyOfRange(hashFile, x * 64, x * 64 + 64);
      export[x] = hash;
    }
    return export;
  }

  @Override
  public long hashCount() throws NoAdapterException {
    byte[] hashFile = getFile();
    return hashFile.length / 64;
  }

  @Override
  public void addNoSession(NoSession session) {
    if (containsNoSession(session.getEncryptedUuid())) {
      throw new NoDashFatalException("No such session exists.");
    }
    sessions.put(session.getUuid(), session);
  }

  @Override
  public boolean containsNoSession(byte[] encryptedUuid) {
    String uuid;
    try {
      uuid = Base64.encodeBase64String(NoUtil.decrypt(encryptedUuid));
    } catch (IllegalBlockSizeException | BadPaddingException e) {
      throw new NoDashFatalException("Could not decrypt given UUID.", e);
    }

    return sessions.containsKey(uuid);
  }

  @Override
  public void shredNoSession(byte[] encryptedUuid) {
    NoSession session = getNoSession(encryptedUuid);
    addNoByteSets(session.getIncomingSafe(), session.getNoUserSafe().getRsaPublicKey());
    sessions.remove(session.getUuid());
  }

  @Override
  public NoSession getNoSession(byte[] encryptedUuid) {
    if (containsNoSession(encryptedUuid)) {
      String uuid;
      try {
        uuid = Base64.encodeBase64String(NoUtil.decrypt(encryptedUuid));
      } catch (IllegalBlockSizeException | BadPaddingException e) {
        throw new NoDashFatalException("Could not decrypt given UUID.", e);
      }
      return sessions.get(uuid);
    }
    throw new NoDashFatalException("No such session exists.");
  }

  @Override
  public Collection<NoByteSet> pollNoByteSets(PublicKey address) {
    if (byteSets.containsKey(address)) {
      Collection<NoByteSet> storedByteSets = byteSets.get(address);
      Collection<NoByteSet> result = new ArrayList<NoByteSet>();
      for (NoByteSet byteSet : storedByteSets) {
        result.add(byteSet);
        storedByteSets.remove(byteSet);
      }
      return result;
    } else {
      return new ArrayList<NoByteSet>();
    }
  }

  @Override
  public void addNoByteSets(Collection<NoByteSet> addedByteSets, PublicKey address) {
    if (byteSets.containsKey(address)) {
      byteSets.get(address).addAll(addedByteSets);
    } else {
      byteSets.put(address, addedByteSets);
    }
  }

  @Override
  public void goOnline(byte[] hash) throws NoUserAlreadyOnlineException {
    String hashString = Base64.encodeBase64String(hash);
    if (online.contains(hashString)) {
      throw new NoUserAlreadyOnlineException();
    }
    online.add(hashString);
  }

  @Override
  public boolean isOnline(byte[] hash) {
    String hashString = Base64.encodeBase64String(hash);
    return online.contains(hashString);
  }

  @Override
  public void goOffline(byte[] hash) {
    String hashString = Base64.encodeBase64String(hash);
    online.remove(hashString);
  }

}