9 |
10 | ## Static members example
11 |
12 | 
13 |
14 | ## Instance members example
15 |
16 | 
--------------------------------------------------------------------------------
/build.gradle:
--------------------------------------------------------------------------------
1 | plugins {
2 | id "org.jetbrains.intellij" version "0.6.1"
3 | }
4 |
5 | group 'com.github.braisdom'
6 | version '1.3.4'
7 |
8 | repositories {
9 | mavenLocal()
10 | maven { url 'http://dl.bintray.com/jetbrains/intellij-plugin-service' }
11 | maven { url 'https://jitpack.io' }
12 | mavenCentral()
13 | }
14 |
15 | dependencies {
16 | testCompile group: 'junit', name: 'junit', version: '4.12'
17 | }
18 |
19 | apply plugin: 'java'
20 | apply plugin: 'idea'
21 | idea {
22 | project {
23 | jdkName = '1.8'
24 | languageLevel = '1.8'
25 | }
26 | }
27 |
28 | apply plugin: 'java'
29 | compileJava {
30 | options.encoding = 'UTF-8'
31 | options.compilerArgs << "-Xlint:deprecation"
32 | }
33 |
34 | apply plugin: 'org.jetbrains.intellij'
35 | intellij {
36 | version = "2020.2"
37 | type = 'IC'
38 | updateSinceUntilBuild = false
39 | downloadSources = true
40 | plugins = ['java']
41 | }
42 |
43 | patchPluginXml {
44 | changeNotes """
45 | 1. Generating setter and getter methods for model who has annotation @DomainModel
46 |2. Generating query and persistence methods for model who has annotation @DomainModel
47 |3. Generating query method for field who has annotation @Queryable
48 |4. Generating relation method for field who has annotation @Relation
49 |...
""" 50 | } -------------------------------------------------------------------------------- /doc/install.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braisdom/ObjectiveSql-IntelliJ-Plugin/1cde1897d21ae9dbcfaa1049707f588f22c3f809/doc/install.png -------------------------------------------------------------------------------- /doc/instance_methods.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braisdom/ObjectiveSql-IntelliJ-Plugin/1cde1897d21ae9dbcfaa1049707f588f22c3f809/doc/instance_methods.png -------------------------------------------------------------------------------- /doc/object-sql-intellij-1.0.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braisdom/ObjectiveSql-IntelliJ-Plugin/1cde1897d21ae9dbcfaa1049707f588f22c3f809/doc/object-sql-intellij-1.0.zip -------------------------------------------------------------------------------- /doc/static_methods.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braisdom/ObjectiveSql-IntelliJ-Plugin/1cde1897d21ae9dbcfaa1049707f588f22c3f809/doc/static_methods.png -------------------------------------------------------------------------------- /gradlew: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env sh 2 | 3 | ############################################################################## 4 | ## 5 | ## Gradle start up script for UN*X 6 | ## 7 | ############################################################################## 8 | 9 | # Attempt to set APP_HOME 10 | # Resolve links: $0 may be a link 11 | PRG="$0" 12 | # Need this for relative symlinks. 13 | while [ -h "$PRG" ] ; do 14 | ls=`ls -ld "$PRG"` 15 | link=`expr "$ls" : '.*-> \(.*\)$'` 16 | if expr "$link" : '/.*' > /dev/null; then 17 | PRG="$link" 18 | else 19 | PRG=`dirname "$PRG"`"/$link" 20 | fi 21 | done 22 | SAVED="`pwd`" 23 | cd "`dirname \"$PRG\"`/" >/dev/null 24 | APP_HOME="`pwd -P`" 25 | cd "$SAVED" >/dev/null 26 | 27 | APP_NAME="Gradle" 28 | APP_BASE_NAME=`basename "$0"` 29 | 30 | # Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. 31 | DEFAULT_JVM_OPTS="" 32 | 33 | # Use the maximum available, or set MAX_FD != -1 to use that value. 34 | MAX_FD="maximum" 35 | 36 | warn () { 37 | echo "$*" 38 | } 39 | 40 | die () { 41 | echo 42 | echo "$*" 43 | echo 44 | exit 1 45 | } 46 | 47 | # OS specific support (must be 'true' or 'false'). 48 | cygwin=false 49 | msys=false 50 | darwin=false 51 | nonstop=false 52 | case "`uname`" in 53 | CYGWIN* ) 54 | cygwin=true 55 | ;; 56 | Darwin* ) 57 | darwin=true 58 | ;; 59 | MINGW* ) 60 | msys=true 61 | ;; 62 | NONSTOP* ) 63 | nonstop=true 64 | ;; 65 | esac 66 | 67 | CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar 68 | 69 | # Determine the Java command to use to start the JVM. 70 | if [ -n "$JAVA_HOME" ] ; then 71 | if [ -x "$JAVA_HOME/jre/sh/java" ] ; then 72 | # IBM's JDK on AIX uses strange locations for the executables 73 | JAVACMD="$JAVA_HOME/jre/sh/java" 74 | else 75 | JAVACMD="$JAVA_HOME/bin/java" 76 | fi 77 | if [ ! -x "$JAVACMD" ] ; then 78 | die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME 79 | 80 | Please set the JAVA_HOME variable in your environment to match the 81 | location of your Java installation." 82 | fi 83 | else 84 | JAVACMD="java" 85 | which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. 86 | 87 | Please set the JAVA_HOME variable in your environment to match the 88 | location of your Java installation." 89 | fi 90 | 91 | # Increase the maximum file descriptors if we can. 92 | if [ "$cygwin" = "false" -a "$darwin" = "false" -a "$nonstop" = "false" ] ; then 93 | MAX_FD_LIMIT=`ulimit -H -n` 94 | if [ $? -eq 0 ] ; then 95 | if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then 96 | MAX_FD="$MAX_FD_LIMIT" 97 | fi 98 | ulimit -n $MAX_FD 99 | if [ $? -ne 0 ] ; then 100 | warn "Could not set maximum file descriptor limit: $MAX_FD" 101 | fi 102 | else 103 | warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT" 104 | fi 105 | fi 106 | 107 | # For Darwin, add options to specify how the application appears in the dock 108 | if $darwin; then 109 | GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\"" 110 | fi 111 | 112 | # For Cygwin, switch paths to Windows format before running java 113 | if $cygwin ; then 114 | APP_HOME=`cygpath --path --mixed "$APP_HOME"` 115 | CLASSPATH=`cygpath --path --mixed "$CLASSPATH"` 116 | JAVACMD=`cygpath --unix "$JAVACMD"` 117 | 118 | # We build the pattern for arguments to be converted via cygpath 119 | ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null` 120 | SEP="" 121 | for dir in $ROOTDIRSRAW ; do 122 | ROOTDIRS="$ROOTDIRS$SEP$dir" 123 | SEP="|" 124 | done 125 | OURCYGPATTERN="(^($ROOTDIRS))" 126 | # Add a user-defined pattern to the cygpath arguments 127 | if [ "$GRADLE_CYGPATTERN" != "" ] ; then 128 | OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)" 129 | fi 130 | # Now convert the arguments - kludge to limit ourselves to /bin/sh 131 | i=0 132 | for arg in "$@" ; do 133 | CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -` 134 | CHECK2=`echo "$arg"|egrep -c "^-"` ### Determine if an option 135 | 136 | if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then ### Added a condition 137 | eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"` 138 | else 139 | eval `echo args$i`="\"$arg\"" 140 | fi 141 | i=$((i+1)) 142 | done 143 | case $i in 144 | (0) set -- ;; 145 | (1) set -- "$args0" ;; 146 | (2) set -- "$args0" "$args1" ;; 147 | (3) set -- "$args0" "$args1" "$args2" ;; 148 | (4) set -- "$args0" "$args1" "$args2" "$args3" ;; 149 | (5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;; 150 | (6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;; 151 | (7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;; 152 | (8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;; 153 | (9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;; 154 | esac 155 | fi 156 | 157 | # Escape application args 158 | save () { 159 | for i do printf %s\\n "$i" | sed "s/'/'\\\\''/g;1s/^/'/;\$s/\$/' \\\\/" ; done 160 | echo " " 161 | } 162 | APP_ARGS=$(save "$@") 163 | 164 | # Collect all arguments for the java command, following the shell quoting and substitution rules 165 | eval set -- $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS "\"-Dorg.gradle.appname=$APP_BASE_NAME\"" -classpath "\"$CLASSPATH\"" org.gradle.wrapper.GradleWrapperMain "$APP_ARGS" 166 | 167 | # by default we should be in the correct project dir, but when run from Finder on Mac, the cwd is wrong 168 | if [ "$(uname)" = "Darwin" ] && [ "$HOME" = "$PWD" ]; then 169 | cd "$(dirname "$0")" 170 | fi 171 | 172 | exec "$JAVACMD" "$@" 173 | -------------------------------------------------------------------------------- /gradlew.bat: -------------------------------------------------------------------------------- 1 | @if "%DEBUG%" == "" @echo off 2 | @rem ########################################################################## 3 | @rem 4 | @rem Gradle startup script for Windows 5 | @rem 6 | @rem ########################################################################## 7 | 8 | @rem Set local scope for the variables with windows NT shell 9 | if "%OS%"=="Windows_NT" setlocal 10 | 11 | set DIRNAME=%~dp0 12 | if "%DIRNAME%" == "" set DIRNAME=. 13 | set APP_BASE_NAME=%~n0 14 | set APP_HOME=%DIRNAME% 15 | 16 | @rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. 17 | set DEFAULT_JVM_OPTS= 18 | 19 | @rem Find java.exe 20 | if defined JAVA_HOME goto findJavaFromJavaHome 21 | 22 | set JAVA_EXE=java.exe 23 | %JAVA_EXE% -version >NUL 2>&1 24 | if "%ERRORLEVEL%" == "0" goto init 25 | 26 | echo. 27 | echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. 28 | echo. 29 | echo Please set the JAVA_HOME variable in your environment to match the 30 | echo location of your Java installation. 31 | 32 | goto fail 33 | 34 | :findJavaFromJavaHome 35 | set JAVA_HOME=%JAVA_HOME:"=% 36 | set JAVA_EXE=%JAVA_HOME%/bin/java.exe 37 | 38 | if exist "%JAVA_EXE%" goto init 39 | 40 | echo. 41 | echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME% 42 | echo. 43 | echo Please set the JAVA_HOME variable in your environment to match the 44 | echo location of your Java installation. 45 | 46 | goto fail 47 | 48 | :init 49 | @rem Get command-line arguments, handling Windows variants 50 | 51 | if not "%OS%" == "Windows_NT" goto win9xME_args 52 | 53 | :win9xME_args 54 | @rem Slurp the command line arguments. 55 | set CMD_LINE_ARGS= 56 | set _SKIP=2 57 | 58 | :win9xME_args_slurp 59 | if "x%~1" == "x" goto execute 60 | 61 | set CMD_LINE_ARGS=%* 62 | 63 | :execute 64 | @rem Setup the command line 65 | 66 | set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar 67 | 68 | @rem Execute Gradle 69 | "%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %CMD_LINE_ARGS% 70 | 71 | :end 72 | @rem End local scope for the variables with windows NT shell 73 | if "%ERRORLEVEL%"=="0" goto mainEnd 74 | 75 | :fail 76 | rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of 77 | rem the _cmd.exe /c_ return code! 78 | if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1 79 | exit /b 1 80 | 81 | :mainEnd 82 | if "%OS%"=="Windows_NT" endlocal 83 | 84 | :omega 85 | -------------------------------------------------------------------------------- /settings.gradle: -------------------------------------------------------------------------------- 1 | rootProject.name = 'object-sql-intellij' 2 | -------------------------------------------------------------------------------- /src/main/java/com/github/braisdom/objsql/intellij/ModelMethodBuilder.java: -------------------------------------------------------------------------------- 1 | package com.github.braisdom.objsql.intellij; 2 | 3 | import com.intellij.openapi.project.Project; 4 | import com.intellij.psi.PsiClass; 5 | import com.intellij.psi.PsiModifier; 6 | import com.intellij.psi.PsiType; 7 | import com.intellij.psi.impl.light.LightFieldBuilder; 8 | 9 | import java.util.List; 10 | 11 | import static com.github.braisdom.objsql.intellij.ObjSqlPsiAugmentProvider.*; 12 | 13 | final class ModelMethodBuilder { 14 | 15 | static void buildField(PsiClass psiClass, List result) { 16 | buildTableName(psiClass, result); 17 | buildRawAttributes(psiClass, result); 18 | } 19 | 20 | static void buildMethod(PsiClass psiClass, List result) { 21 | buildGetRawAttributes(psiClass, result); 22 | buildGetRawAttribute(psiClass, result); 23 | } 24 | 25 | private static void buildRawAttributes(PsiClass psiClass, List result) { 26 | PsiType fieldType = createParameterType(psiClass.getProject(), 27 | "java.util.Map", "java.lang.String", "java.lang.Object"); 28 | ObjsqlLightFieldBuilder primaryBuilder = new ObjsqlLightFieldBuilder("rawAttributes", fieldType, psiClass); 29 | primaryBuilder.setNavigationElement(psiClass); 30 | primaryBuilder.setModifiers(PsiModifier.PRIVATE, PsiModifier.FINAL); 31 | primaryBuilder.setContainingClass(psiClass); 32 | 33 | result.add(primaryBuilder); 34 | } 35 | 36 | private static void buildTableName(PsiClass psiClass, List result) { 37 | PsiType fieldType = createParameterType(psiClass.getProject(), "java.lang.String"); 38 | ObjsqlLightFieldBuilder tableNameBuilder = new ObjsqlLightFieldBuilder("TABLE_NAME", fieldType, psiClass); 39 | tableNameBuilder.setModifiers(PsiModifier.PUBLIC, PsiModifier.STATIC, PsiModifier.FINAL); 40 | tableNameBuilder.setContainingClass(psiClass); 41 | tableNameBuilder.setNavigationElement(tableNameBuilder); 42 | 43 | result.add(tableNameBuilder); 44 | } 45 | 46 | private static void buildGetRawAttributes(PsiClass psiClass, List result) { 47 | Project project = psiClass.getProject(); 48 | ObjSqlLightMethodBuilder methodBuilder = new ObjSqlLightMethodBuilder(psiClass.getManager(), "getRawAttributes"); 49 | PsiType psiType = getProjectType("java.util.Map", project); 50 | methodBuilder.withMethodReturnType(psiType) 51 | .withContainingClass(psiClass) 52 | .withModifier(PsiModifier.PUBLIC, PsiModifier.FINAL); 53 | 54 | if(!checkMethodExists(psiClass, methodBuilder)) { 55 | result.add(methodBuilder); 56 | } 57 | } 58 | 59 | private static void buildGetRawAttribute(PsiClass psiClass, List result) { 60 | Project project = psiClass.getProject(); 61 | ObjSqlLightMethodBuilder methodBuilder = new ObjSqlLightMethodBuilder(psiClass.getManager(), "getRawAttribute"); 62 | PsiType psiType = getProjectType("java.lang.Object", project); 63 | methodBuilder.withParameter("key", "java.lang.String") 64 | .withMethodReturnType(psiType) 65 | .withContainingClass(psiClass) 66 | .withModifier(PsiModifier.PUBLIC, PsiModifier.FINAL); 67 | 68 | if(!checkMethodExists(psiClass, methodBuilder)) { 69 | result.add(methodBuilder); 70 | } 71 | } 72 | } 73 | -------------------------------------------------------------------------------- /src/main/java/com/github/braisdom/objsql/intellij/ObjSqlImplicitUsageProvider.java: -------------------------------------------------------------------------------- 1 | package com.github.braisdom.objsql.intellij; 2 | 3 | import com.intellij.codeInsight.daemon.ImplicitUsageProvider; 4 | import com.intellij.psi.PsiClass; 5 | import com.intellij.psi.PsiElement; 6 | import com.intellij.psi.PsiField; 7 | 8 | public class ObjSqlImplicitUsageProvider implements ImplicitUsageProvider { 9 | 10 | @Override 11 | public boolean isImplicitUsage(PsiElement element) { 12 | return isImplicitWrite(element) || isImplicitRead(element); 13 | } 14 | 15 | @Override 16 | public boolean isImplicitRead(PsiElement element) { 17 | return checkUsage(element); 18 | } 19 | 20 | @Override 21 | public boolean isImplicitWrite(PsiElement element) { 22 | return checkUsage(element); 23 | } 24 | 25 | private boolean checkUsage(PsiElement element) { 26 | if (element instanceof PsiField) { 27 | PsiClass psiClass = ((PsiField)element).getContainingClass(); 28 | return psiClass.hasAnnotation(ObjSqlPsiAugmentProvider.DOMAIN_MODEL_CLASSNAME); 29 | } 30 | return false; 31 | } 32 | 33 | } -------------------------------------------------------------------------------- /src/main/java/com/github/braisdom/objsql/intellij/ObjSqlLightClassBuilder.java: -------------------------------------------------------------------------------- 1 | package com.github.braisdom.objsql.intellij; 2 | 3 | import com.intellij.psi.PsiElement; 4 | import com.intellij.psi.impl.light.LightPsiClassBuilder; 5 | import org.jetbrains.annotations.NotNull; 6 | 7 | public class ObjSqlLightClassBuilder extends LightPsiClassBuilder { 8 | private final String myQualifiedName; 9 | 10 | public ObjSqlLightClassBuilder(@NotNull PsiElement context, 11 | @NotNull String simpleName, 12 | @NotNull String qualifiedName) { 13 | super(context, simpleName); 14 | myQualifiedName = qualifiedName; 15 | } 16 | 17 | @Override 18 | public boolean equals(Object o) { 19 | if (this == o) { 20 | return true; 21 | } 22 | 23 | if (o == null || getClass() != o.getClass()) { 24 | return false; 25 | } 26 | 27 | ObjSqlLightClassBuilder that = (ObjSqlLightClassBuilder) o; 28 | 29 | return myQualifiedName.equals(that.myQualifiedName); 30 | } 31 | 32 | @Override 33 | public int hashCode() { 34 | return myQualifiedName.hashCode(); 35 | } 36 | } 37 | -------------------------------------------------------------------------------- /src/main/java/com/github/braisdom/objsql/intellij/ObjSqlLightMethodBuilder.java: -------------------------------------------------------------------------------- 1 | package com.github.braisdom.objsql.intellij; 2 | 3 | import com.intellij.ide.util.PsiNavigationSupport; 4 | import com.intellij.lang.ASTNode; 5 | import com.intellij.openapi.util.TextRange; 6 | import com.intellij.pom.Navigatable; 7 | import com.intellij.psi.*; 8 | import com.intellij.psi.impl.CheckUtil; 9 | import com.intellij.psi.impl.light.LightMethodBuilder; 10 | import com.intellij.psi.impl.light.LightModifierList; 11 | import com.intellij.psi.impl.light.LightTypeParameterListBuilder; 12 | import com.intellij.util.IncorrectOperationException; 13 | import org.jetbrains.annotations.NonNls; 14 | import org.jetbrains.annotations.NotNull; 15 | import org.jetbrains.annotations.Nullable; 16 | 17 | import java.util.Collection; 18 | import java.util.Objects; 19 | 20 | public class ObjSqlLightMethodBuilder extends LightMethodBuilder { 21 | private PsiMethod myMethod; 22 | private ASTNode myASTNode; 23 | 24 | public ObjSqlLightMethodBuilder(@NotNull PsiManager manager, @NotNull String name) { 25 | super(manager, name); 26 | } 27 | 28 | public ObjSqlLightMethodBuilder withModifier(@PsiModifier.ModifierConstant @NotNull @NonNls String modifier) { 29 | addModifier(modifier); 30 | return this; 31 | } 32 | 33 | public ObjSqlLightMethodBuilder withModifier(@PsiModifier.ModifierConstant @NotNull @NonNls String... modifiers) { 34 | for (String modifier : modifiers) { 35 | addModifier(modifier); 36 | } 37 | return this; 38 | } 39 | 40 | public ObjSqlLightMethodBuilder withMethodReturnType(PsiType returnType) { 41 | setMethodReturnType(returnType); 42 | return this; 43 | } 44 | 45 | @Override 46 | public void navigate(boolean requestFocus) { 47 | PsiClassUtil.navigate(getProject(), getContainingClass(), (psiClasses -> { 48 | for (PsiClass psiClass : psiClasses) { 49 | PsiMethod[] psiMethods = psiClass.getMethods(); 50 | for (PsiMethod psiMethod : psiMethods) { 51 | if (getName().equals(psiMethod.getName())) { 52 | Navigatable descriptor = PsiNavigationSupport.getInstance().getDescriptor(psiMethod); 53 | if (descriptor != null) { 54 | descriptor.navigate(requestFocus); 55 | return true; 56 | } 57 | } 58 | } 59 | } 60 | return false; 61 | })); 62 | } 63 | 64 | @Override 65 | public boolean canNavigate() { 66 | return PsiNavigationSupport.getInstance().canNavigate(this); 67 | } 68 | 69 | @Override 70 | public boolean canNavigateToSource() { 71 | return false; 72 | } 73 | 74 | @Override 75 | public LightMethodBuilder setMethodReturnType(PsiType returnType) { 76 | return super.setMethodReturnType(returnType); 77 | } 78 | 79 | @Override 80 | public LightMethodBuilder setMethodReturnType(String returnType) { 81 | return super.setMethodReturnType(returnType); 82 | } 83 | 84 | public ObjSqlLightMethodBuilder withParameter(@NotNull String name, @NotNull PsiType type) { 85 | addParameter(name, type); 86 | return this; 87 | } 88 | 89 | public ObjSqlLightMethodBuilder withParameter(@NotNull String name, @NotNull String type) { 90 | addParameter(name, type); 91 | return this; 92 | } 93 | 94 | public ObjSqlLightMethodBuilder withParameter(@NotNull String name, @NotNull PsiType type, boolean isVarArgs) { 95 | addParameter(name, type, isVarArgs); 96 | return this; 97 | } 98 | 99 | public ObjSqlLightMethodBuilder withParameter(@NotNull String name, @NotNull String type, boolean isVarArgs) { 100 | addParameter(name, JavaPsiFacade.getElementFactory(getProject()).createTypeFromText(type, this), isVarArgs); 101 | return this; 102 | } 103 | 104 | public ObjSqlLightMethodBuilder withException(@NotNull PsiClassType type) { 105 | addException(type); 106 | return this; 107 | } 108 | 109 | public ObjSqlLightMethodBuilder withContainingClass(@NotNull PsiClass containingClass) { 110 | setContainingClass(containingClass); 111 | return this; 112 | } 113 | 114 | // add Parameter as is, without wrapping with LightTypeParameter 115 | @Override 116 | public LightMethodBuilder addTypeParameter(PsiTypeParameter parameter) { 117 | ((LightTypeParameterListBuilder) getTypeParameterList()).addParameter(parameter); 118 | return this; 119 | } 120 | 121 | @Override 122 | public PsiElement getParent() { 123 | PsiElement result = super.getParent(); 124 | result = null != result ? result : getContainingClass(); 125 | return result; 126 | } 127 | 128 | @Nullable 129 | @Override 130 | public PsiFile getContainingFile() { 131 | PsiClass containingClass = getContainingClass(); 132 | return containingClass != null ? containingClass.getContainingFile() : null; 133 | } 134 | 135 | @Override 136 | public String getText() { 137 | ASTNode node = getNode(); 138 | if (null != node) { 139 | return node.getText(); 140 | } 141 | return ""; 142 | } 143 | 144 | @Override 145 | public ASTNode getNode() { 146 | if (null == myASTNode) { 147 | final PsiElement myPsiMethod = getOrCreateMyPsiMethod(); 148 | myASTNode = null == myPsiMethod ? null : myPsiMethod.getNode(); 149 | } 150 | return myASTNode; 151 | } 152 | 153 | @Override 154 | public TextRange getTextRange() { 155 | TextRange r = super.getTextRange(); 156 | return r == null ? TextRange.EMPTY_RANGE : r; 157 | } 158 | 159 | private String getAllModifierProperties(LightModifierList modifierList) { 160 | final StringBuilder builder = new StringBuilder(); 161 | for (String modifier : modifierList.getModifiers()) { 162 | if (!PsiModifier.PACKAGE_LOCAL.equals(modifier)) { 163 | builder.append(modifier).append(' '); 164 | } 165 | } 166 | return builder.toString(); 167 | } 168 | 169 | private PsiMethod rebuildMethodFromString() { 170 | PsiMethod result; 171 | try { 172 | final StringBuilder methodTextDeclaration = new StringBuilder(); 173 | methodTextDeclaration.append(getAllModifierProperties((LightModifierList) getModifierList())); 174 | PsiType returnType = getReturnType(); 175 | if (null != returnType && returnType.isValid()) { 176 | methodTextDeclaration.append(returnType.getCanonicalText()).append(' '); 177 | } 178 | methodTextDeclaration.append(getName()); 179 | methodTextDeclaration.append('('); 180 | if (getParameterList().getParametersCount() > 0) { 181 | for (PsiParameter parameter : getParameterList().getParameters()) { 182 | methodTextDeclaration.append(parameter.getType().getCanonicalText()).append(' ').append(parameter.getName()).append(','); 183 | } 184 | methodTextDeclaration.deleteCharAt(methodTextDeclaration.length() - 1); 185 | } 186 | methodTextDeclaration.append(')'); 187 | methodTextDeclaration.append('{').append(" ").append('}'); 188 | 189 | final PsiElementFactory elementFactory = JavaPsiFacade.getElementFactory(getManager().getProject()); 190 | 191 | result = elementFactory.createMethodFromText(methodTextDeclaration.toString(), getContainingClass()); 192 | if (null != getBody()) { 193 | result.getBody().replace(getBody()); 194 | } 195 | } catch (Exception ex) { 196 | result = null; 197 | } 198 | return result; 199 | } 200 | 201 | @Override 202 | public PsiElement copy() { 203 | final PsiElement myPsiMethod = getOrCreateMyPsiMethod(); 204 | return null == myPsiMethod ? null : myPsiMethod.copy(); 205 | } 206 | 207 | private PsiElement getOrCreateMyPsiMethod() { 208 | if (null == myMethod) { 209 | myMethod = rebuildMethodFromString(); 210 | } 211 | return myMethod; 212 | } 213 | 214 | @NotNull 215 | @Override 216 | public PsiElement[] getChildren() { 217 | final PsiElement myPsiMethod = getOrCreateMyPsiMethod(); 218 | return null == myPsiMethod ? PsiElement.EMPTY_ARRAY : myPsiMethod.getChildren(); 219 | } 220 | 221 | @Override 222 | public String toString() { 223 | return getClass().getSimpleName() + "#" + getName(); 224 | } 225 | 226 | @Override 227 | public PsiElement replace(@NotNull PsiElement newElement) throws IncorrectOperationException { 228 | // just add new element to the containing class 229 | final PsiClass containingClass = getContainingClass(); 230 | if (null != containingClass) { 231 | CheckUtil.checkWritable(containingClass); 232 | return containingClass.add(newElement); 233 | } 234 | return null; 235 | } 236 | 237 | @Override 238 | public boolean equals(Object o) { 239 | if (this == o) { 240 | return true; 241 | } 242 | 243 | if (!(o instanceof PsiMethod)) { 244 | return false; 245 | } 246 | 247 | PsiMethod that = (PsiMethod) o; 248 | if (!getName().equals(that.getName())) { 249 | return false; 250 | } 251 | if (isConstructor() != that.isConstructor()) { 252 | return false; 253 | } 254 | final PsiClass containingClass = getContainingClass(); 255 | final PsiClass thatContainingClass = that.getContainingClass(); 256 | if (containingClass != null ? !containingClass.equals(thatContainingClass) : thatContainingClass != null) { 257 | return false; 258 | } 259 | if (!getModifierList().equals(that.getModifierList())) { 260 | return false; 261 | } 262 | if (!getParameterList().equals(that.getParameterList())) { 263 | return false; 264 | } 265 | return Objects.equals(getReturnType().getPresentableText(), that.getReturnType().getPresentableText()); 266 | } 267 | 268 | @Override 269 | public int hashCode() { 270 | // should be constant because of RenameJavaMethodProcessor#renameElement and fixNameCollisionsWithInnerClassMethod(...) 271 | return 1; 272 | } 273 | 274 | @Override 275 | public void delete() throws IncorrectOperationException { 276 | // simple do nothing 277 | } 278 | 279 | @Override 280 | public void checkDelete() throws IncorrectOperationException { 281 | // simple do nothing 282 | } 283 | } 284 | -------------------------------------------------------------------------------- /src/main/java/com/github/braisdom/objsql/intellij/ObjSqlPsiAugmentProvider.java: -------------------------------------------------------------------------------- 1 | package com.github.braisdom.objsql.intellij; 2 | 3 | import com.intellij.openapi.project.Project; 4 | import com.intellij.openapi.util.RecursionGuard; 5 | import com.intellij.openapi.util.RecursionManager; 6 | import com.intellij.psi.*; 7 | import com.intellij.psi.augment.PsiAugmentProvider; 8 | import com.intellij.psi.impl.source.PsiExtensibleClass; 9 | import com.intellij.psi.search.GlobalSearchScope; 10 | import com.intellij.psi.util.CachedValueProvider; 11 | import com.intellij.psi.util.CachedValuesManager; 12 | import com.intellij.psi.util.PsiModificationTracker; 13 | import org.jetbrains.annotations.NotNull; 14 | import org.jetbrains.annotations.Nullable; 15 | 16 | import java.util.ArrayList; 17 | import java.util.Arrays; 18 | import java.util.Collections; 19 | import java.util.List; 20 | 21 | public class ObjSqlPsiAugmentProvider extends PsiAugmentProvider { 22 | 23 | public static final String DOMAIN_MODEL_CLASSNAME = "com.github.braisdom.objsql.annotations.DomainModel"; 24 | 25 | private static final List LANG_PRIMARY_TYPES = Arrays.asList(new String[]{"Long", "Integer", "String", "Short"}); 26 | 27 | @NotNull 28 | @Override 29 | protectedConversion between singular and plural form of a noun word.
8 | * 9 | * @author (Fei) John Chen 10 | */ 11 | public class WordUtil { 12 | 13 | private static final Map
543 | * Examples:
544 | * camelize("hello") "Hello"
545 | * camelize("hello world") "Hello world"
546 | * camelize("active_record") "ActiveRecord"
547 | * camelize("active_record", true) "activeRecord"
548 | *
549 | *
550 | * @param word the word to be converted to camelized form
551 | * @param firstLetterInLowerCase true if the first character should be in lower case
552 | * @return a camelized string
553 | */
554 | public static String camelize(String word, boolean firstLetterInLowerCase) {
555 | if (word == null || "".equals(word)) return word;
556 |
557 | String result = "";
558 | if (word.indexOf('_') != -1) {
559 | StringBuilder sb = new StringBuilder();
560 | int count = 0;
561 | StringTokenizer st = new StringTokenizer(word, "_");
562 | while(st.hasMoreTokens()) {
563 | String token = st.nextToken();
564 | count++;
565 | if (count == 1) {
566 | sb.append(camelizeOneWord(token, firstLetterInLowerCase));
567 | }
568 | else {
569 | sb.append(camelizeOneWord(token, false));
570 | }
571 | }
572 | result = sb.toString();
573 | }
574 | else {
575 | result = camelizeOneWord(word, firstLetterInLowerCase);
576 | }
577 | return result;
578 | }
579 |
580 | private static String camelizeOneWord(String word, boolean firstLetterInLowerCase) {
581 | if (word == null || "".equals(word)) return word;
582 |
583 | String firstChar = word.substring(0,1);
584 | String result = (firstLetterInLowerCase)?firstChar.toLowerCase():firstChar.toUpperCase();
585 | if (word.length() > 1) {
586 | result += word.substring(1);
587 | }
588 | return result;
589 | }
590 |
591 | /**
592 | * underscore is the reverse of camelize method.
593 | *
594 | *
595 | * Examples:
596 | * underscore("Hello world") "hello world"
597 | * underscore("ActiveRecord") "active_record"
598 | * underscore("The RedCross") "the red_cross"
599 | * underscore("ABCD") "abcd"
600 | *
601 | *
602 | * @param phase the original string
603 | * @return an underscored string
604 | */
605 | public static String underscore(String phase) {
606 | if (phase == null || "".equals(phase)) return phase;
607 |
608 | phase = phase.replace('-', '_');
609 | StringBuilder sb = new StringBuilder();
610 | int total = phase.length();
611 | for (int i = 0; i < total; i++) {
612 | char c = phase.charAt(i);
613 | if (i == 0) {
614 | if (isInA2Z(c)) {
615 | sb.append(("" + c).toLowerCase());
616 | }
617 | else {
618 | sb.append(c);
619 | }
620 | }
621 | else {
622 | if (isInA2Z(c)) {
623 | if (isIna2z(phase.charAt(i-1))) {
624 | sb.append(("_" + c).toLowerCase());
625 | }
626 | else {
627 | sb.append(("" + c).toLowerCase());
628 | }
629 | }
630 | else {
631 | sb.append(c);
632 | }
633 | }
634 | }
635 | return sb.toString();
636 | }
637 |
638 | private static String A2Z = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
639 | private static String a2z = "abcdefghijklmnopqrstuvwxyz";
640 |
641 | private static boolean isInA2Z(char c) {
642 | return (A2Z.indexOf(c) != -1)?true:false;
643 | }
644 |
645 | private static boolean isIna2z(char c) {
646 | return (a2z.indexOf(c) != -1)?true:false;
647 | }
648 |
649 | /**
650 | * Replaces all dashes and underscores by spaces and capitalizes all the words.
651 | *
652 | *
653 | * Examples:
654 | * titleize("ch 1: Java-ActiveRecordIsFun") "Ch 1: Java Active Record Is Fun"
655 | *
656 | *
657 | * @param phase the original string
658 | * @return a titleized string
659 | */
660 | public static String titleize(String phase) {
661 | if (phase == null || "".equals(phase)) return phase;
662 |
663 | phase = humanize(phase);
664 | StringBuilder sb = new StringBuilder();
665 | int total = phase.length();
666 | for (int i = 0; i < total; i++) {
667 | char c = phase.charAt(i);
668 | if (i == 0) {
669 | if (isIna2z(c)) {
670 | sb.append(("" + c).toUpperCase());
671 | }
672 | else {
673 | sb.append(c);
674 | }
675 | }
676 | else {
677 | if (isIna2z(c) && ' ' == phase.charAt(i-1)) {
678 | sb.append(("" + c).toUpperCase());
679 | }
680 | else {
681 | sb.append(c);
682 | }
683 | }
684 | }
685 | return sb.toString();
686 | }
687 |
688 | /**
689 | * Replaces all dashes and underscores by spaces and capitalizes the first
690 | * word. Also removes
691 | *
692 | *
693 | * Examples:
694 | * humanize("active_record") "Active record"
695 | * humanize("post_id") "Post"
696 | *
697 | *
698 | * @param phase the original string
699 | * @return a humanized string
700 | */
701 | public static String humanize(String phase) {
702 | if (phase == null || "".equals(phase)) return phase;
703 | phase = underscore(phase);
704 | if (phase.endsWith("_id")) phase += " ";
705 | return camelize(phase.replaceAll("_id ", " ").replace('_', ' ').trim());
706 | }
707 |
708 | /**
709 | * Returns a database table tableName corresponding to the input entityClass class
710 | * tableName.
711 | *
712 | *
713 | * Examples:
714 | * tableize("Person") "people"
715 | * tableize("LineItem") "line_items"
716 | *
717 | *
718 | * @param modelClassName
719 | * @return the table tableName of the java entityClass class tableName
720 | */
721 | public static String tableize(String modelClassName) {
722 | return pluralize(underscore(modelClassName));
723 | }
724 |
725 | /**
726 | * Returns a entityClass class tableName corresponding to the input database
727 | * table tableName.
728 | *
729 | *
730 | * Examples:
731 | * classify("people") "Person"
732 | * classify("line_items") "LineItem"
733 | *
734 | *
735 | * @param tableName java class tableName of the entityClass
736 | * @return a java entityClass class tableName
737 | */
738 | public static String classify(String tableName) {
739 | return camelize(singularize(tableName));
740 | }
741 |
742 | public static boolean isEmpty(String str) {
743 | return str == null || str.length() == 0;
744 | }
745 |
746 | public static String removeEnd(String str, String remove) {
747 | if (isEmpty(str) || isEmpty(remove)) {
748 | return str;
749 | }
750 | if (str.endsWith(remove)) {
751 | return str.substring(0, str.length() - remove.length());
752 | }
753 | return str;
754 | }
755 |
756 | /**
757 | * Returns an ordinalized string.
758 | *
759 | * 760 | * Examples: 761 | * ordinalize(100) "100th" 762 | * ordinalize(1003) "1003rd" 763 | *764 | * 765 | * @param number the number 766 | * @return an ordinalized string for the number 767 | */ 768 | public static String ordinalize(int number) { 769 | String result = "" + number; 770 | if (result.endsWith("1")) result = result + "st"; 771 | else if (result.endsWith("2")) result = result + "nd"; 772 | else if (result.endsWith("3")) result = result + "rd"; 773 | else result = result + "th"; 774 | return result; 775 | } 776 | } -------------------------------------------------------------------------------- /src/main/java/com/github/braisdom/objsql/intellij/oo/OOComponent.java: -------------------------------------------------------------------------------- 1 | /* Copyright 2013 Artem Melentyev
Supported annotations of ObjectiveSql:
13 |See more: ObjectiveSql
23 | ]]>1. Generating setter and getter methods for model who has annotation @DomainModel
27 |2. Generating query and persistence methods for model who has annotation @DomainModel
28 |3. Generating query method for field who has annotation @Queryable
29 |4. Generating relation method for field who has annotation @Relation
30 |...
31 | ]]>