opendaylight/config/yang-test-plugin/src/main/java/org/opendaylight/controller/config/yang/test/plugin/ProcessSources.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.controller.config.yang.test.plugin;
   9
  10 import org.apache.commons.io.FileUtils;
  11 import org.apache.maven.plugin.AbstractMojo;
  12 import org.apache.maven.plugin.MojoExecutionException;
  13 import org.apache.maven.plugin.MojoFailureException;
  14
  15 import java.io.File;
  16 import java.io.IOException;
  17 import java.util.regex.Pattern;
  18
  19 /**
  20  * Add implementation code from stub.txt
  21  *
  22  * @goal process-sources
  23  *
  24  * @phase process-sources
  25  *
  26  */
  27 public class ProcessSources extends AbstractMojo{
  28     /**
  29      * @parameter expression="${project.build.sourceDirectory}"
  30      * @readOnly
  31      * @required
  32      */
  33     private File directory;
  34
  35     @Override
  36     public void execute() throws MojoExecutionException, MojoFailureException {
  37         if (directory == null || !directory.exists()) {
  38             super.getLog().error("Directory does not exists.");
  39         }
  40         File sourceDirectory = new File(directory.getPath() + Util.replaceDots(".org.opendaylight.controller.config.yang.test.impl"));
  41         if (!sourceDirectory.exists()) {
  42             super.getLog().error("Source directory does not exists " + sourceDirectory.getPath());
  43         }
  44
  45         File[] sourceFiles = sourceDirectory.listFiles();
  46         for (File sourceFile: sourceFiles) {
  47             if (sourceFile.getName().endsWith(".java")) {
  48                 String sourceContent;
  49                 try {
  50                     sourceContent = FileUtils.readFileToString(sourceFile);
  51                 } catch (IOException e) {
  52                     getLog().error("Cannot read " + sourceFile.getAbsolutePath(), e);
  53                     continue;
  54                 }
  55                 if (sourceFile.getName().endsWith("Module.java") || sourceFile.getName().endsWith("ModuleFactory.java")) {
  56                     File stubFile = new File(sourceFile.getPath().replace(".java", "Stub.txt"));
  57                     if (stubFile.exists()) {
  58                         String stubContent = null;
  59                         try {
  60                             stubContent = FileUtils.readFileToString(stubFile);
  61                         } catch (IOException e) {
  62                             getLog().error("Cannot read " + stubFile.getAbsolutePath(), e);
  63                         }
  64                         if (stubContent != null) {
  65                             sourceContent = rewriteStub(sourceContent, stubContent);
  66                         }
  67                     }
  68                 }
  69                 // remove copyright headers as they can contain timestamp
  70                 sourceContent = removeCopyrights(sourceContent);
  71
  72                 // replace the file content
  73                 try {
  74                     FileUtils.write(sourceFile, sourceContent);
  75                 } catch (IOException e) {
  76                     getLog().error("Cannot write " + sourceFile.getAbsolutePath(), e);
  77                 }
  78             }
  79
  80         }
  81     }
  82
  83     private static Pattern MULTILINE_COMMENT_PATTERN = Pattern.compile("/\\*.*\\*/", Pattern.MULTILINE | Pattern.DOTALL);
  84     private static String removeCopyrights(String source) {
  85         String target = MULTILINE_COMMENT_PATTERN.matcher(source).replaceAll("\n");
  86         //FileUtils.write(sourceFile, target);
  87         return target;
  88     }
  89
  90     private static Pattern UNSUPPORTED_OP_PATTERN = Pattern.compile("^.*TODO.*\n.*throw new java.lang.UnsupportedOperationException.*$", Pattern.MULTILINE);
  91
  92     private static String rewriteStub(String source, String replaceTODOWith) {
  93         String target = UNSUPPORTED_OP_PATTERN.matcher(source).replaceFirst(replaceTODOWith);
  94         return target;
  95     }
  96 }