Laboratoire ICAR » Plateforme TXM

package org.txm.sw

import org.kohsuke.args4j.*

import groovy.transform.Field

import org.txm.rcpapplication.swt.widget.parameters.*

// PARAMETERS //

@Field @Option(name="infile",usage="input file", widget="File", required=true, def="")

File infile			// the file to process

@Field @Option(name="outdir",usage="output folder", widget="Folder", required=false, def="")

File outdir		// the directory which received result files

@Field @Option(name="regex",usage="search pattern", widget="String", required=false, def="(<\\?(.+))")

def regex         						// matching line test

@Field @Option(name="prefix",usage="result files prefix", widget="String", required=false, def="")

String prefix							// result files prefix

@Field @Option(name="replace",usage="replace with, use the matching line to build the filename", widget="String", required=false, def="\$2")

String replace							// use the matching line to build the filename

@Field @Option(name="ext",usage=" result files extension", widget="String", required=false, def=".xml")

String ext                                     			// result files extension

@Field @Option(name="includeSeparatorLine",usage="should include matching line in result files", widget="Boolean", required=false, def="true")

boolean includeSeparatorLine					// should include matching line in result files

@Field @Option(name="writeAtStartup",usage="should create files only if a matching line is found ", widget="Boolean", required=false, def="true")

boolean writeAtStartup						// should create files only if a matching line is found

@Field @Option(name="encoding",usage="file encoding", widget="String", required=false, def="UTF-8")

String encoding

// Open the parameters input dialog box

if (!ParametersDialog.open(this)) return;

regex = /$regex/ // create a pattern object

// CALL //

splitByRegExp(infile, outdir,

	regex, prefix, replace, ext,

//	regex, prefix, null, ext,

	includeSeparatorLine, writeAtStartup)

// PUBLIC METHODS //

def splitByRegExp(def infile, def outdir, def regex, def prefix, def replace, def ext, def includeSeparatorLine, def writeAtStartup) {

	int count = 1;

	outdir.deleteDir()

	outdir.mkdirs()

	if (!outdir.exists()) return false

	if (!infile.isFile()) return false

	File outfile = null;

	def writer = null

	if (writeAtStartup) {

			outfile = new File(outdir, prefix+(count++)+ext)

		writer = outfile.newWriter(encoding)

	}

	infile.eachLine(encoding) { line ->

		if (line ==~ regex) { // new file

			if (replace == null || replace == "") {

				outfile = new File(outdir, prefix+(count++)+ext)

			} else {

				String name = prefix+line.replaceAll(regex, replace)+ext

				outfile = new File(outdir, name)

			}

			if (writer != null)	writer.close() // close previous writer

			writer = outfile.newWriter(encoding)

			println "new file $outfile"

			if (includeSeparatorLine) {

				writer.println line

			}

		} else {

			if (writer != null)

				writer.println line

		}

	}

	if (writer != null)	writer.close()

}