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

cmenu: better relative path algorithm

Changeset 0272052c9684…

Parent a93ecb406a4b…

by Steve Borho

Changes to 2 files · Browse files at 0272052c9684 Showing diff from parent a93ecb406a4b Diff from another changeset...

Change 1 of 1 Show Entire File contrib/​nautilus-thg.py Stacked
 
165
166
167
168
 
 
 
 
 
 
169
170
171
 
165
166
167
 
168
169
170
171
172
173
174
175
176
@@ -165,7 +165,12 @@
  if repo:   menus = self.menu.get_commands(repo, cwd, files)   for f in files: - self.files.append(util.canonpath(repo.root, cwd, f)) + cpath = util.canonpath(repo.root, cwd, f) + if cpath.startswith(cwd): + cpath = cpath[len(cwd+os.sep):] + self.files.append(cpath) + else: + self.files.append(f)   else:   menus = self.menu.get_norepo_commands(cwd, files)   self.cwd = cwd
Change 1 of 1 Show Changes Only tortoise/​contextmenu.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
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
 # 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  import pythoncom  from win32com.shell import shell, shellcon  import win32con  import win32process  import win32event  import win32ui  import win32gui  import win32gui_struct  import win32api  import _winreg  from mercurial import hg  from thgutil import *  import menuthg    try:   from mercurial.error import RepoError  except ImportError:   from mercurial.repo import RepoError    # FIXME: quick workaround traceback caused by missing "closed"  # attribute in win32trace.  import sys  from mercurial import ui  def write_err(self, *args):   for a in args:   sys.stderr.write(str(a))  ui.ui.write_err = write_err    S_OK = 0  S_FALSE = 1    def get_clone_repo_name(dir, repo_name):   dest_clone = os.path.join(dir, repo_name)   if os.path.exists(dest_clone):   dest_clone = os.path.join(dir, "Clone of " + repo_name)     i = 2   while os.path.exists(dest_clone):   dest_clone = os.path.join(dir, "Clone of (%s) %s" % (i, repo_name))   i += 1   return dest_clone    """Windows shell extension that adds context menu items to Mercurial repository"""  class ContextMenuExtension(menuthg.menuThg):   _reg_progid_ = "Mercurial.ShellExtension.ContextMenu"   _reg_desc_ = "Mercurial Shell Extension"   _reg_clsid_ = "{EEE9936B-73ED-4D45-80C9-AF918354F885}"   _com_interfaces_ = [shell.IID_IShellExtInit, shell.IID_IContextMenu]   _public_methods_ = [   "Initialize", # From IShellExtInit   "QueryContextMenu", "InvokeCommand", "GetCommandString" # IContextMenu   ]     registry_keys = [   (_winreg.HKEY_CLASSES_ROOT,   r"*\shellex\ContextMenuHandlers\TortoiseHg",   [(None, _reg_clsid_)]),   (_winreg.HKEY_CLASSES_ROOT,   r"Directory\Background\shellex\ContextMenuHandlers\TortoiseHg",   [(None, _reg_clsid_)]),   (_winreg.HKEY_CLASSES_ROOT,   r"Directory\shellex\ContextMenuHandlers\TortoiseHg",   [(None, _reg_clsid_)]),   (_winreg.HKEY_CLASSES_ROOT,   r"Folder\shellex\ContextMenuHandlers\TortoiseHg",   [(None, _reg_clsid_)]),   (_winreg.HKEY_CLASSES_ROOT,   r"Directory\shellex\DragDropHandlers\TortoiseHg",   [(None, _reg_clsid_)]),   (_winreg.HKEY_CLASSES_ROOT,   r"Folder\shellex\DragDropHandlers\TortoiseHg",   [(None, _reg_clsid_)]),   ]     def __init__(self):   self.folder = None   self.fnames = []   self.menuitems = {}   menuthg.menuThg.__init__(self)     def Initialize(self, folder, dataobj, hkey):   if folder:   self.folder = shell.SHGetPathFromIDList(folder)     if dataobj:   format_etc = win32con.CF_HDROP, None, 1, -1, pythoncom.TYMED_HGLOBAL   sm = dataobj.GetData(format_etc)   num_files = shell.DragQueryFile(sm.data_handle, -1)   for i in range(num_files):   self.fnames.append(shell.DragQueryFile(sm.data_handle, i))     def _create_menu(self, parent, menus, pos, idCmd, idCmdFirst):   for menu_info in menus:   if menu_info.isSep():   win32gui.InsertMenu(parent, pos,   win32con.MF_BYPOSITION|win32con.MF_SEPARATOR,   idCmdFirst + idCmd, None)   elif menu_info.isSubmenu():   submenu = win32gui.CreatePopupMenu()   idCmd = self._create_menu(submenu, menu_info.get_menus(), 0,   idCmd, idCmdFirst)   opt = {   'text' : menu_info.menutext,   'wID' : idCmdFirst + idCmd,   'hSubMenu' : submenu,   }     if menu_info.icon:   icon_path = get_icon_path("tortoise", menu_info.icon)   opt['hbmpChecked'] = opt['hbmpUnchecked'] = \   icon_to_bitmap(icon_path)     item, _ = win32gui_struct.PackMENUITEMINFO(**opt)   win32gui.InsertMenuItem(parent, pos, True, item)   self._handlers[idCmd] = ("", "")   else:   fstate = win32con.MF_BYCOMMAND   if menu_info.state is False:   fstate |= win32con.MF_GRAYED     opt = {   'text' : menu_info.menutext,   'fState' : fstate,   'wID' : idCmdFirst + idCmd,   }     if menu_info.icon:   icon_path = get_icon_path("tortoise", menu_info.icon)   opt['hbmpChecked'] = opt['hbmpUnchecked'] = \   icon_to_bitmap(icon_path)     item, _ = win32gui_struct.PackMENUITEMINFO(**opt)   win32gui.InsertMenuItem(parent, pos, True, item)   self._handlers[idCmd] = (menu_info.helptext, menu_info.hgcmd)   idCmd += 1   pos += 1   return idCmd     def QueryContextMenu(self, hMenu, indexMenu, idCmdFirst, idCmdLast, uFlags):   if uFlags & shellcon.CMF_DEFAULTONLY:   return 0     thgmenu = []     # a brutal hack to detect if we are the first menu to go on to the   # context menu. If we are not the first, then add a menu separator   # The number '30000' is just a guess based on my observation   print "idCmdFirst = ", idCmdFirst   if idCmdFirst >= 30000:   thgmenu.append(menuthg.TortoiseMenuSep())   # As we are a context menu handler, we can ignore verbs.     if self.folder:   cwd = self.folder   elif self.fnames:   f = self.fnames[0]   cwd = os.path.isdir(f) and f or os.path.dirname(f)     self._handlers = {}   if self.folder and self.fnames:   # get menus with drag-n-drop support   thgmenu += self.get_commands_dragdrop(self.fnames, self.folder)   repo = menuthg.open_repo(self.folder)   else:   # get menus for hg menu   repo = menuthg.open_repo(cwd)   if repo:   thgmenu += self.get_commands(repo, cwd, self.fnames)   else:   thgmenu += self.get_norepo_commands(cwd, self.fnames)     self.cwd = cwd   self.repo = repo   idCmd = self._create_menu(hMenu, thgmenu, indexMenu, 0, idCmdFirst)   # Return total number of menus & submenus we've added   return idCmd     def InvokeCommand(self, ci):   mask, hwnd, verb, params, dir, nShow, hotkey, hicon = ci   if verb >> 16:   # This is a textual verb invocation... not supported.   return S_FALSE   if verb not in self.menuitems:   raise Exception("Unsupported command id %i!" % verb)   self.run_dialog(self.menuitems[verb][1])     def GetCommandString(self, cmd, uFlags):   if uFlags & shellcon.GCS_VALIDATEA or uFlags & shellcon.GCS_VALIDATEW:   if cmd in self.menuitems:   return S_OK   return S_FALSE   if uFlags & shellcon.GCS_VERBA or uFlags & shellcon.GCS_VERBW:   return S_FALSE   if uFlags & shellcon.GCS_HELPTEXTA or uFlags & shellcon.GCS_HELPTEXTW:   # The win32com.shell implementation encodes the resultant   # string into the correct encoding depending on the flags.   return self.menuitems[cmd][0]   return S_FALSE     def run_dialog(self, hgcmd):   cwd = self.cwd   if self.repo:   # Convert filenames to be relative to cwd   files = []   for f in self.fnames: - files.append(util.canonpath(self.repo.root, cwd, f)) + cpath = util.canonpath(self.repo.root, cwd, f) + if cpath.startswith(cwd): + cpath = cpath[len(cwd+os.sep):] + files.append(cpath) + else: + files.append(f)   self.fnames = files   gpopts = "--command %s " % hgcmd   if filelist:   fd, tmpfile = tempfile.mkstemp(prefix="tortoisehg_filelist_")   os.write(fd, "\n".join(self.fnames))   os.close(fd)   gpopts += " --listfile %s --deletelistfile" % (shellquote(tmpfile))   app_path = find_path("hgproc", get_prog_root(), '.EXE;.BAT')   cmdline = shellquote(app_path) + gpopts   try:   import subprocess   pop = subprocess.Popen(cmdline,   shell=False,   cwd=cwd,   creationflags=win32con.CREATE_NO_WINDOW,   stderr=subprocess.STDOUT,   stdout=subprocess.PIPE,   stdin=subprocess.PIPE)   except win32api.error, details:   win32ui.MessageBox("Error executing command - %s" % (details),   "gpopen")