org.onosproject.yang.runtime.helperutils.YangApacheUtils Maven / Gradle / Ivy
/*
* Copyright 2017-present Open Networking Foundation
*
* 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 org.onosproject.yang.runtime.helperutils;
import org.onosproject.yang.compiler.datamodel.YangNode;
import org.onosproject.yang.model.YangModel;
import org.osgi.framework.Bundle;
import org.osgi.framework.BundleContext;
import org.slf4j.Logger;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Pattern;
import static org.onosproject.yang.compiler.tool.YangCompilerManager.getYangNodes;
import static org.onosproject.yang.compiler.tool.YangCompilerManager.parseJarFile;
import static org.onosproject.yang.compiler.tool.YangCompilerManager.processYangModel;
import static org.osgi.framework.FrameworkUtil.getBundle;
import static org.slf4j.LoggerFactory.getLogger;
/**
* Utilities for runtime to use apache tools.
*/
public final class YangApacheUtils {
private static final String SLASH = File.separator;
private static final String HYPHEN = "-";
private static final String PERIOD = ".";
private static final String YANG_RESOURCES = "yang/resources";
private static final String SYSTEM = SLASH + "system" + SLASH;
private static final String MAVEN = "mvn:";
private static final String JAR = ".jar";
private static final String USER_DIRECTORY = "user.dir";
private static final Logger log = getLogger(YangApacheUtils.class);
// Forbid construction.
private YangApacheUtils() {
}
/**
* Returns YANG model for generated module class.
*
* @param modClass generated module class
* @return YANG model
*/
public static YangModel getYangModel(Class modClass) {
BundleContext context = getBundle(modClass).getBundleContext();
if (context != null) {
Bundle bundle = context.getBundle();
List curNodes = new ArrayList<>();
String jarPath;
String metaPath;
jarPath = getJarPathFromBundleLocation(
bundle.getLocation(), context.getProperty(USER_DIRECTORY));
metaPath = jarPath + SLASH + YANG_RESOURCES + SLASH;
YangModel model = processJarParsingOperations(jarPath);
if (model != null) {
curNodes.addAll(getYangNodes(model));
// process model creations.
if (!curNodes.isEmpty()) {
try {
return processYangModel(metaPath, curNodes,
model.getYangModelId(), false);
} catch (IOException e) {
log.error(" failed to create process YANG model " +
e.getMessage(), e);
}
}
}
}
return null;
}
/**
* Returns jar path from bundle mvnLocationPath.
*
* @param mvnLocationPath mvnLocationPath of bundle
* @return path of jar
*/
private static String getJarPathFromBundleLocation(String mvnLocationPath,
String currentDirectory) {
StringBuilder builder = new StringBuilder();
builder.append(currentDirectory).append(SYSTEM);
StringBuilder ver = new StringBuilder();
if (mvnLocationPath.contains(MAVEN)) {
String[] strArray = mvnLocationPath
.replaceFirst("\\$Bundle-.*$", "")
.split(MAVEN);
if (strArray[1].contains(File.separator)) {
String[] split = strArray[1].split(File.separator);
if (split[0].contains(PERIOD)) {
String[] groupId = split[0].split(Pattern.quote(PERIOD));
for (String s : groupId) {
builder.append(s).append(SLASH);
}
for (int i = 1; i < split.length; i++) {
builder.append(split[i]).append(SLASH);
ver.append(split[i]).append(HYPHEN);
}
builder.append(ver);
builder.deleteCharAt(builder.length() - 1);
return builder.toString();
}
}
}
return null;
}
/**
* Process jar file for fetching YANG model.
*
* @param path jar file path
* @return YANG model
*/
private static YangModel processJarParsingOperations(String path) {
//Deserialize data model and get the YANG node set.
String jar = path + JAR;
try {
File file = new File(jar);
if (file.exists()) {
return parseJarFile(path + JAR, path);
}
} catch (IOException e) {
log.error(" failed to parse the jar file in path {} : {} ", path,
e.getMessage());
}
return null;
}
}