truffle: mxtool/mx.py comparison

comparison mxtool/mx.py @ 5455:271e83dcfcef

added findclass command to mx tool

author	Doug Simon <doug.simon@oracle.com>
date	Wed, 30 May 2012 18:07:38 +0200
parents	bfc9aeb1264c
children	9911227e6dd3

comparison

equal deleted inserted replaced

-:3d2ad5c8db10
+:271e83dcfcef
 by extension commands.
 Property values can use environment variables with Bash syntax (e.g. ${HOME}).
 """
-import sys, os, errno, time, subprocess, shlex, types, urllib2, contextlib, StringIO, zipfile, signal, xml.sax.saxutils
+import sys, os, errno, time, subprocess, shlex, types, urllib2, contextlib, StringIO, zipfile, signal, xml.sax.saxutils, tempfile
 import shutil, fnmatch, re, xml.dom.minidom
 from collections import Callable
 from threading import Thread
 from argparse import ArgumentParser, REMAINDER
 from os.path import join, basename, dirname, exists, getmtime, isabs, expandvars, isdir, isfile
 _libs[l.name] = l
 class XMLElement(xml.dom.minidom.Element):
 def writexml(self, writer, indent="", addindent="", newl=""):
 writer.write(indent+"<" + self.tagName)
 attrs = self._get_attributes()
 a_names = attrs.keys()
 a_names.sort()
 for a_name in a_names:
 writer.write(" %s=\"" % a_name)
 xml.dom.minidom._write_data(writer, attrs[a_name].value)
 writer.write("\"")
 if self.childNodes:
 if not self.ownerDocument.padTextNodeWithoutSiblings and len(self.childNodes) == 1 and isinstance(self.childNodes[0], xml.dom.minidom.Text):
 # if the only child of an Element node is a Text node, then the
 # text is printed without any indentation or new line padding
 writer.write(">")
 self.childNodes[0].writexml(writer)
 writer.write("</%s>%s" % (self.tagName,newl))
 else:
 writer.write(">%s"%(newl))
 def createElement(self, tagName):
 # overwritten to create XMLElement
 e = XMLElement(tagName)
 e.ownerDocument = self
 return e
 def open(self, tag, attributes={}, data=None):
 element = self.createElement(tag)
 for key, value in attributes.items():
 element.setAttribute(key, value)
 self.current.appendChild(element)
 element.appendChild(self.createTextNode(data))
 return self
 def close(self, tag):
 assert self.current != self
 assert tag == self.current.tagName, str(tag) + ' != ' + self.current.tagName
 self.current = self.current.parentNode
 return self
 def element(self, tag, attributes={}, data=None):
 return self.open(tag, attributes, data).close(tag)
 def xml(self, indent='', newl='', escape=False):
 assert self.current == self
 result = self.toprettyxml(indent, newl, encoding="UTF-8")
 if escape:
 entities = { '"':  "&quot;", "'":  "&apos;", '\n': '&#10;' }
 d.append_to_classpath(cp, resolve)
 if _opts.cp_suffix is not None:
 cp += [_opts.cp_suffix]
 return os.pathsep.join(cp)
-def classpath(names=None, resolve=True, includeSelf=True):
+def classpath(names=None, resolve=True, includeSelf=True, includeBootClasspath=False):
 """
 Get the class path for a list of given projects, resolving each entry in the
 path (e.g. downloading a missing library) if 'resolve' is true.
 """
 if names is None:
-return _as_classpath(sorted_deps(includeLibs=True), resolve)
+result = _as_classpath(sorted_deps(includeLibs=True), resolve)
-deps = []
-if isinstance(names, types.StringTypes):
-project(names).all_deps(deps, True, includeSelf)
 else:
-for n in names:
+deps = []
-project(n).all_deps(deps, True, includeSelf)
+if isinstance(names, types.StringTypes):
-return _as_classpath(deps, resolve)
+project(names).all_deps(deps, True, includeSelf)
+else:
+for n in names:
+project(n).all_deps(deps, True, includeSelf)
+result = _as_classpath(deps, resolve)
+if includeBootClasspath:
+result = os.pathsep.join([java().bootclasspath(), result])
+return result
+def classpath_walk(names=None, resolve=True, includeSelf=True, includeBootClasspath=False):
+"""
+Walks the resources available in a given classpath, yielding a tuple for each resource
+where the first member of the tuple is a directory path or ZipFile object for a
+classpath entry and the second member is the qualified path of the resource relative
+to the classpath entry.
+"""
+cp = classpath(names, resolve, includeSelf, includeBootClasspath)
+for entry in cp.split(os.pathsep):
+if not exists(entry):
+continue
+if isdir(entry):
+for root, dirs, files in os.walk(entry):
+for d in dirs:
+entryPath = join(root[len(entry) + 1:], d)
+yield entry, entryPath
+for f in files:
+entryPath = join(root[len(entry) + 1:], f)
+yield entry, entryPath
+elif entry.endswith('.jar') or entry.endswith('.zip'):
+with zipfile.ZipFile(entry, 'r') as zf:
+for zi in zf.infolist():
+entryPath = zi.filename
+yield zf, entryPath
 def sorted_deps(projectNames=None, includeLibs=False):
 """
 Gets projects and libraries sorted such that dependencies
 are before the projects that depend on them. Unless 'includeLibs' is
 self.debug_port = opts.java_dbg_port
 self.java =  exe_suffix(join(self.jdk, 'bin', 'java'))
 self.javac = exe_suffix(join(self.jdk, 'bin', 'javac'))
 self.javap = exe_suffix(join(self.jdk, 'bin', 'javap'))
 self.javadoc = exe_suffix(join(self.jdk, 'bin', 'javadoc'))
+self._bootclasspath = None
 if not exists(self.java):
 abort('Java launcher derived from JAVA_HOME does not exist: ' + self.java)
 def delAtAndSplit(s):
 if self.debug_port is not None:
 self.java_args += ['-Xdebug', '-Xrunjdwp:transport=dt_socket,server=y,suspend=y,address=' + str(self.debug_port)]
 def format_cmd(self, args):
 return [self.java] + self.java_args_pfx + self.java_args + self.java_args_sfx + args
+def bootclasspath(self):
+if self._bootclasspath is None:
+tmpDir = tempfile.mkdtemp()
+try:
+src = join(tmpDir, 'bootclasspath.java')
+with open(src, 'w') as fp:
+print >> fp, """
+public class bootclasspath {
+public static void main(String[] args) {
+String s = System.getProperty("sun.boot.class.path");
+if (s != null) {
+System.out.println(s);
+}
+}
+}"""
+subprocess.check_call([self.javac, '-d', tmpDir, src])
+self._bootclasspath = subprocess.check_output([self.java, '-cp', tmpDir, 'bootclasspath'])
+finally:
+shutil.rmtree(tmpDir)
+return self._bootclasspath
 def check_get_env(key):
 """
 Gets an environment variable, aborting with a useful message if it is not set.
 """
 parser = ArgumentParser(prog='mx build')
 javaCompliance = java().javaCompliance
 defaultEcjPath = join(_mainSuite.dir, 'mx', 'ecj.jar')
 parser = parser if parser is not None else ArgumentParser(prog='mx build')
 parser.add_argument('-f', action='store_true', dest='force', help='force build (disables timestamp checking)')
 parser.add_argument('-c', action='store_true', dest='clean', help='removes existing build output')
 parser.add_argument('--source', dest='compliance', help='Java compliance level', default=str(javaCompliance))
 parser.add_argument('--Wapi', action='store_true', dest='warnAPI', help='show warnings about using internal APIs')
 print 'node [shape=rect];'
 for p in projects():
 for dep in p.canonical_deps():
 print '"' + p.name + '"->"' + dep + '"'
 print '}'
 def _source_locator_memento(deps):
 slm = XMLDoc()
 slm.open('sourceLookupDirector')
 slm.open('sourceContainers', {'duplicates' : 'false'})
 # Every Java program depends on the JRE
 memento = XMLDoc().element('classpathContainer', {'path' : 'org.eclipse.jdt.launching.JRE_CONTAINER'}).xml()
 slm.element('classpathContainer', {'memento' : memento, 'typeId':'org.eclipse.jdt.launching.sourceContainer.classpathContainer'})
 for dep in deps:
 if dep.isLibrary():
 if hasattr(dep, 'eclipse.container'):
 memento = XMLDoc().element('classpathContainer', {'path' : getattr(dep, 'eclipse.container')}).xml()
 slm.element('classpathContainer', {'memento' : memento, 'typeId':'org.eclipse.jdt.launching.sourceContainer.classpathContainer'})
 launch.close('mapAttribute')
 launch.element('stringAttribute', {'key' : 'org.eclipse.jdt.launching.PROJECT_ATTR', 'value' : ''})
 launch.element('stringAttribute', {'key' : 'org.eclipse.jdt.launching.VM_CONNECTOR_ID', 'value' : 'org.eclipse.jdt.launching.socketAttachConnector'})
 launch.close('launchConfiguration')
 launch = launch.xml(newl='\n') % slm.xml(escape=True)
 if name is None:
 name = 'attach-' + hostname + '-' + port
 eclipseLaunches = join('mx', 'eclipse-launches')
 if not exists(eclipseLaunches):
 os.makedirs(eclipseLaunches)
 vmArgs.append(a)
 else:
 mainClass = a
 appArgs = list(reversed(argsCopy))
 break
 if mainClass is None:
 log('Cannot create Eclipse launch configuration without main class or jar file: java ' + ' '.join(javaArgs))
 return False
 if name is None:
 if mainClass == '-jar':
 name = basename(appArgs[0])
 if len(appArgs) > 1 and not appArgs[1].startswith('-'):
 name = name + '_' + appArgs[1]
 if cp is not None:
 for e in cp.split(os.pathsep):
 for s in suites():
 deps += [p for p in s.projects if e == p.output_dir()]
 deps += [l for l in s.libs if e == l.get_path(False)]
 slm = _source_locator_memento(deps)
 launch = XMLDoc()
 launch.open('launchConfiguration', {'type' : 'org.eclipse.jdt.launching.localJavaApplication'})
 launch.element('stringAttribute', {'key' : 'org.eclipse.debug.core.source_locator_id', 'value' : 'org.eclipse.jdt.launching.sourceLocator.JavaSourceLookupDirector'})
 launch.element('stringAttribute', {'key' : 'org.eclipse.debug.core.source_locator_memento', 'value' : '%s'})
 launch.element('stringAttribute', {'key' : 'org.eclipse.jdt.launching.JRE_CONTAINER', 'value' : 'org.eclipse.jdt.launching.JRE_CONTAINER/org.eclipse.jdt.internal.debug.ui.launcher.StandardVMType/' + jre})
 suite = _mainSuite
 for p in projects():
 if p.native:
 continue
 if not exists(p.dir):
 os.makedirs(p.dir)
 out = XMLDoc()
 out.open('classpath')
 for src in p.srcDirs:
 srcDir = join(p.dir, src)
 if not exists(srcDir):
 os.mkdir(srcDir)
 out.element('classpathentry', {'kind' : 'src', 'path' : src})
 with open(join(eclipseSettingsDir, name)) as f:
 content = f.read()
 content = content.replace('${javaCompliance}', str(p.javaCompliance))
 update_file(join(settingsDir, name), content)
 make_eclipse_attach('localhost', '8000', deps=projects())
 def netbeansinit(args, suite=None):
 """(re)generate NetBeans project configurations"""
 if suite is None:
 out.close('source-roots')
 out.open('test-roots')
 out.element('root', {'id' : 'test.src.dir'})
 out.close('test-roots')
 out.close('data')
 firstDep = True
 for dep in p.all_deps([], True):
 if dep == p:
 continue;
 eclipseinit(args, suite)
 netbeansinit(args, suite)
 def javadoc(args):
 """generate javadoc for some/all Java projects"""
 parser = ArgumentParser(prog='mx javadoc')
 parser.add_argument('--unified', action='store_true', help='put javadoc in a single directory instead of one per project')
 parser.add_argument('--force', action='store_true', help='(re)generate javadoc even if package-list file exists')
 parser.add_argument('--projects', action='store', help='comma separated projects to process (omit to process all projects)')
 parser.add_argument('--argfile', action='store', help='name of file containing extra javadoc options')
 sp = os.pathsep.join(sp)
 log('Generating {2} for {0} in {1}'.format(', '.join(names), out, docDir))
 run([java().javadoc, memory, '-classpath', cp, '-quiet', '-d', out, '-sourcepath', sp] + docletArgs + links + extraArgs + list(pkgs))
 log('Generated {2} for {0} in {1}'.format(', '.join(names), out, docDir))
+def findclass(args):
+"""find all classes matching a given substring"""
+for entry, filename in classpath_walk(includeBootClasspath=True):
+if filename.endswith('.class'):
+if isinstance(entry, zipfile.ZipFile):
+classname = filename.replace('/', '.')
+else:
+classname = filename.replace(os.sep, '.')
+classname = classname[:-len('.class')]
+for a in args:
+if a in classname:
+log(classname)
 def javap(args):
 """launch javap with a -classpath option denoting all available classes
 Run the JDK javap class file disassembler with the following prepended options:
 'build': [build, '[options]'],
 'checkstyle': [checkstyle, ''],
 'canonicalizeprojects': [canonicalizeprojects, ''],
 'clean': [clean, ''],
 'eclipseinit': [eclipseinit, ''],
+'findclass': [findclass, ''],
 'help': [help_, '[command]'],
 'ideclean': [ideclean, ''],
 'ideinit': [ideinit, ''],
 'projectgraph': [projectgraph, ''],
 'javap': [javap, ''],

Mercurial > hg > truffle

comparison mxtool/mx.py @ 5455:271e83dcfcef