示例#1
0
 public static final void usageAndExit(String[] expectedArgs) {
   String clazz = threadRootClass();
   StringBuilder b = new StringBuilder();
   b.append("USAGE: ").append(clazz).append(" ");
   for (String arg : expectedArgs) {
     b.append(arg).append(" ");
   }
   System.out.println(b.toString());
   System.exit(ExitCode.INVALID_ARGUMENTS.code);
 }
示例#2
0
 public static final void usageAndExit(Map<String, String[]> argToFlagsMap) {
   String clazz = threadRootClass();
   StringBuilder b = new StringBuilder();
   b.append("USAGE: ").append(clazz).append("\n\t");
   for (String arg : argToFlagsMap.keySet()) {
     String[] flags = argToFlagsMap.get(arg);
     if (flags == null || flags.length == 0) {
       throw new IllegalArgumentException("No flags registered for arg: " + arg);
     }
     b.append("{");
     for (int i = 0; i < flags.length - 1; i++) {
       b.append(flags[i]).append(",");
     }
     b.append(flags[flags.length - 1]).append("}");
   }
   System.out.println(b.toString());
   System.exit(ExitCode.INVALID_ARGUMENTS.code);
 }
示例#3
0
  private static final Class<?>[] getVisibleClasses(Map<String, String> options) {
    // --Variables
    List<Class<?>> classes = new ArrayList<Class<?>>();
    // (get classpath)
    String pathSep = System.getProperty("path.separator");
    String[] cp = System.getProperties().getProperty("java.class.path", null).split(pathSep);
    // --Configuration
    ensureScalaPath(options, cp);
    // --Fill Options
    // (get classes)
    for (String entry : cp) {
      // (should skip?)
      if (entry.equals(".") || entry.trim().length() == 0) {
        continue;
      }
      boolean isIgnored = false;
      for (String pattern : ignoredClasspath) {
        if (entry.matches(pattern)) {
          log.debug(LOG_TAG, "Ignoring options in classpath element: " + entry);
          isIgnored = true;
          break;
        }
      }
      if (isIgnored) {
        continue;
      }
      // (no, don't skip)
      File f = new File(entry);
      if (f.isDirectory()) {
        // --Case: Files
        LazyFileIterator iter = new LazyFileIterator(f, ".*class$");
        while (iter.hasNext()) {
          // (get the associated class)
          Class<?> clazz = filePathToClass(entry, iter.next().getPath());
          if (clazz != null) {
            // (add the class if it's valid)
            classes.add(clazz);
          }
        }
      } else if (!isIgnored(entry)) {
        // --Case: Jar
        try {
          JarFile jar = new JarFile(f);
          Enumeration<JarEntry> e = jar.entries();
          while (e.hasMoreElements()) {
            // (for each jar file element)
            JarEntry jarEntry = e.nextElement();
            String clazz = jarEntry.getName();
            if (clazz.matches(".*class$")) {
              // (if it's a class)
              clazz = clazz.substring(0, clazz.length() - 6).replaceAll("/", ".");
              // (add it)
              try {
                classes.add(Class.forName(clazz, false, ClassLoader.getSystemClassLoader()));
              } catch (ClassNotFoundException ex) {
                throw Log.internal("Could not load class in jar: " + f + " at path: " + clazz);
              } catch (NoClassDefFoundError ex) {
                log.debug(LOG_TAG, "Could not scan class: " + clazz + " (in jar: " + f + ")");
              }
            }
          }
        } catch (IOException e) {
          throw log.fail("Could not open jar file: " + f + "(are you sure the file exists?)");
        }
      } else {
        // case: ignored jar
      }
    }

    return classes.toArray(new Class<?>[classes.size()]);
  }
示例#4
0
  @SuppressWarnings("rawtypes")
  protected static final Map<String, Field> fillOptions(
      Class<?>[] classes, Map<String, String> options, boolean ensureAllOptions) {

    // --Get Fillable Options
    Map<String, Field> canFill = new HashMap<String, Field>();
    Map<String, Marker> required = new HashMap<String, Marker>();
    Map<String, String> interner = new HashMap<String, String>();
    for (Class c : classes) {
      Field[] fields = null;
      try {
        fields = c.getDeclaredFields();
      } catch (Throwable e) {
        log.debug(
            LOG_TAG,
            "Could not check fields for class: "
                + c.getName()
                + "  (caused by "
                + e.getClass()
                + ": "
                + e.getMessage()
                + ")");
        continue;
      }

      for (Field f : fields) {
        Option o = f.getAnnotation(Option.class);
        if (o != null) {
          // (check if field is static)
          if ((f.getModifiers() & Modifier.STATIC) == 0) {
            log.err(LOG_TAG, "Option can only be applied to static field: " + c + "." + f);
            System.exit(ExitCode.BAD_OPTION.code);
          }
          // (required marker)
          Marker mark = null;
          if (o.required()) {
            mark = new Marker();
            mark.unset();
          }
          // (add main name)
          String name = o.name().toLowerCase();
          if (name.equals("")) {
            name = f.getName().toLowerCase();
          }
          if (canFill.containsKey(name)) {
            String name1 =
                canFill.get(name).getDeclaringClass().getCanonicalName()
                    + "."
                    + canFill.get(name).getName();
            String name2 = f.getDeclaringClass().getCanonicalName() + "." + f.getName();
            if (!name1.equals(name2)) {
              log.err(
                  LOG_TAG,
                  "Multiple declarations of option " + name + ": " + name1 + " and " + name2);
              System.exit(ExitCode.BAD_OPTION.code);
            } else {
              log.err(
                  LOG_TAG,
                  "Class is in classpath multiple times: "
                      + canFill.get(name).getDeclaringClass().getCanonicalName());
            }
          }
          canFill.put(name, f);
          if (mark != null) required.put(name, mark);
          interner.put(name, name);
          // (add alternate names)
          if (!o.alt().equals("")) {
            for (String alt : o.alt().split(" *, *")) {
              alt = alt.toLowerCase();
              if (canFill.containsKey(alt) && !alt.equals(name))
                throw new IllegalArgumentException(
                    "Multiple declarations of option "
                        + alt
                        + ": "
                        + canFill.get(alt)
                        + " and "
                        + f);
              canFill.put(alt, f);
              if (mark != null) required.put(alt, mark);
              interner.put(alt, name);
            }
          }
        }
      }
    }

    // --Fill Options
    for (String key : options.keySet()) {
      String rawKey = key;
      key = key.toLowerCase();
      // (get values)
      String value = options.get(rawKey);
      Field target = canFill.get(key);
      // (mark required option as fulfilled)
      Marker mark = required.get(key);
      if (mark != null) {
        mark.set();
      }
      // (fill the field)
      if (target != null) {
        // (case: declared option)
        fillField(target, value);
      } else if (ensureAllOptions) {
        // (case: undeclared option)
        // split the key
        int lastDotIndex = rawKey.lastIndexOf('.');
        if (lastDotIndex < 0) {
          log.err(LOG_TAG, "Unrecognized option: " + key);
          System.exit(ExitCode.BAD_OPTION.code);
        }
        String className = rawKey.substring(0, lastDotIndex);
        String fieldName = rawKey.substring(lastDotIndex + 1);
        // get the class
        Class clazz = null;
        try {
          clazz = ClassLoader.getSystemClassLoader().loadClass(className);
        } catch (Exception e) {
          log.err(LOG_TAG, "Could not set option: " + rawKey + "; no such class: " + className);
          System.exit(ExitCode.BAD_OPTION.code);
        }
        // get the field
        try {
          target = clazz.getField(fieldName);
        } catch (Exception e) {
          log.err(
              LOG_TAG,
              "Could not set option: "
                  + rawKey
                  + "; no such field: "
                  + fieldName
                  + " in class: "
                  + className);
          System.exit(ExitCode.BAD_OPTION.code);
        }
        fillField(target, value);
      }
    }

    // --Ensure Required
    boolean good = true;
    for (String key : required.keySet()) {
      if (!required.get(key).isSet()) {
        log.err(
            LOG_TAG,
            "Missing required option: "
                + interner.get(key)
                + "   <in class: "
                + canFill.get(key).getDeclaringClass()
                + ">");
        required.get(key).set(); // don't duplicate error messages
        good = false;
      }
    }
    if (!good) {
      System.exit(ExitCode.BAD_OPTION.code);
    }

    return canFill;
  }
示例#5
0
 private static void fillField(Field f, String value) {
   try {
     // --Permissions
     boolean accessState = true;
     if (Modifier.isFinal(f.getModifiers())) {
       log.err(LOG_TAG, "Option cannot be final: " + f);
       System.exit(ExitCode.BAD_OPTION.code);
     }
     if (!f.isAccessible()) {
       accessState = false;
       f.setAccessible(true);
     }
     // --Set Value
     Object objVal = Utils.cast(value, f.getGenericType());
     if (objVal != null) {
       if (objVal.getClass().isArray()) {
         // (case: array)
         Object[] array = (Object[]) objVal;
         // error check
         if (!f.getType().isArray()) {
           log.err(
               LOG_TAG,
               "Setting an array to a non-array field. field: "
                   + f
                   + " value: "
                   + Arrays.toString(array)
                   + " src: "
                   + value);
           System.exit(ExitCode.BAD_OPTION.code);
         }
         // create specific array
         Object toSet = Array.newInstance(f.getType().getComponentType(), array.length);
         for (int i = 0; i < array.length; i++) {
           Array.set(toSet, i, array[i]);
         }
         // set value
         f.set(null, toSet);
       } else {
         // case: not array
         f.set(null, objVal);
       }
     } else {
       log.err(
           LOG_TAG, "Cannot assign option field: " + f + " value: " + value + "; invalid type");
       System.exit(ExitCode.BAD_OPTION.code);
     }
     // --Permissions
     if (!accessState) {
       f.setAccessible(false);
     }
   } catch (IllegalArgumentException e) {
     log.err(
         LOG_TAG,
         "Cannot assign option field: "
             + f.getDeclaringClass().getCanonicalName()
             + "."
             + f.getName()
             + " value: "
             + value
             + " cause: "
             + e.getMessage());
     System.exit(ExitCode.BAD_OPTION.code);
   } catch (IllegalAccessException e) {
     log.err(
         LOG_TAG,
         "Cannot access option field: "
             + f.getDeclaringClass().getCanonicalName()
             + "."
             + f.getName());
     System.exit(ExitCode.BAD_OPTION.code);
   } catch (Exception e) {
     log.err(
         LOG_TAG,
         "Cannot assign option field: "
             + f.getDeclaringClass().getCanonicalName()
             + "."
             + f.getName()
             + " value: "
             + value
             + " cause: "
             + e.getMessage());
     System.exit(ExitCode.BAD_OPTION.code);
   }
 }