nbbuild/antsrc/org/netbeans/nbbuild/LocFiles.java - netbeans - Git at Google

 /*
  * Licensed to the Apache Software Foundation (ASF) under one
  * or more contributor license agreements.  See the NOTICE file
  * distributed with this work for additional information
  * regarding copyright ownership.  The ASF licenses this file
  * to you 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.netbeans.nbbuild;

 import java.io.File;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.List;
 import java.util.Locale;
 import java.util.StringTokenizer;
 import org.apache.tools.ant.BuildException;
 import org.apache.tools.ant.DirectoryScanner;
 import org.apache.tools.ant.Project;
 import org.apache.tools.ant.Task;
 import org.apache.tools.ant.taskdefs.Jar;
 import org.apache.tools.ant.taskdefs.Mkdir;
 import org.apache.tools.ant.types.PatternSet;

 public final class LocFiles extends Task {
     private String patternset;
     public void setPatternSet(String s) {
         patternset = s;
     }

     private String cluster;
     public void setCluster(String cluster) {
         this.cluster = cluster;
     }

     private String locales;
     public void setLocales(String locales) {
         this.locales = locales;
     }

     private File srcDir;

     public void setSrc(File f) {
         srcDir = f;
     }
     File distDir = null;

     public void setDestDir(File d) {
         distDir = d;
     }
     String cnbDashes;

     public void setCodeNameBase(String d) {
         assert d.indexOf('-') == -1;
         cnbDashes = d.replace('.', '-');
     }
     File nbmsLocation = null;

     public void setNBMs(File f) {
         nbmsLocation = f;
     }

     private String baseFiles;
     public void setBaseFilesRef(String refid) {
         baseFiles = refid;
     }

     @Override
     public void execute() throws BuildException {
         List<String> includes = new ArrayList<>();
         if (locales != null && !locales.isEmpty()) {
             if (!srcDir.exists()) {
                 log("No l10n files present. Do hg clone http://hg.netbeans.org/main/l10n!", Project.MSG_VERBOSE);
                 return;
             }

             StringTokenizer tok = new StringTokenizer(locales, ",");
             while (tok.hasMoreElements()) {
                 String locale = tok.nextToken();
                 if (locale.equals("default")) { // NOI18N
                     locale = Locale.getDefault().toString();
                 }
                 for (String l = locale; l != null; l = trailingUnderscore(l)) {
                     processLocale(l, includes, l.equals(locale));
                 }
             }
         }

         if (patternset != null) {
             PatternSet ps = new PatternSet();
             ps.setProject(getProject());
             if (includes.isEmpty()) {
                 ps.createInclude().setName("I/dont/exist/at/all");
             } else {
                 for (String s : includes) {
                     ps.createInclude().setName(s);
                 }
             }
             getProject().addReference(patternset, ps);
         }

     }

     private static String trailingUnderscore(String s) {
         int under = s.lastIndexOf('_');
         if (under == -1) {
             return null;
         }
         return s.substring(0, under);
     }

     private static String findModuleDir(String cnbDashes) {
         String pref = "org-netbeans-modules-";
         if (cnbDashes.startsWith(pref)) {
             return cnbDashes.substring(pref.length());
         }
         return cnbDashes;
     }

     private void processLocale(String locale, Collection<? super String> toAdd, boolean warn) {
         File baseSrcDir = fileFrom(srcDir, locale);
         if (!baseSrcDir.exists()) {
             if (warn) {
                 log("No files for locale: " + locale);
             }
             return;
         }
         log("Found L10N dir " + baseSrcDir, Project.MSG_VERBOSE);

         final String moduleDir = findModuleDir(cnbDashes);
         File locBaseDir = fileFrom(baseSrcDir, cluster, moduleDir);
         if (!locBaseDir.exists()) {
             log("Can't find directory " + locBaseDir, Project.MSG_WARN);
             return;
         }
         File[] ch = locBaseDir.listFiles();
         if (ch == null) {
             throw new BuildException("Surprising content of " + locBaseDir);
         }
         for (File f : ch) {
             processLocaleJar(f, locale, toAdd, moduleDir, locBaseDir, baseSrcDir);
         }
     }

     private void processLocaleJar(
         File file, String locale, Collection<? super String> toAdd,
         String moduleDir, File locBaseDir, File baseSrcDir
     ) {
         DirectoryScanner ds = new DirectoryScanner();
         int segments = -1;
         int dirIndex = -1;
         final File root;
         final String prefixRoot;
         final int remove;
         if (file.getName().equals(moduleDir)) {
             ds.setIncludes(new String[] { moduleDir });
             root = fileFrom(distDir, cluster);
             prefixRoot = null;
             remove = 0;
         } else {
             if (file.getName().equals("netbeans")) {
                 ds.setIncludes(new String[] {
                     "netbeans/**/org", "netbeans/**/com"
                 });
                 remove = 4;
                 segments = 3;
                 dirIndex = 1;
                 root = fileFrom(distDir, cluster);
                 prefixRoot = "";
             } else {
                 assert file.getName().equals("ext") ||
                     file.getName().equals("locale");

                 ds.setIncludes(new String[] { file.getName() + "/*" });
                 segments = 2;
                 dirIndex = 0;
                 root = fileFrom(distDir, cluster, "modules");
                 prefixRoot = "modules/";
                 remove = 0;
             }
         }

         ds.setBasedir(locBaseDir);
         ds.scan();
         for (String dir : ds.getIncludedDirectories()) {
             if (remove > 0) {
                 dir = dir.substring(0, dir.length() - remove);
             }

             Jar jar = new Jar();
             jar.setProject(getProject());
             Mkdir mkdir = new Mkdir();
             mkdir.setProject(getProject());

             String jarFileName;
             String subPath = "";
             File jarDir;
             String prefixDir;
             if (dir.equals(moduleDir)) {
                 jarDir = fileFrom(root, "modules", "locale");
                 prefixDir = "modules/locale/";
                 jar.setBasedir(fileFrom(locBaseDir, dir));
                 jarFileName = cnbDashes + "_" + locale + ".jar";
             } else {
                 // netbeans/*/* case or ext/* case
                 jar.setBasedir(fileFrom(locBaseDir, dir));
                 String[] arr = dir.split("/");
                 assert arr.length >= segments : "Expected segments: " + dir;
                 final int jarIndex = arr.length - 1;
                 jarDir = fileFromIf(
                     fileFrom(root, arr, dirIndex, jarIndex, true),
                     "locale"
                 );
                 prefixDir = nameFromIf(
                     nameFrom(prefixRoot, arr, dirIndex, jarIndex, true),
                     "locale"
                 );
                 jarFileName = fixJarName(prefixDir, arr[jarIndex]) + "_" + locale + ".jar";
             }
             String fullName = prefixDir + jarFileName;
             toAdd.add(fullName);
             if (jarDir == null) {
                 // in patternSet only mode
                 continue;
             }

             /*
             String subPath = dir.substring((cluster + File.separator + nbm + File.separator).length() - 1, dir.lastIndexOf(File.separator));
             if (!subPath.startsWith(File.separator + "netbeans")) {
                 subPath = File.separator + "modules" + subPath;
                 if (!name.startsWith("org-") && !(subPath.endsWith(File.separator + "ext") || subPath.endsWith(File.separator + "ext" + File.separator + "locale"))) {
                     name = "org-netbeans-modules-" + name;
                 } else {
                     // Handle exception from ext/
                     if (name.startsWith("web-httpmonitor") || name.startsWith("deployment-deviceanywhere")) {
                         name = "org-netbeans-modules-" + name;
                     }
                 }
             } else {
                 subPath = subPath.substring((File.separator + "netbeans").length());
                 //Handle exceptions form nblib
                 if (name.startsWith("j2ee-ant") || name.startsWith("deployment-deviceanywhere") || name.startsWith("mobility-project") || name.startsWith("java-j2seproject-copylibstask")) {
                     name = "org-netbeans-modules-" + name;
                 }
             }
             nbm = nbm.replaceAll("^vw-rh", "visualweb-ravehelp-rave_nbpack");
             nbm = nbm.replaceAll("^vw-", "visualweb-");
             if (!nbm.startsWith("org-") && !nbm.startsWith("com-")) {
                 nbm = "org-netbeans-modules-" + nbm;
             }
             */

             if (subPath.matches(".*/docs$")) {
                 ds.setBasedir(fileFrom(baseSrcDir, dir));
                 ds.setIncludes(new String[]{"**/*.hs"});
                 ds.setExcludes(new String[]{""});
                 ds.scan();
                 if (ds.getIncludedFilesCount() != 1) {
                     throw new BuildException("Can't find .hs file for " + cnbDashes + " module.");
                 }
                 File hsFile = fileFrom(baseSrcDir, dir, ds.getIncludedFiles()[0]);
                 File baseJHDir = hsFile.getParentFile();

                 Task locJH = getProject().createTask("locjhindexer");
                 try {
                     System.out.println("Basedir: " + baseJHDir.getAbsolutePath());
                     locJH.getClass().getMethod("setBasedir", File.class).invoke(locJH, baseJHDir);
                     locJH.getClass().getMethod("setLocales", String.class).invoke(locJH, locale);
                     locJH.getClass().getMethod("setDbdir", String.class).invoke(locJH, "JavaHelpSearch");
               //      ((Path)locJH.getClass().getMethod("createClasspath").invoke(locJH)).add(classpath);

                 } catch (Exception ex) {
                     throw new BuildException("Can't run locJHInxeder", ex);
                 }
                 locJH.execute();
             }
             mkdir.setDir(jarDir);
             mkdir.execute();
             jar.setDestFile(fileFrom(jarDir, jarFileName));
             jar.execute();
         }
     }

     private static File fileFrom(File base, String... paths) {
         return fileFrom(base, paths, 0, paths.length, true);
     }
     private static File fileFrom(File base, String[] paths, int from, int upTo, boolean addIfPresent) {
         if (base == null) {
             return null;
         }
         File f = base;
         while (from < upTo) {
             if (addIfPresent || !f.getName().equals(paths[from])) {
                 f = new File(f, paths[from]);
             }
             from++;
         }
         return f;
     }
     private static String nameFrom(String base, String... paths) {
         return nameFrom(base, paths, 0, paths.length, true);
     }
     private static String nameFrom(String base, String[] paths, int from, int upTo, boolean addIfPresent) {
         if (base == null) {
             return null;
         }
         String f = base;
         while (from < upTo) {
             if (addIfPresent || !f.endsWith(paths[from] + "/")) {
                 f += paths[from];
                 f += '/';
             }
             from++;
         }
         return f;
     }

     private static File fileFromIf(File fileFrom, String... add) {
         return fileFrom(fileFrom, add, 0, add.length, false);
     }
     private static String nameFromIf(String fileFrom, String... add) {
         return nameFrom(fileFrom, add, 0, add.length, false);
     }

     private String fixJarName(String prefixDir, String name) {
         if (prefixDir.endsWith("locale/") && baseFiles != null) {
             prefixDir = prefixDir.substring(0, prefixDir.length() - 7);
             PatternSet bf = (PatternSet)getProject().getReference(baseFiles);
             for (String p : bf.getIncludePatterns(getProject())) {
                 if (p.startsWith(prefixDir)) {
                     String realName = p.substring(prefixDir.length());
                     if (realName.indexOf('/') != -1 || realName.indexOf('\\') != -1) {
                         continue;
                     }
                     int indx = realName.indexOf(name);
                     if (indx == -1) {
                         continue;
                     }
                     return realName.substring(0, indx + name.length());
                 }
             }
         }


         if (name.equals("autoupdate-ui_nb")) {
             return "org-netbeans-modules-autoupdate-ui_nb";
         }
         if (name.equals("options-api_nb")) {
             return "org-netbeans-modules-options-api_nb";
         }
         if (name.equals("uihandler_nb")) {
             return "org-netbeans-modules-uihandler_nb";
         }
         return name;
     }
 }
	/*
	* Licensed to the Apache Software Foundation (ASF) under one
	* or more contributor license agreements. See the NOTICE file
	* distributed with this work for additional information
	* regarding copyright ownership. The ASF licenses this file
	* to you 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.netbeans.nbbuild;

	import java.io.File;
	import java.util.ArrayList;
	import java.util.Collection;
	import java.util.List;
	import java.util.Locale;
	import java.util.StringTokenizer;
	import org.apache.tools.ant.BuildException;
	import org.apache.tools.ant.DirectoryScanner;
	import org.apache.tools.ant.Project;
	import org.apache.tools.ant.Task;
	import org.apache.tools.ant.taskdefs.Jar;
	import org.apache.tools.ant.taskdefs.Mkdir;
	import org.apache.tools.ant.types.PatternSet;

	public final class LocFiles extends Task {
	private String patternset;
	public void setPatternSet(String s) {
	patternset = s;
	}

	private String cluster;
	public void setCluster(String cluster) {
	this.cluster = cluster;
	}

	private String locales;
	public void setLocales(String locales) {
	this.locales = locales;
	}

	private File srcDir;

	public void setSrc(File f) {
	srcDir = f;
	}
	File distDir = null;

	public void setDestDir(File d) {
	distDir = d;
	}
	String cnbDashes;

	public void setCodeNameBase(String d) {
	assert d.indexOf('-') == -1;
	cnbDashes = d.replace('.', '-');
	}
	File nbmsLocation = null;

	public void setNBMs(File f) {
	nbmsLocation = f;
	}

	private String baseFiles;
	public void setBaseFilesRef(String refid) {
	baseFiles = refid;
	}

	@Override
	public void execute() throws BuildException {
	List<String> includes = new ArrayList<>();
	if (locales != null && !locales.isEmpty()) {
	if (!srcDir.exists()) {
	log("No l10n files present. Do hg clone http://hg.netbeans.org/main/l10n!", Project.MSG_VERBOSE);
	return;
	}

	StringTokenizer tok = new StringTokenizer(locales, ",");
	while (tok.hasMoreElements()) {
	String locale = tok.nextToken();
	if (locale.equals("default")) { // NOI18N
	locale = Locale.getDefault().toString();
	}
	for (String l = locale; l != null; l = trailingUnderscore(l)) {
	processLocale(l, includes, l.equals(locale));
	}
	}
	}

	if (patternset != null) {
	PatternSet ps = new PatternSet();
	ps.setProject(getProject());
	if (includes.isEmpty()) {
	ps.createInclude().setName("I/dont/exist/at/all");
	} else {
	for (String s : includes) {
	ps.createInclude().setName(s);
	}
	}
	getProject().addReference(patternset, ps);
	}

	}

	private static String trailingUnderscore(String s) {
	int under = s.lastIndexOf('_');
	if (under == -1) {
	return null;
	}
	return s.substring(0, under);
	}

	private static String findModuleDir(String cnbDashes) {
	String pref = "org-netbeans-modules-";
	if (cnbDashes.startsWith(pref)) {
	return cnbDashes.substring(pref.length());
	}
	return cnbDashes;
	}

	private void processLocale(String locale, Collection<? super String> toAdd, boolean warn) {
	File baseSrcDir = fileFrom(srcDir, locale);
	if (!baseSrcDir.exists()) {
	if (warn) {
	log("No files for locale: " + locale);
	}
	return;
	}
	log("Found L10N dir " + baseSrcDir, Project.MSG_VERBOSE);

	final String moduleDir = findModuleDir(cnbDashes);
	File locBaseDir = fileFrom(baseSrcDir, cluster, moduleDir);
	if (!locBaseDir.exists()) {
	log("Can't find directory " + locBaseDir, Project.MSG_WARN);
	return;
	}
	File[] ch = locBaseDir.listFiles();
	if (ch == null) {
	throw new BuildException("Surprising content of " + locBaseDir);
	}
	for (File f : ch) {
	processLocaleJar(f, locale, toAdd, moduleDir, locBaseDir, baseSrcDir);
	}
	}

	private void processLocaleJar(
	File file, String locale, Collection<? super String> toAdd,
	String moduleDir, File locBaseDir, File baseSrcDir
	) {
	DirectoryScanner ds = new DirectoryScanner();
	int segments = -1;
	int dirIndex = -1;
	final File root;
	final String prefixRoot;
	final int remove;
	if (file.getName().equals(moduleDir)) {
	ds.setIncludes(new String[] { moduleDir });
	root = fileFrom(distDir, cluster);
	prefixRoot = null;
	remove = 0;
	} else {
	if (file.getName().equals("netbeans")) {
	ds.setIncludes(new String[] {
	"netbeans//org", "netbeans//com"
	});
	remove = 4;
	segments = 3;
	dirIndex = 1;
	root = fileFrom(distDir, cluster);
	prefixRoot = "";
	} else {
	assert file.getName().equals("ext") \|\|
	file.getName().equals("locale");

	ds.setIncludes(new String[] { file.getName() + "/*" });
	segments = 2;
	dirIndex = 0;
	root = fileFrom(distDir, cluster, "modules");
	prefixRoot = "modules/";
	remove = 0;
	}
	}

	ds.setBasedir(locBaseDir);
	ds.scan();
	for (String dir : ds.getIncludedDirectories()) {
	if (remove > 0) {
	dir = dir.substring(0, dir.length() - remove);
	}

	Jar jar = new Jar();
	jar.setProject(getProject());
	Mkdir mkdir = new Mkdir();
	mkdir.setProject(getProject());

	String jarFileName;
	String subPath = "";
	File jarDir;
	String prefixDir;
	if (dir.equals(moduleDir)) {
	jarDir = fileFrom(root, "modules", "locale");
	prefixDir = "modules/locale/";
	jar.setBasedir(fileFrom(locBaseDir, dir));
	jarFileName = cnbDashes + "_" + locale + ".jar";
	} else {
	// netbeans// case or ext/* case
	jar.setBasedir(fileFrom(locBaseDir, dir));
	String[] arr = dir.split("/");
	assert arr.length >= segments : "Expected segments: " + dir;
	final int jarIndex = arr.length - 1;
	jarDir = fileFromIf(
	fileFrom(root, arr, dirIndex, jarIndex, true),
	"locale"
	);
	prefixDir = nameFromIf(
	nameFrom(prefixRoot, arr, dirIndex, jarIndex, true),
	"locale"
	);
	jarFileName = fixJarName(prefixDir, arr[jarIndex]) + "_" + locale + ".jar";
	}
	String fullName = prefixDir + jarFileName;
	toAdd.add(fullName);
	if (jarDir == null) {
	// in patternSet only mode
	continue;
	}

	/*
	String subPath = dir.substring((cluster + File.separator + nbm + File.separator).length() - 1, dir.lastIndexOf(File.separator));
	if (!subPath.startsWith(File.separator + "netbeans")) {
	subPath = File.separator + "modules" + subPath;
	if (!name.startsWith("org-") && !(subPath.endsWith(File.separator + "ext") \|\| subPath.endsWith(File.separator + "ext" + File.separator + "locale"))) {
	name = "org-netbeans-modules-" + name;
	} else {
	// Handle exception from ext/
	if (name.startsWith("web-httpmonitor") \|\| name.startsWith("deployment-deviceanywhere")) {
	name = "org-netbeans-modules-" + name;
	}
	}
	} else {
	subPath = subPath.substring((File.separator + "netbeans").length());
	//Handle exceptions form nblib
	if (name.startsWith("j2ee-ant") \|\| name.startsWith("deployment-deviceanywhere") \|\| name.startsWith("mobility-project") \|\| name.startsWith("java-j2seproject-copylibstask")) {
	name = "org-netbeans-modules-" + name;
	}
	}
	nbm = nbm.replaceAll("^vw-rh", "visualweb-ravehelp-rave_nbpack");
	nbm = nbm.replaceAll("^vw-", "visualweb-");
	if (!nbm.startsWith("org-") && !nbm.startsWith("com-")) {
	nbm = "org-netbeans-modules-" + nbm;
	}
	*/

	if (subPath.matches(".*/docs$")) {
	ds.setBasedir(fileFrom(baseSrcDir, dir));
	ds.setIncludes(new String[]{"*/.hs"});
	ds.setExcludes(new String[]{""});
	ds.scan();
	if (ds.getIncludedFilesCount() != 1) {
	throw new BuildException("Can't find .hs file for " + cnbDashes + " module.");
	}
	File hsFile = fileFrom(baseSrcDir, dir, ds.getIncludedFiles()[0]);
	File baseJHDir = hsFile.getParentFile();

	Task locJH = getProject().createTask("locjhindexer");
	try {
	System.out.println("Basedir: " + baseJHDir.getAbsolutePath());
	locJH.getClass().getMethod("setBasedir", File.class).invoke(locJH, baseJHDir);
	locJH.getClass().getMethod("setLocales", String.class).invoke(locJH, locale);
	locJH.getClass().getMethod("setDbdir", String.class).invoke(locJH, "JavaHelpSearch");
	// ((Path)locJH.getClass().getMethod("createClasspath").invoke(locJH)).add(classpath);

	} catch (Exception ex) {
	throw new BuildException("Can't run locJHInxeder", ex);
	}
	locJH.execute();
	}
	mkdir.setDir(jarDir);
	mkdir.execute();
	jar.setDestFile(fileFrom(jarDir, jarFileName));
	jar.execute();
	}
	}

	private static File fileFrom(File base, String... paths) {
	return fileFrom(base, paths, 0, paths.length, true);
	}
	private static File fileFrom(File base, String[] paths, int from, int upTo, boolean addIfPresent) {
	if (base == null) {
	return null;
	}
	File f = base;
	while (from < upTo) {
	if (addIfPresent \|\| !f.getName().equals(paths[from])) {
	f = new File(f, paths[from]);
	}
	from++;
	}
	return f;
	}
	private static String nameFrom(String base, String... paths) {
	return nameFrom(base, paths, 0, paths.length, true);
	}
	private static String nameFrom(String base, String[] paths, int from, int upTo, boolean addIfPresent) {
	if (base == null) {
	return null;
	}
	String f = base;
	while (from < upTo) {
	if (addIfPresent \|\| !f.endsWith(paths[from] + "/")) {
	f += paths[from];
	f += '/';
	}
	from++;
	}
	return f;
	}

	private static File fileFromIf(File fileFrom, String... add) {
	return fileFrom(fileFrom, add, 0, add.length, false);
	}
	private static String nameFromIf(String fileFrom, String... add) {
	return nameFrom(fileFrom, add, 0, add.length, false);
	}

	private String fixJarName(String prefixDir, String name) {
	if (prefixDir.endsWith("locale/") && baseFiles != null) {
	prefixDir = prefixDir.substring(0, prefixDir.length() - 7);
	PatternSet bf = (PatternSet)getProject().getReference(baseFiles);
	for (String p : bf.getIncludePatterns(getProject())) {
	if (p.startsWith(prefixDir)) {
	String realName = p.substring(prefixDir.length());
	if (realName.indexOf('/') != -1 \|\| realName.indexOf('\\') != -1) {
	continue;
	}
	int indx = realName.indexOf(name);
	if (indx == -1) {
	continue;
	}
	return realName.substring(0, indx + name.length());
	}
	}
	}


	if (name.equals("autoupdate-ui_nb")) {
	return "org-netbeans-modules-autoupdate-ui_nb";
	}
	if (name.equals("options-api_nb")) {
	return "org-netbeans-modules-options-api_nb";
	}
	if (name.equals("uihandler_nb")) {
	return "org-netbeans-modules-uihandler_nb";
	}
	return name;
	}
	}