org.onos.yangtools.yang2sources.plugin.ConfigArg.java Source code

Introduction

Here is the source code for org.onos.yangtools.yang2sources.plugin.ConfigArg.java

Source

/*
 * Copyright (c) 2013 Cisco Systems, Inc. and others.  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 org.onos.yangtools.yang2sources.plugin;

import java.io.File;
import java.util.Map;

import org.apache.maven.project.MavenProject;

import com.google.common.base.Preconditions;
import com.google.common.collect.Maps;

/**
 * Base complex configuration arguments
 */
public abstract class ConfigArg {

    private final File outputBaseDir;

    public ConfigArg(String outputBaseDir) {
        this.outputBaseDir = outputBaseDir == null ? null : new File(outputBaseDir);
    }

    public File getOutputBaseDir(MavenProject project) {
        if (outputBaseDir == null) {
            return null;
        }
        if (outputBaseDir.isAbsolute()) {
            return outputBaseDir;
        } else {
            return new File(project.getBasedir(), outputBaseDir.getPath());
        }
    }

    public abstract void check();

    /**
     * Configuration argument for code generator class and output directory.
     */
    public static final class CodeGeneratorArg extends ConfigArg {
        private static final String TARGET_GENERATED_SOURCES = "target" + File.separator + "generated-sources";
        private static final String CODE_GEN_DEFAULT_RESOURCE_DIR = TARGET_GENERATED_SOURCES + File.separator
                + "spi";
        public static final String YANG_GENERATED_DIR = TARGET_GENERATED_SOURCES + File.separator + "yang";

        private String codeGeneratorClass;
        private File resourceBaseDir = new File(CODE_GEN_DEFAULT_RESOURCE_DIR);

        private Map<String, String> additionalConfiguration = Maps.newHashMap();

        public CodeGeneratorArg() {
            super(null);
        }

        public CodeGeneratorArg(String codeGeneratorClass) {
            this(codeGeneratorClass, null);
        }

        public CodeGeneratorArg(String codeGeneratorClass, String outputBaseDir) {
            super(outputBaseDir);
            this.codeGeneratorClass = codeGeneratorClass;
        }

        public CodeGeneratorArg(String codeGeneratorClass, String outputBaseDir, String resourceBaseDir) {
            super(outputBaseDir);
            this.codeGeneratorClass = codeGeneratorClass;
            this.resourceBaseDir = new File(resourceBaseDir);
        }

        @Override
        public void check() {
            Preconditions.checkNotNull(codeGeneratorClass, "codeGeneratorClass for CodeGenerator cannot be null");
        }

        public String getCodeGeneratorClass() {
            return codeGeneratorClass;
        }

        public File getResourceBaseDir(MavenProject project) {
            if (resourceBaseDir.isAbsolute()) {
                return resourceBaseDir;
            } else {
                return new File(project.getBasedir(), resourceBaseDir.getPath());
            }
        }

        public Map<String, String> getAdditionalConfiguration() {
            return additionalConfiguration;
        }
    }
}