htmlpage/src/main/java/org/apidesign/bck2brwsr/htmlpage/PageProcessor.java
changeset 28 81ad7a739fed
parent 26 03e4aaa4ef3d
child 100 029e6eed60e9
     1.1 --- a/htmlpage/src/main/java/org/apidesign/bck2brwsr/htmlpage/PageProcessor.java	Mon Sep 24 15:06:43 2012 +0200
     1.2 +++ b/htmlpage/src/main/java/org/apidesign/bck2brwsr/htmlpage/PageProcessor.java	Tue Sep 25 09:55:34 2012 +0200
     1.3 @@ -28,11 +28,15 @@
     1.4  import javax.annotation.processing.RoundEnvironment;
     1.5  import javax.annotation.processing.SupportedAnnotationTypes;
     1.6  import javax.lang.model.element.Element;
     1.7 +import javax.lang.model.element.ElementKind;
     1.8 +import javax.lang.model.element.ExecutableElement;
     1.9 +import javax.lang.model.element.Modifier;
    1.10  import javax.lang.model.element.PackageElement;
    1.11  import javax.lang.model.element.TypeElement;
    1.12  import javax.tools.Diagnostic;
    1.13  import javax.tools.FileObject;
    1.14  import javax.tools.StandardLocation;
    1.15 +import org.apidesign.bck2brwsr.htmlpage.api.OnClick;
    1.16  import org.apidesign.bck2brwsr.htmlpage.api.Page;
    1.17  import org.openide.util.lookup.ServiceProvider;
    1.18  
    1.19 @@ -42,7 +46,10 @@
    1.20   * @author Jaroslav Tulach <jtulach@netbeans.org>
    1.21   */
    1.22  @ServiceProvider(service=Processor.class)
    1.23 -@SupportedAnnotationTypes("org.apidesign.bck2brwsr.htmlpage.api.Page")
    1.24 +@SupportedAnnotationTypes({
    1.25 +    "org.apidesign.bck2brwsr.htmlpage.api.Page",
    1.26 +    "org.apidesign.bck2brwsr.htmlpage.api.OnClick"
    1.27 +})
    1.28  public final class PageProcessor extends AbstractProcessor {
    1.29      @Override
    1.30      public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
    1.31 @@ -64,18 +71,26 @@
    1.32              try {
    1.33                  FileObject java = processingEnv.getFiler().createSourceFile(pkg + '.' + p.name(), e);
    1.34                  w = new OutputStreamWriter(java.openOutputStream());
    1.35 -                w.append("package " + pkg + ";\n");
    1.36 -                w.append("import org.apidesign.bck2brwsr.htmlpage.api.*;\n");
    1.37 -                w.append("class ").append(p.name()).append(" {\n");
    1.38 -                for (String id : pp.ids()) {
    1.39 -                    String tag = pp.tagNameForId(id);
    1.40 -                    String type = type(tag);
    1.41 -                    w.append("  ").append("public static final ").
    1.42 -                        append(type).append(' ').append(cnstnt(id)).append(" = new ").
    1.43 -                        append(type).append("(\"").append(id).append("\");\n");
    1.44 +                try {
    1.45 +                    w.append("package " + pkg + ";\n");
    1.46 +                    w.append("import org.apidesign.bck2brwsr.htmlpage.api.*;\n");
    1.47 +                    w.append("class ").append(p.name()).append(" {\n");
    1.48 +                    for (String id : pp.ids()) {
    1.49 +                        String tag = pp.tagNameForId(id);
    1.50 +                        String type = type(tag);
    1.51 +                        w.append("  ").append("public static final ").
    1.52 +                            append(type).append(' ').append(cnstnt(id)).append(" = new ").
    1.53 +                            append(type).append("(\"").append(id).append("\");\n");
    1.54 +                    }
    1.55 +                    w.append("  static {\n");
    1.56 +                    if (!initializeOnClick(pe, w, pp)) {
    1.57 +                        return false;
    1.58 +                    }
    1.59 +                    w.append("  }\n");
    1.60 +                    w.append("}\n");
    1.61 +                } finally {
    1.62 +                    w.close();
    1.63                  }
    1.64 -                w.append("}");
    1.65 -                w.close();
    1.66              } catch (IOException ex) {
    1.67                  processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "Can't create " + p.name() + ".java", e);
    1.68                  return false;
    1.69 @@ -110,4 +125,41 @@
    1.70      private static String cnstnt(String id) {
    1.71          return id.toUpperCase(Locale.ENGLISH).replace('.', '_');
    1.72      }
    1.73 +
    1.74 +    private boolean initializeOnClick(PackageElement pe, Writer w, ProcessPage pp) throws IOException {
    1.75 +        for (Element clazz : pe.getEnclosedElements()) {
    1.76 +            if (clazz.getKind() != ElementKind.CLASS) {
    1.77 +                continue;
    1.78 +            }
    1.79 +            TypeElement type = (TypeElement)clazz;
    1.80 +            for (Element method : clazz.getEnclosedElements()) {
    1.81 +                OnClick oc = method.getAnnotation(OnClick.class);
    1.82 +                if (oc != null) {
    1.83 +                    if (pp.tagNameForId(oc.id()) == null) {
    1.84 +                        processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "id = " + oc.id() + " does not exist in the HTML page. Found only " + pp.ids(), method);
    1.85 +                        return false;
    1.86 +                    }
    1.87 +                    ExecutableElement ee = (ExecutableElement)method;
    1.88 +                    if (!ee.getParameters().isEmpty()) {
    1.89 +                        processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "@OnClose method can't take arguments", ee);
    1.90 +                        return false;
    1.91 +                    }
    1.92 +                    if (!ee.getModifiers().contains(Modifier.STATIC)) {
    1.93 +                        processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "@OnClose method has to be static", ee);
    1.94 +                        return false;
    1.95 +                    }
    1.96 +                    if (ee.getModifiers().contains(Modifier.PRIVATE)) {
    1.97 +                        processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, "@OnClose method can't be private", ee);
    1.98 +                        return false;
    1.99 +                    }
   1.100 +                    w.append("  ").append(cnstnt(oc.id())).
   1.101 +                        append(".addOnClick(new Runnable() { public void run() {\n");
   1.102 +                    w.append("    ").append(type.getSimpleName().toString()).
   1.103 +                        append('.').append(ee.getSimpleName()).append("();\n");
   1.104 +                    w.append("  }});\n");
   1.105 +                }
   1.106 +            }
   1.107 +        }
   1.108 +        return true;
   1.109 +    }
   1.110  }