org.beangle.style.maven.FormatSpaceMojo.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of style-maven-plugin Show documentation
There is a newer version: 28.2
/*
 * Beangle, Agile Development Scaffold and Toolkits.
 *
 * Copyright © 2005, The Beangle Software.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see .
 */
package org.beangle.style.maven

import java.io.File
import org.apache.maven.plugin.AbstractMojo
import org.apache.maven.plugins.annotations.{ Mojo, Parameter }
import org.apache.maven.project.MavenProject
import org.beangle.style.util.Files./
import org.beangle.style.core.{ DefaultWhiteSpaceFormater, WhiteSpaceFormater }

@Mojo(name = "ws-format")
class FormatSpaceMojo extends AbstractMojo {

  @Parameter(defaultValue = "${project}", readonly = true)
  private var project: MavenProject = _

  def execute(): Unit = {
    val builder = DefaultWhiteSpaceFormater.newBuilder()
    var tabLength = System.getProperty("tablength")
    if (null == tabLength) tabLength = "2"
    builder.enableTab2space(Integer.parseInt(tabLength)).enableTrimTrailingWhiteSpace().insertFinalNewline()
    val formater = builder.build()

    val dir = System.getProperty("dir")
    val ext = Option(System.getProperty("ext"))

    if (null == dir) {
      import scala.jdk.CollectionConverters._
      project.getCompileSourceRoots.asScala foreach { resource =>
        format(resource, formater, ext)
      }

      project.getTestCompileSourceRoots.asScala foreach { resource =>
        format(resource, formater, ext)
      }

      project.getResources.asScala foreach { resource =>
        format(resource.getDirectory, formater, ext)
      }

      project.getTestResources.asScala foreach { resource =>
        format(resource.getDirectory, formater, ext)
      }
    } else {
      if (new File(dir).exists()) {
        format(dir, formater, ext)
      } else {
        format(project.getBasedir.getAbsolutePath + / + dir, formater, ext)
      }
    }
  }

  private def format(path: String, formater: WhiteSpaceFormater, ext: Option[String]): Unit = {
    if (new File(path).exists()) {
      println(s"formating $path ...")
      WhiteSpaceFormater.format(formater, new File(path), ext)
    }
  }
}