/**
*
* APDPlat - Application Product Development Platform
* Copyright (c) 2013, 杨尚川, yang-shangchuan@qq.com
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
package org.apdplat.module.module.service;
import org.apdplat.module.module.model.Command;
import org.apdplat.module.module.model.Module;
import org.apdplat.module.system.service.PropertyHolder;
import org.apdplat.platform.log.APDPlatLogger;
import org.apdplat.platform.util.FileUtils;
import org.apdplat.platform.util.XMLFactory;
import org.apdplat.platform.util.XMLUtils;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import org.apdplat.platform.log.APDPlatLoggerFactory;
/**
*
* @author 杨尚川
*/
public class ModuleParser {
private static final APDPlatLogger LOG = APDPlatLoggerFactory.getAPDPlatLogger(ModuleParser.class);
private static final String dtdFile="/target/module.dtd";
private static final String disableModules=PropertyHolder.getProperty("module.disable");
/**
* 获取各个module.xml中的根模块
* @return
*/
public static List<Module> getRootModules(){
LOG.info("module.disable:"+disableModules);
List<Module> modules=new ArrayList<>();
try{
Enumeration<URL> ps = Thread.currentThread().getContextClassLoader().getResources("META-INF/services/module.xml");
while(ps.hasMoreElements()) {
InputStream in = null;
try {
URL url=ps.nextElement();
LOG.info("找到模块描述文件："+url.getPath());
in = url.openStream();
Module root=getRootModule(in);
modules.add(root);
}catch(Exception e)
{
LOG.error("获取根模块出错",e);
}
finally {
try {
in.close();
} catch (IOException e) {
LOG.error("获取根模块出错",e);
}
}
}
}catch(Exception e){
LOG.error("获取根模块出错",e);
}
return modules;
}
/**
* 解析并获得根模块
* @param in
* @return
*/
public static Module getRootModule(InputStream in){
//准备模块DTD文件以供校验使用
prepareDtd();
try {
byte[] data=FileUtils.readAll(in);
String xml=new String(data,"utf-8");
LOG.info("将DTD文件替换为绝对路径");
xml=xml.replace("module.dtd", FileUtils.getAbsolutePath(dtdFile));
LOG.info("将模块文件读取到字节数组中，长度为："+data.length);
ByteArrayInputStream bin=new ByteArrayInputStream(xml.getBytes("utf-8"));
LOG.info("注册module.xml文件");
LOG.info("验证module.xml文件");
//校验文件
verifyFile(bin);
// 解析模块
Module module=parseModule(xml);
return module;
} catch (UnsupportedEncodingException e) {
LOG.error("获取根模块出错",e);
}
return null;
}
private static Module parseModule(String xml) {
XMLFactory factory=new XMLFactory(Module.class,Command.class);
Module module=factory.unmarshal(xml);
//将XML表示的模块转变为JAVA对象表示的模块
assembleMudule(module);
return module;
}
/**
* 采用递归的方式建立模块树的引用关系
* @param module 模块树的根
*/
private static void assembleMudule(Module module){
int order=1;
for(Command c : module.getCommands()){
c.setModule(module);
if(c.getOrderNum()==0){
c.setOrderNum(order++);
}
}
order=1;
List<Module> toDelete=new ArrayList<>();
for(Module m : module.getSubModules()){
//根据参数module.hide来设置模块
if(disableModules.contains(m.getEnglish())){
LOG.info("禁用: "+m.getChinese());
toDelete.add(m);
continue;
}else{
LOG.info("启用: "+m.getChinese());
}
m.setParentModule(module);
if(m.getOrderNum()==0){
m.setOrderNum(order++);
}
assembleMudule(m);
}
for(Module m : toDelete){
module.removeSubModule(m);
}
}
private static void prepareDtd(){
try{
Enumeration<URL> ps = Thread.currentThread().getContextClassLoader().getResources("META-INF/services/module.dtd");
if(ps.hasMoreElements()) {
InputStream in = null;
try {
URL url=ps.nextElement();
LOG.info("找到模块DTD文件："+url.getPath());
in = url.openStream();
byte[] data=FileUtils.readAll(in);
LOG.info("将DTD复制到："+dtdFile);
FileUtils.createAndWriteFile(dtdFile, data);
}catch(Exception e)
{
LOG.error("获取根模块出错",e);
}
finally {
try {
in.close();
} catch (IOException e) {
LOG.error("获取根模块出错",e);
}
}
}else{
LOG.info("没有找到模块DTD文件");
}
}catch(Exception e){
LOG.error("获取根模块出错",e);
}
}
private static void verifyFile(InputStream in){
boolean pass=XMLUtils.validateXML(in);
if(!pass){
LOG.info("验证没有通过，请参考module.dtd文件");
return ;
}
LOG.info("验证通过");
}
private static void print(Module module,String pre){
System.out.println(pre+module.getChinese()+":"+module.getEnglish()+":"+module.getOrderNum()+":"+module.isDisplay());
for(Command command : module.getCommands()){
System.out.println(pre+" "+command.getChinese()+":"+command.getEnglish()+":"+command.getOrderNum());
}
for(Module sub : module.getSubModules()){
print(sub,pre+" ");
}
}
public static void main(String[] args){
for(Module module : getRootModules()){
print(module,"");
}
}
}