SublimeDlangAutoModuleName/auto_module_name.py at master · vitalfadeev/SublimeDlangAutoModuleName · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
import sublime, sublime_plugin
import re
import os

from Default import history_list


class DlangAutoModuleNameCommand( sublime_plugin.TextCommand ):
    def is_visible( self ):
        return self.view.match_selector( 0, "source.d" )


    def _check_exists( self, edit ):
        # check "module ..."
        regions = self.view.find_all( r"^module\s.*" )

        if regions:
            return regions[ 0 ].b


    def _get_module_name_from_abs_path( self, abs_path ):
        import os.path

        # remove ext
        abs_path, ext = os.path.splitext( abs_path )

        # trim.  project/source/ui/event to ui/event
        folders = []
        folder_path, folder = os.path.split( abs_path )

        while ( folder_path and folder ):
            if folder == "source" or folder == "src":
                break

            folders.insert( 0, folder )
            folder_path, folder = os.path.split( folder_path )

        return ".".join( folders )


    def _get_module_name_from_file_name( self ):
        # file_name
        import os.path

        abs_name = self.view.file_name()
        if abs_name:
            name = self._get_module_name_from_abs_path( abs_name )
            return name


    def _get_module_name_from_class( self ):
        # grep "class ..."
        extracts = []
        regions = self.view.find_all( r"^class\s[.]*(\w+)", 0, r"\1", extracts )

        if extracts:
            return extracts[ 0 ].lower()


    def _get_module_name_from_interface( self ):
        # grep "class ..."
        extracts = []
        regions = self.view.find_all( r"^interface\s[.]*(\w+)", 0, r"\1", extracts )

        if extracts:
            return extracts[ 0 ].lower()


    def _get_module_name_from_struct( self ):
        # grep "class ..."
        extracts = []
        regions = self.view.find_all( r"^struct\s[.]*(\w+)", 0, r"\1", extracts )

        if extracts:
            return extracts[ 0 ].lower()


    def _get_module_name( self ):
        name = self._get_module_name_from_file_name()
        if name:
            return name

        name = self._get_module_name_from_class()
        if name:
            return name

        name = self._get_module_name_from_interface()
        if name:
            return name

        name = self._get_module_name_from_struct()
        if name:
            return name


    def _at_top( self, edit, name ):
        # at top in file
        insert_string = "module {};\n\n".format( name )

        self.view.insert( edit, 0, insert_string )

        return len( "module " )


    def _select( self, sel_i ):
        sel = self.view.sel()
        sel.clear()
        sel.add( sublime.Region( sel_i, sel_i ) )

        # scroll t show it
        self.view.show( sel_i )


    def run(self, edit, **args):
        history_list.get_jump_history_for_view(self.view).push_selection(self.view)

        # Check
        exist_point = self._check_exists( edit )
        if exist_point:
            self._select( exist_point )
            return

        # Get Name
        name = self._get_module_name()
        if name:
            inserted = self._at_top( edit, name )

            # Select inserted
            if inserted is not None:
                self._select( inserted )