com.google.eclipse.protobuf.ui/src/com/google/eclipse/protobuf/ui/protoc/command/CodeGenerationProtocOption.java

/*
 * Copyright (c) 2012 Google Inc.
 *
 * All rights reserved. This program and the accompanying materials are made available under the terms of the Eclipse
 * Public License v1.0 which accompanies this distribution, and is available at
 *
 * http://www.eclipse.org/legal/epl-v10.html
 */
package com.google.eclipse.protobuf.ui.protoc.command;

import static java.io.File.separator;
import static org.eclipse.core.runtime.IPath.SEPARATOR;

import org.eclipse.core.resources.*;
import org.eclipse.core.runtime.*;

import com.google.common.annotations.VisibleForTesting;
import com.google.eclipse.protobuf.ui.preferences.compiler.CodeGenerationPreference;

/**
 * @author alruiz@google.com (Alex Ruiz)
 */
class CodeGenerationProtocOption implements ProtocOption {
  private static final NullProgressMonitor NO_MONITOR = new NullProgressMonitor();

  private final String optionName;
  private final CodeGenerationPreference preferences;

  private boolean initialized;
  private boolean enabled;
  private IFolder outputDirectory;
  private String outputDirectoryLocation;

  CodeGenerationProtocOption(String optionName, CodeGenerationPreference preferences) {
    this.optionName = optionName;
    this.preferences = preferences;
  }

  @Override public final void addOptionTo(ProtocCommand command) throws CoreException {
    ensureIsInitialized();
    if (enabled) {
      command.appendOption(optionName, outputDirectoryLocation);
    }
  }

  final IFolder outputDirectory() throws CoreException {
    ensureIsInitialized();
    return outputDirectory;
  }

  private void ensureIsInitialized() throws CoreException {
    if (!initialized) {
      initialize();
    }
  }

  private void initialize() throws CoreException {
    initialized = true;
    enabled = preferences.isEnabled();
    if (enabled) {
      outputDirectory = findOrCreateDirectory(preferences.outputDirectory());
      outputDirectoryLocation = outputDirectory.getLocation().toOSString();
    }
  }

  private IFolder findOrCreateDirectory(String directoryName) throws CoreException {
    IFolder directory = null;
    StringBuilder path = new StringBuilder();
    IProject project = preferences.project();
    for (String segment : segmentsOf(directoryName)) {
      path.append(segment);
      directory = project.getFolder(path.toString());
      if (!directory.exists()) {
        directory.create(true, true, NO_MONITOR);
      }
      path.append(SEPARATOR);
    }
    return directory;
  }

  @VisibleForTesting static String[] segmentsOf(String path) {
    if (path == null) {
      throw new NullPointerException("The given path should not be null");
    }
    return path.split("\\Q" + separator + "\\E");
  }
}