add info for @OnlyIn side only marker

Author: ZZZank

src/main/java/zzzank/probejs/lang/java/base/AnnotationHolder.java

@@ -8,6 +8,7 @@
 import java.lang.annotation.Annotation;
 import java.util.Arrays;
 import java.util.List;
+import java.util.Optional;
 import java.util.stream.Collectors;
 
 public class AnnotationHolder {
@@ -43,4 +44,8 @@ public <T extends Annotation> T getAnnotation(Class<T> type) {
         }
         return null;
     }
+
+    public <T extends Annotation> Optional<T> getAnnotationOptional(Class<T> type) {
+        return Optional.ofNullable(getAnnotation(type));
+    }
 }

src/main/java/zzzank/probejs/lang/transpiler/transformation/impl/InjectAnnotation.java

@@ -3,6 +3,7 @@
 import dev.latvian.mods.rhino.annotations.typing.JSInfo;
 import dev.latvian.mods.rhino.annotations.typing.JSParams;
 import lombok.val;
+import net.minecraftforge.api.distmarker.OnlyIn;
 import zzzank.probejs.features.rhizo.RhizoState;
 import zzzank.probejs.lang.java.base.AnnotationHolder;
 import zzzank.probejs.lang.java.clazz.Clazz;
@@ -30,17 +31,16 @@ public void transformMethod(Clazz clazz, MethodInfo methodInfo, MethodDecl decl)
         applyInfo(methodInfo, decl);
         applyDeprecated(methodInfo, decl);
         applyJSParams(methodInfo, decl.params);
+        applySideOnly(methodInfo, decl);
 
         if (RhizoState.INFO_ANNOTATION) {
             val paramLines = methodInfo.params.stream()
                 .filter(p -> p.hasAnnotation(JSInfo.class))
                 .map(p -> String.format("@param %s - %s", p.name, p.getAnnotation(JSInfo.class).value()))
-                .collect(Collectors.toList());
-            if (!paramLines.isEmpty()) {
+                .toArray(String[]::new);
+            if (paramLines.length != 0) {
                 decl.linebreak();
-                for (String line : paramLines) {
-                    decl.addComment(line);
-                }
+                decl.addComment(paramLines);
             }
         }
     }
@@ -49,13 +49,15 @@ public void transformMethod(Clazz clazz, MethodInfo methodInfo, MethodDecl decl)
     public void transformField(Clazz clazz, FieldInfo fieldInfo, FieldDecl decl) {
         applyInfo(fieldInfo, decl);
         applyDeprecated(fieldInfo, decl);
+        applySideOnly(fieldInfo, decl);
     }
 
     @Override
     public void transformConstructor(Clazz clazz, ConstructorInfo constructorInfo, ConstructorDecl decl) {
         applyInfo(constructorInfo, decl);
         applyDeprecated(constructorInfo, decl);
         applyJSParams(constructorInfo, decl.params);
+        applySideOnly(constructorInfo, decl);
     }
 
     public void applyDeprecated(AnnotationHolder info, CommentableCode decl) {
@@ -94,4 +96,19 @@ public void applyJSParams(AnnotationHolder parent, Collection<ParamDecl> params)
             }
         }
     }
+
+    public void applySideOnly(AnnotationHolder target, CommentableCode decl) {
+        val dist = target.getAnnotationOptional(OnlyIn.class)
+            .map(OnlyIn::value)
+            .orElse(null);
+        if (dist == null) {
+            return;
+        }
+        decl.linebreak();
+        if (dist.isClient()) {
+            decl.addComment("Client only, do not use in server scripts");
+        } else if (dist.isDedicatedServer()) {
+            decl.addComment("Server only, do not use in client scripts");
+        }
+    }
 }