ClassFinder.java
上传用户:gwt600
上传日期:2021-06-03
资源大小:704k
文件大小:4k
源码类别:

游戏

开发平台:

Java

  1. /*
  2. This file is part of the OdinMS Maple Story Server
  3.     Copyright (C) 2008 Patrick Huy <patrick.huy@frz.cc> 
  4.     This program is free software: you can redistribute it and/or modify
  5.     it under the terms of the GNU Affero General Public License version 3
  6.     as published by the Free Software Foundation. You may not use, modify
  7.     or distribute this program under any other version of the
  8.     GNU Affero General Public License.
  9.     This program is distributed in the hope that it will be useful,
  10.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.     GNU Affero General Public License for more details.
  13.     You should have received a copy of the GNU Affero General Public License
  14.     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15.  */
  16. package net.sf.odinms.tools;
  17. import java.io.File;
  18. import java.io.IOException;
  19. import java.util.Enumeration;
  20. import java.util.LinkedList;
  21. import java.util.List;
  22. import java.util.jar.JarEntry;
  23. import java.util.jar.JarFile;
  24. public class ClassFinder {
  25. private static org.slf4j.Logger log = org.slf4j.LoggerFactory.getLogger(ClassFinder.class);
  26. List<JarFile> jars = new LinkedList<JarFile>();
  27. List<File> dirs = new LinkedList<File>();
  28. public ClassFinder() {
  29. String classpath = System.getProperty("java.class.path");
  30. String[] splittedPath = classpath.split(File.pathSeparator);
  31. for (String cpe : splittedPath) {
  32. File cpeFile = new File(cpe);
  33. if (cpeFile.isDirectory()) {
  34. dirs.add(cpeFile);
  35. } else {
  36. try {
  37. jars.add(new JarFile(cpeFile));
  38. } catch (IOException e) {
  39. log.error("ERROR", e);
  40. }
  41. }
  42. }
  43. }
  44. private void addClassesInFolder(List<String> classes, File folder, String packageName, boolean recurse) {
  45. for (File f : folder.listFiles()) {
  46. if (!f.isDirectory()) {
  47. if (f.getName().endsWith(".class")) {
  48. classes.add(packageName + "." + f.getName().substring(0, f.getName().length() - 6));
  49. }
  50. } else if (f.isDirectory() && recurse) {
  51. addClassesInFolder(classes, f, packageName + "." + f.getName(), recurse);
  52. }
  53. }
  54. }
  55. public String[] listClasses(String packageName, boolean recurse) {
  56. List<String> ret = new LinkedList<String>();
  57. // scan dirs
  58. final String fileSystemPackagePath = packageName.replace('.', File.separatorChar);
  59. for (File dir : dirs) {
  60. File subfolder = new File(dir, fileSystemPackagePath);
  61. if (subfolder.exists() && subfolder.isDirectory()) {
  62. addClassesInFolder(ret, subfolder, packageName, recurse);
  63. }
  64. }
  65. // scan jars
  66. final String jarPackagePath = packageName.replace('.', '/');
  67. for (JarFile jar : jars) {
  68.         for (Enumeration<JarEntry> entries = jar.entries(); entries.hasMoreElements();) {
  69.             // Get the entry name
  70.             String entryName = (entries.nextElement()).getName();
  71.             if (entryName.endsWith(".class") && entryName.startsWith(jarPackagePath)) {
  72.              int lastSlash = entryName.lastIndexOf('/');
  73.              if (lastSlash <= jarPackagePath.length() || recurse) {
  74. String path = entryName.substring(0, lastSlash);
  75. String className = entryName.substring(lastSlash + 1, entryName.length() - 6);
  76. ret.add(path.replace('/', '.') + "." + className);
  77. }
  78.             }
  79.         }
  80. }
  81. return ret.toArray(new String[ret.size()]);
  82. }
  83. public void dispose() {
  84. for (JarFile jar : jars) {
  85. try {
  86. jar.close();
  87. } catch (IOException e) {
  88. log.error("THROW", e);
  89. }
  90. }
  91. }
  92. }