feat: add ImplicitUsageProvider extension

annotations/src/main/java/net/staticstudios/data/Column.java

@@ -7,6 +7,7 @@
 
 @Retention(RetentionPolicy.RUNTIME)
 @Target(ElementType.FIELD)
+@ImplicitWrite
 public @interface Column {
     String name();
 

annotations/src/main/java/net/staticstudios/data/ForeignColumn.java

@@ -7,6 +7,7 @@
 
 @Retention(RetentionPolicy.RUNTIME)
 @Target(ElementType.FIELD)
+@ImplicitWrite
 public @interface ForeignColumn {
     String name();
 

annotations/src/main/java/net/staticstudios/data/IdColumn.java

@@ -7,6 +7,7 @@
 
 @Retention(RetentionPolicy.RUNTIME)
 @Target(ElementType.FIELD)
+@ImplicitWrite
 public @interface IdColumn {
     String name();
 }

annotations/src/main/java/net/staticstudios/data/Identifier.java

@@ -10,6 +10,7 @@
  */
 @Retention(RetentionPolicy.RUNTIME)
 @Target(ElementType.FIELD)
+@ImplicitWrite
 public @interface Identifier {
     String value();
 

annotations/src/main/java/net/staticstudios/data/ImplicitWrite.java

@@ -0,0 +1,4 @@
+package net.staticstudios.data;
+
+public @interface ImplicitWrite {
+}

annotations/src/main/java/net/staticstudios/data/ManyToMany.java

@@ -8,6 +8,7 @@
 
 @Retention(RetentionPolicy.RUNTIME)
 @Target(ElementType.FIELD)
+@ImplicitWrite
 public @interface ManyToMany {
     /**
      * How should this relation be linked?

annotations/src/main/java/net/staticstudios/data/OneToMany.java

@@ -8,6 +8,7 @@
 
 @Retention(RetentionPolicy.RUNTIME)
 @Target(ElementType.FIELD)
+@ImplicitWrite
 public @interface OneToMany {
     /**
      * How should this relation be linked?

annotations/src/main/java/net/staticstudios/data/OneToOne.java

@@ -8,6 +8,7 @@
 
 @Retention(RetentionPolicy.RUNTIME)
 @Target(ElementType.FIELD)
+@ImplicitWrite
 public @interface OneToOne {
     /**
      * How should this relation be linked?

intellij-plugin/src/main/java/net/staticstudios/data/ide/intellij/DataImplicitUsageProvider.java

@@ -0,0 +1,30 @@
+package net.staticstudios.data.ide.intellij;
+
+import com.intellij.codeInsight.daemon.ImplicitUsageProvider;
+import com.intellij.psi.PsiElement;
+import com.intellij.psi.PsiField;
+import net.staticstudios.data.utils.Constants;
+import org.jetbrains.annotations.NotNull;
+
+public class DataImplicitUsageProvider implements ImplicitUsageProvider {
+
+    @Override
+    public boolean isImplicitUsage(@NotNull PsiElement psiElement) {
+        return false;
+    }
+
+    @Override
+    public boolean isImplicitRead(@NotNull PsiElement psiElement) {
+        return false;
+    }
+
+    @Override
+    public boolean isImplicitWrite(@NotNull PsiElement psiElement) {
+        if (!(psiElement instanceof PsiField field)) {
+            return false;
+        }
+
+        return IntelliJPluginUtils.hasAnnotationRecursive(field, Constants.IMPLICIT_WRITE_FQN);
+    }
+
+}

intellij-plugin/src/main/java/net/staticstudios/data/ide/intellij/IntelliJPluginUtils.java

@@ -1,8 +1,11 @@
 package net.staticstudios.data.ide.intellij;
 
+import com.intellij.lang.jvm.annotation.JvmAnnotationAttribute;
 import com.intellij.psi.*;
 import net.staticstudios.data.utils.Constants;
 
+import java.util.Arrays;
+
 public class IntelliJPluginUtils {
     public static boolean genericTypeIs(PsiType type, String classFqn) {
         if (!(type instanceof PsiClassType psiClassType)) {
@@ -43,9 +46,39 @@ public static boolean hasAnnotation(PsiModifierListOwner element, String annotat
         if (element.getModifierList() == null) {
             return false;
         }
+
         return element.getModifierList().findAnnotation(annotationFqn) != null;
     }
 
+    public static boolean hasAnnotationRecursive(PsiModifierListOwner element, String annotationFqn) {
+        if (element.getModifierList() == null) {
+            return false;
+        }
+
+        if (hasAnnotation(element, annotationFqn)) {
+            return true;
+        }
+
+        for (PsiAnnotation annotation : element.getModifierList().getAnnotations()) {
+            PsiJavaCodeReferenceElement referenceElement = annotation.getNameReferenceElement();
+
+            if (referenceElement == null) {
+                continue;
+            }
+
+            PsiElement resolved = referenceElement.resolve();
+            if (!(resolved instanceof PsiClass psiClass)) {
+                continue;
+            }
+
+            if (hasAnnotationRecursive(psiClass, annotationFqn)) {
+                return true;
+            }
+        }
+
+        return false;
+    }
+
     public static PsiType getGenericParameter(PsiClassType type, PsiManager manager) {
         PsiType[] params = type.getParameters();
         return (params.length > 0) ? params[0] : PsiType.getJavaLangObject(manager, type.getResolveScope());

intellij-plugin/src/main/resources/META-INF/plugin.xml

@@ -14,5 +14,7 @@
         <lang.psiAugmentProvider implementation="net.staticstudios.data.ide.intellij.DataPsiAugmentProvider"/>
 
         <dependencySupport coordinate="net.staticstudios:static-data" kind="java" displayName="Static Data"/>
+
+        <implicitUsageProvider implementation="net.staticstudios.data.ide.intellij.DataImplicitUsageProvider"/>
     </extensions>
 </idea-plugin>

utils/src/main/java/net/staticstudios/data/utils/Constants.java

@@ -20,5 +20,6 @@ public class Constants {
     public static final String INSERT_MODE_FQN = "net.staticstudios.data.InsertMode";
     public static final String INSERT_CONTEXT_FQN = "net.staticstudios.data.insert.InsertContext";
     public static final String BATCH_INSERT_FQN = "net.staticstudios.data.insert.BatchInsert";
+    public static final String IMPLICIT_WRITE_FQN = "net.staticstudios.data.ImplicitWrite";
 
 }