Kiln » TortoiseHg » TortoiseHg
Clone URL:  
Pushed to one repository · View In Graph Contained in 0.8, 0.8.1, and 0.8.2

add separate module for menu creation

Changeset 9521df28903d

Parent a7d4a0c9e5e5

by Simon Heimberg

Changes to one file · Browse files at 9521df28903d Showing diff from parent a7d4a0c9e5e5 Diff from another changeset...

Change 1 of 1 Show Entire File tortoise/​menuthg.py Stacked
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
@@ -0,0 +1,271 @@
+# Published under the GNU GPL, v2 or later. +# Copyright (C) 2007 Jelmer Vernooij <jelmer@samba.org> +# Copyright (C) 2007 Henry Ludemann <misc@hl.id.au> +# Copyright (C) 2007 TK Soh <teekaysoh@gmail.com> + +import os +#import tempfile +from mercurial import hg +from thgutil import * +from mercurial import ui +from mercurial.i18n import _ + +try: + from mercurial.error import RepoError +except ImportError: + from mercurial.repo import RepoError + + +class TortoiseMenu(object): + + def __init__(self, menutext, helptext, handler, icon=None, state=True): + self.menutext = menutext + self.helptext = helptext + self.handler = handler + self.state = state + self.icon = icon + + def isSubmenu(): + return false + + def isSep(): + return false + + +class TortoiseSubmenu(TortoiseMenu): + + def __init__(self, menutext, helptext, menus=[], icon=None): + TortoiseMenu.__init(self, menutext, helptext, None, icon) + self.menus = menus[:] + + def add_menu(self, menutext, helptext, handler, icon=None, state=True): + self.menus.append(TortoiseMenu(menutext, helptext, + handler, icon, state)) + + def add_sep(self): + self.menus.append(TortoiseMenuSep()) + + def get_menus(self): + return self.menus + + def append(self, entry): + self.menus.append(entry) + + def isSubmenu(): + return true + + +class TortoiseMenuSep(object): + + def isSubmenu(): + return false + + def isSep(): + return true + + +class menuThg: + """shell extension that adds context menu items""" + + def __init__(self): + self.handlers = self + self.name = "TortoiseHG" + + def get_commands_dragdrop(self, srcfiles, destfolder): + """ + Get a list of commands valid for the current selection. + + Commands are instances of TortoiseMenu, TortoiseMenuSep or TortoiseMenu + """ + + print "_get_commands_dragdrop() on %s" % ", ".join(self._filenames) + + # we can only accept dropping one item + if len(srcfiles) > 1: + return [] + + # open repo + drag_repo = None + drop_repo = None + + print "drag = %s" % self.srcfiles[0] + print "drop = %s" % self.destfolder + + drag_path = self.srcfiles[0] + drag_repo = open_repo(drag_path) + if not drag_repo: + return [] + if drag_repo and drag_repo.root != drag_path: + return [] # dragged item must be a hg repo root directory + + drop_repo = open_repo(self._folder) + + thgmenu = [] + thgmenu.append(TortoiseMenu(_("Create Clone"), + _("Create clone here from source"), + self.handlers._clone_here, icon="menuclone.ico")) + + if drop_repo: + thgmenu.append(TortoiseMenu(_("Synchronize"), + _("Synchronize with dragged repository"), + self.handlers._synch_here, icon="menusynch.py")) + return thgmenu + + def get_commands(self, files): + """ + Get a list of commands valid for the current selection. + + Commands are instances of TortoiseMenu, TortoiseMenuSep or TortoiseMenu + """ + + print "_get_commands() on %s" % ", ".join(self._filenames) + + # open repo + repo = open_repo(files[0]) + + if repo: + thgmenu.append(TortoiseMenu(_("HG Commit..."), + _("Commit changes in repository"), + self.handlers._commit, icon="menucommit.ico")) + + menu = TortoiseSubmenu(self.name, "Mercurial", [], icon="hg.ico") + + if repo is None: + menu.append(TortoiseMenu(_("Clone a Repository"), + _("clone a repository"), + self.handlers._clone, icon="menuclone.ico")) + menu.append(TortoiseMenu(_("Create Repository Here"), + _("create a new repository in this directory"), + self.handlers._init, icon="menucreaterepos.ico", + state=os.path.isdir(rpath))) + else: + + for f in self._filenames: + if f.endswith('.hgignore'): + menu.append(TortoiseMenu(_("Modify ignore filter"), + _("Modify repository ignore filter"), + self.handlers._hgignore, icon="general.ico")) # needs ico + break + + menu.append(TortoiseMenu(_("View File Status"), + _("Repository status"), + self.handlers._status, icon="menushowchanged.ico")) + + menu.append(TortoiseMenu(_("(Un)Shelve Changes"), + _("Shelve repository changes"), + self.handlers._shelve, icon="general.ico")) # needs ico + + # Visual Diff (any extdiff command) + has_vdiff = repo.ui.config('tortoisehg', 'vdiff', '') != '' + menu.append(TortoiseMenu(_("Visual Diff"), + _("View changes using GUI diff tool"), + self.handlers._vdiff, icon="TortoiseMerge.ico", + state=has_vdiff)) + + if len(self._filenames) == 0: + menu.append(TortoiseMenu(_("Guess Renames"), + _("Detect renames and copies"), + self.handlers._guess_rename, icon="general.ico")) # needs ico + elif len(self._filenames) == 1: + result.append(TortoiseMenu(_("Rename file"), + _("Rename file or directory"), + self.handlers._rename, icon="general.ico")) # needs ico + + menu.append(TortoiseMenu(_("Add Files"), + _("Add files to Hg repository"), + self.handlers._add, icon="menuadd.ico")) + menu.append(TortoiseMenu(_("Remove Files"), + _("Remove selected files on the next commit"), + self.handlers._remove, icon="menudelete.ico")) + menu.append(TortoiseMenu(_("Undo Changes"), + _("Revert selected files"), + self.handlers._revert, icon="menurevert.ico")) + + # we can only annotate file but not directories + annotatible = len(files) > 0 + for f in self._filenames: + if not os.path.isfile(f): + annotatible = False + break + menu.append(TortoiseMenu(_("Annotate Files"), + _("show changeset information per file line"), + self.handlers._annotate, icon="menublame.ico", + state=annotatible)) + + menu.append(TortoiseMenuSep()) + menu.append(TortoiseMenu(_("Update To Revision"), + _("update working directory"), + self.handlers._update, icon="menucheckout.ico")) + + can_merge = len(repo.heads()) > 1 and \ + len(repo.changectx(None).parents()) < 2 + menu.append(TortoiseMenu(_("Merge Revisions"), + _("merge working directory with another revision"), + self.handlers._merge, icon="menumerge.ico", + state=can_merge)) + + in_merge = len(repo.changectx(None).parents()) > 1 + menu.append(TortoiseMenu(_("Undo Merge"), + _("Undo merge by updating to revision"), + self.handlers._merge, icon="menuunmerge.ico", + state=in_merge)) + + menu.append(TortoiseMenuSep()) + + menu.append(TortoiseMenu(_("View Changelog"), + _("View revision history"), + self.handlers._history, icon="menulog.ico")) + + menu.append(TortoiseMenu(_("Search Repository"), + _("Search revisions of files for a text pattern"), + self.handlers._grep, icon="menurepobrowse.ico")) + + if repo.ui.config('tortoisehg', 'view'): + menu.append(TortoiseMenu(_("Revision Graph"), + _("View history with DAG graph"), + self.handlers._view, icon="menurevisiongraph.ico")) + + menu.append(TortoiseMenuSep()) + + menu.append(TortoiseMenu(_("Synchronize..."), + _("Synchronize with remote repository"), + self.handlers._synch, icon="menusynch.ico")) + menu.append(TortoiseMenu(_("Recovery..."), + _("General repair and recovery of repository"), + self.handlers._recovery, icon="general.ico")) + menu.append(TortoiseMenu(_("Web Server"), + _("start web server for this repository"), + self.handlers._serve, icon="proxy.ico")) + + menu.append(TortoiseMenuSep()) + menu.append(TortoiseMenu(_("Create Clone"), + _("Clone a repository here"), + self.handlers._clone, icon="menuclone.ico")) + can_init = repo.root != rpath and os.path.isdir(rpath) + menu.append(TortoiseMenu(_("Create Repository Here"), + _("create a new repository in this directory"), + self.handlers._init, icon="menucreaterepos.ico", + state=can_init)) + + # config settings menu + menu.append(TortoiseMenuSep()) + optmenu = TortoiseSubmenu(_("Settings"), '', + icon="menusettings.ico") + optmenu.add_menu(_("Global"), + _("Configure user wide settings"), + self.handlers._config_user, icon="settings_user.ico") + if repo: + optmenu.add_menu(_("Repository"), + _("Configure settings local to this repository"), + self.handlers._config_repo, + icon="settings_repo.ico") + menu.append(optmenu) + + # add common menu items + menu.append(TortoiseMenuSep()) + menu.append(TortoiseMenu(_("About"), _("About TortoiseHg"), + self.handlers._about, icon="menuabout.ico")) + + thgmenu.append(menu) + thgmenu.append(TortoiseMenuSep()) + return thgmenu