Jump to table identifier if possible

Author: cedricziel

src/main/java/com/cedricziel/idea/typo3/util/TableUtil.java

@@ -22,7 +22,7 @@ public class TableUtil {
     public static final String EXT_TABLES_SQL_FILENAME = "ext_tables.sql";
 
     public static Set<String> getAvailableTableNames(@NotNull Project project) {
-        PsiFile[] extSqlFiles = FilenameIndex.getFilesByName(project, EXT_TABLES_SQL_FILENAME, GlobalSearchScope.allScope(project));
+        PsiFile[] extSqlFiles = getExtTablesSqlFilesInProject(project);
 
         Set<String> tableNames = new HashSet<>();
 
@@ -53,17 +53,55 @@ public static Set<String> getAvailableTableNames(@NotNull Project project) {
         return tableNames;
     }
 
-    public static PsiElement[] getExtTablesSqlFilesForTable(@NotNull String tableName, @NotNull Project project) {
-        PsiFile[] extSqlFiles = FilenameIndex.getFilesByName(project, EXT_TABLES_SQL_FILENAME, GlobalSearchScope.allScope(project));
+    public static PsiElement[] getTableDefinitionElements(@NotNull String tableName, @NotNull Project project) {
+
+        PsiFile[] extTablesSqlFilesInProjectContainingTable = getExtTablesSqlFilesInProjectContainingTable(tableName, project);
 
         return Arrays
-                .stream(extSqlFiles)
+                .stream(extTablesSqlFilesInProjectContainingTable)
+                .map(file -> {
+                    CharSequence charSequence = LoadTextUtil.loadText(file.getVirtualFile());
+
+                    final Matcher matcher = Pattern
+                            .compile("create\\s+table\\s+(if\\s+not\\s+exists\\s+)?([a-zA-Z_0-9]+)", Pattern.CASE_INSENSITIVE)
+                            .matcher(charSequence);
+                    try {
+                        while (matcher.find()) {
+                            if (matcher.groupCount() < 2) {
+                                continue;
+                            }
+
+                            final String foundTable = matcher.group(2);
+                            if (!foundTable.equals(tableName)) {
+                                continue;
+                            }
+
+                            return file.findElementAt(matcher.end() - 2);
+                        }
+                    } catch (IllegalStateException e) {
+                        // do nothing
+                    }
+
+                    return null;
+                })
                 .filter(Objects::nonNull)
-                .filter(file -> LoadTextUtil.loadText(file.getVirtualFile()).toString().contains(tableName))
-                .map(file -> file.findElementAt(LoadTextUtil.loadText(file.getVirtualFile()).toString().indexOf(tableName)))
                 .toArray(PsiElement[]::new);
     }
 
+    private static PsiFile[] getExtTablesSqlFilesInProjectContainingTable(@NotNull String tableName, @NotNull Project project) {
+
+        return Arrays
+                .stream(getExtTablesSqlFilesInProject(project))
+                .filter(Objects::nonNull)
+                .filter(file -> LoadTextUtil.loadText(file.getVirtualFile()).toString().contains(tableName))
+                .toArray(PsiFile[]::new);
+    }
+
+    @NotNull
+    private static PsiFile[] getExtTablesSqlFilesInProject(@NotNull Project project) {
+        return FilenameIndex.getFilesByName(project, EXT_TABLES_SQL_FILENAME, GlobalSearchScope.allScope(project));
+    }
+
     public static void completeAvailableTableNames(@NotNull Project project, @NotNull CompletionResultSet completionResultSet) {
         for (String name : TableUtil.getAvailableTableNames(project)) {
             completionResultSet.addElement(new LookupElement() {