~ubuntu-branches/ubuntu/intrepid/gedit/intrepid-proposed

« back to all changes in this revision

Viewing changes to plugins/snippets/snippets/SnippetsLibrary.py

  • Committer: Bazaar Package Importer
  • Author(s): Sebastien Bacher
  • Date: 2007-07-04 15:15:07 UTC
  • mto: (2.1.2 lenny)
  • mto: This revision was merged to the branch mainline in revision 43.
  • Revision ID: james.westby@ubuntu.com-20070704151507-1pqdkv92m58z9n2n
Tags: upstream-2.19.2
ImportĀ upstreamĀ versionĀ 2.19.2

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
#    Gedit snippets plugin
2
 
#    Copyright (C) 2005-2006  Jesse van den Kieboom <jesse@icecrew.nl>
3
 
#
4
 
#    This program is free software; you can redistribute it and/or modify
5
 
#    it under the terms of the GNU General Public License as published by
6
 
#    the Free Software Foundation; either version 2 of the License, or
7
 
#    (at your option) any later version.
8
 
#
9
 
#    This program is distributed in the hope that it will be useful,
10
 
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 
#    GNU General Public License for more details.
13
 
#
14
 
#    You should have received a copy of the GNU General Public License
15
 
#    along with this program; if not, write to the Free Software
16
 
#    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17
 
 
18
 
import ElementTree as et
19
 
import os
20
 
import weakref
21
 
import sys
22
 
import gtk
23
 
from functions import *
24
 
 
25
 
class NamespacedId:
26
 
        def __init__(self, namespace, id):
27
 
                if not id:
28
 
                        self.id = None
29
 
                else:
30
 
                        if namespace:
31
 
                                self.id = namespace + '-'
32
 
                        else:
33
 
                                self.id = 'global-'
34
 
                
35
 
                        self.id += id
36
 
 
37
 
class SnippetData:
38
 
        PROPS = {'tag': '', 'text': '', 'description': 'New snippet', 
39
 
                        'accelerator': ''}
40
 
 
41
 
        def __init__(self, node, library):
42
 
                self.priv_id = node.attrib.get('id')
43
 
 
44
 
                self.set_library(library)
45
 
                self.valid = False
46
 
                self.set_node(node)
47
 
 
48
 
        def can_modify(self):
49
 
                return (self.library and (isinstance(self.library(), SnippetsUserFile)))
50
 
 
51
 
        def set_library(self, library):
52
 
                if library:
53
 
                        self.library = weakref.ref(library)
54
 
                else:
55
 
                        self.library = None
56
 
 
57
 
                self.id = NamespacedId(self.language(), self.priv_id).id
58
 
 
59
 
        def set_node(self, node):
60
 
                if self.can_modify():
61
 
                        self.node = node
62
 
                else:
63
 
                        self.node = None
64
 
                
65
 
                self.init_snippet_data(node)
66
 
                
67
 
        def init_snippet_data(self, node):
68
 
                if node == None:
69
 
                        return
70
 
 
71
 
                self.override = node.attrib.get('override')
72
 
 
73
 
                self.properties = {}
74
 
                props = SnippetData.PROPS.copy()
75
 
                
76
 
                # Store all properties present
77
 
                for child in node:
78
 
                        if props.has_key(child.tag):
79
 
                                del props[child.tag]
80
 
                                
81
 
                                # Normalize accelerator
82
 
                                if child.tag == 'accelerator' and child.text != None:
83
 
                                        keyval, mod = gtk.accelerator_parse(child.text)
84
 
                                        
85
 
                                        if not gtk.accelerator_valid(keyval, mod):
86
 
                                                child.text = ''
87
 
                                        else:
88
 
                                                child.text = gtk.accelerator_name(keyval, mod)
89
 
                                
90
 
                                if self.can_modify():
91
 
                                        self.properties[child.tag] = child
92
 
                                else:
93
 
                                        self.properties[child.tag] = child.text or ''
94
 
                
95
 
                # Create all the props that were not found so we stay consistent
96
 
                for prop in props:
97
 
                        if self.can_modify():
98
 
                                child = et.SubElement(node, prop)
99
 
 
100
 
                                child.text = props[prop]
101
 
                                self.properties[prop] = child
102
 
                        else:
103
 
                                self.properties[prop] = props[prop]
104
 
                
105
 
                self.check_validation()
106
 
        
107
 
        def check_validation(self):
108
 
                if not self['tag'] and not self['accelerator']:
109
 
                        return False
110
 
 
111
 
                library = SnippetsLibrary()
112
 
                keyval, mod = gtk.accelerator_parse(self['accelerator'])
113
 
                
114
 
                self.valid = library.valid_tab_trigger(self['tag']) and \
115
 
                                (not self['accelerator'] or library.valid_accelerator(keyval, mod))
116
 
        
117
 
        def __getitem__(self, prop):
118
 
                if prop in self.properties:
119
 
                        if self.can_modify():
120
 
                                return self.properties[prop].text or ''
121
 
                        else:
122
 
                                return self.properties[prop] or ''
123
 
                
124
 
                return ''
125
 
        
126
 
        def __setitem__(self, prop, value):
127
 
                if not prop in self.properties:
128
 
                        return
129
 
                        
130
 
                if not self.can_modify() and self.properties[prop] != value:
131
 
                        # ohoh, this is not can_modify, but it needs to be changed...
132
 
                        # make sure it is transfered to the changes file and set all the
133
 
                        # fields.
134
 
                        # This snippet data container will effectively become the container
135
 
                        # for the newly created node, but transparently to whoever uses
136
 
                        # it
137
 
                        
138
 
                        self._override()
139
 
 
140
 
                if self.can_modify() and self.properties[prop].text != value:
141
 
                        if self.library():
142
 
                                self.library().tainted = True
143
 
 
144
 
                        oldvalue = self.properties[prop].text
145
 
                        self.properties[prop].text = value
146
 
                        
147
 
                        if prop == 'tag' or prop == 'accelerator':
148
 
                                container = SnippetsLibrary().container(self.language())
149
 
                                container.prop_changed(self, prop, oldvalue)
150
 
                
151
 
                self.check_validation()
152
 
 
153
 
        def language(self):
154
 
                if self.library and self.library():
155
 
                        return self.library().language
156
 
                else:
157
 
                        return None
158
 
        
159
 
        def is_override(self):
160
 
                return self.override and SnippetsLibrary().overridden[self.override]
161
 
 
162
 
        def _override(self):
163
 
                # Find the user file
164
 
                target = SnippetsLibrary().get_user_library(self.language())
165
 
 
166
 
                # Create a new node there with override
167
 
                element = et.SubElement(target.root, 'snippet', \
168
 
                                {'override': self.id})
169
 
                
170
 
                # Create all the properties
171
 
                for p in self.properties:
172
 
                        prop = et.SubElement(element, p)
173
 
                        prop.text = self.properties[p]
174
 
                        self.properties[p] = prop
175
 
 
176
 
                # Create an override snippet data, feed it element so that it stores
177
 
                # all the values and then set the node to None so that it only contains
178
 
                # the values in .properties
179
 
                override = SnippetData(element, self.library())
180
 
                override.set_node(None)
181
 
                override.id = self.id
182
 
                
183
 
                # Set our node to the new element
184
 
                self.node = element
185
 
                
186
 
                # Set the override to our id
187
 
                self.override = self.id
188
 
                self.id = None
189
 
                
190
 
                # Set the new library
191
 
                self.set_library(target)
192
 
                
193
 
                # The library is tainted because we added this snippet
194
 
                target.tainted = True
195
 
                
196
 
                # Add the override
197
 
                SnippetsLibrary().overridden[self.override] = override
198
 
        
199
 
        def revert(self, snippet):
200
 
                userlib = self.library()
201
 
                self.set_library(snippet.library())
202
 
                
203
 
                userlib.remove(self.node)
204
 
                
205
 
                self.set_node(None)
206
 
 
207
 
                # Copy the properties
208
 
                self.properties = snippet.properties
209
 
                
210
 
                # Set the id
211
 
                self.id = snippet.id
212
 
 
213
 
                # Reset the override flag
214
 
                self.override = None
215
 
 
216
 
class SnippetsTreeBuilder(et.TreeBuilder):
217
 
        def __init__(self, start=None, end=None):
218
 
                et.TreeBuilder.__init__(self)
219
 
                self.set_start(start)
220
 
                self.set_end(end)
221
 
 
222
 
        def set_start(self, start):
223
 
                self._start_cb = start
224
 
        
225
 
        def set_end(self, end):
226
 
                self._end_cb = end
227
 
 
228
 
        def start(self, tag, attrs):
229
 
                result = et.TreeBuilder.start(self, tag, attrs)
230
 
        
231
 
                if self._start_cb:
232
 
                        self._start_cb(result)
233
 
        
234
 
                return result
235
 
                
236
 
        def end(self, tag):
237
 
                result = et.TreeBuilder.end(self, tag)
238
 
        
239
 
                if self._end_cb:
240
 
                        self._end_cb(result)
241
 
        
242
 
                return result
243
 
 
244
 
class LanguageContainer:
245
 
        def __init__(self, language):
246
 
                self.language = language
247
 
                self.snippets = []
248
 
                self.snippets_by_prop = {'tag': {}, 'accelerator': {}}
249
 
                self.accel_group = gtk.AccelGroup()
250
 
                self._refs = 0
251
 
 
252
 
        def _add_prop(self, snippet, prop, value=0):
253
 
                if value == 0:
254
 
                        value = snippet[prop]
255
 
                
256
 
                if not value or value == '':
257
 
                        return
258
 
 
259
 
                snippets_debug('Added ' + prop + ' ' + value + ' to ' + \
260
 
                                str(self.language))
261
 
                
262
 
                if prop == 'accelerator':
263
 
                        keyval, mod = gtk.accelerator_parse(value)
264
 
                        self.accel_group.connect_group(keyval, mod, 0, \
265
 
                                        SnippetsLibrary().accelerator_activated)
266
 
                
267
 
                snippets = self.snippets_by_prop[prop]
268
 
                
269
 
                if value in snippets:
270
 
                        snippets[value].append(snippet)
271
 
                else:
272
 
                        snippets[value] = [snippet]
273
 
 
274
 
        def _remove_prop(self, snippet, prop, value=0):
275
 
                if value == 0:
276
 
                        value = snippet[prop]
277
 
 
278
 
                if not value or value == '':
279
 
                        return
280
 
 
281
 
                snippets_debug('Removed ' + prop + ' ' + value + ' from ' + \
282
 
                                str(self.language))
283
 
 
284
 
                if prop == 'accelerator':
285
 
                        keyval, mod = gtk.accelerator_parse(value)
286
 
                        self.accel_group.disconnect_key(keyval, mod)
287
 
 
288
 
                snippets = self.snippets_by_prop[prop]
289
 
                
290
 
                try:
291
 
                        snippets[value].remove(snippet)
292
 
                except:
293
 
                        True
294
 
                
295
 
        def append(self, snippet):
296
 
                tag = snippet['tag']
297
 
                accelerator = snippet['accelerator']
298
 
                
299
 
                self.snippets.append(snippet)
300
 
                
301
 
                self._add_prop(snippet, 'tag')
302
 
                self._add_prop(snippet, 'accelerator')
303
 
 
304
 
                return snippet
305
 
        
306
 
        def remove(self, snippet):
307
 
                try:
308
 
                        self.snippets.remove(snippet)
309
 
                except:
310
 
                        True
311
 
                        
312
 
                self._remove_prop(snippet, 'tag')
313
 
                self._remove_prop(snippet, 'accelerator')
314
 
        
315
 
        def prop_changed(self, snippet, prop, oldvalue):
316
 
                snippets_debug('PROP CHANGED (', prop, ')', oldvalue)
317
 
 
318
 
                self._remove_prop(snippet, prop, oldvalue)
319
 
                self._add_prop(snippet, prop)
320
 
        
321
 
        def from_prop(self, prop, value):
322
 
                snippets = self.snippets_by_prop[prop]
323
 
                
324
 
                if value in snippets:
325
 
                        return snippets[value]
326
 
                else:
327
 
                        return []
328
 
        
329
 
        def ref(self):
330
 
                self._refs += 1
331
 
        
332
 
                return True
333
 
 
334
 
        def unref(self):
335
 
                if self._refs > 0:
336
 
                        self._refs -= 1
337
 
                
338
 
                return self._refs != 0
339
 
 
340
 
class SnippetsSystemFile:
341
 
        def __init__(self, path=None):
342
 
                self.path = path
343
 
                self.loaded = False
344
 
                self.language = None
345
 
                self.ok = True
346
 
                self.need_id = True
347
 
                
348
 
        def load_error(self, message):
349
 
                sys.stderr.write("An error occurred loading " + self.path + ":\n")
350
 
                sys.stderr.write(message + "\nSnippets in this file will not be " \
351
 
                                "available, please correct or remove the file.\n")
352
 
 
353
 
        def _add_snippet(self, element):
354
 
                if not self.need_id or element.attrib.get('id'):
355
 
                        self.loading_elements.append(element)
356
 
 
357
 
        def set_language(self, element):
358
 
                self.language = element.attrib.get('language')
359
 
                
360
 
                if self.language:
361
 
                        self.language = self.language.lower()
362
 
        
363
 
        def _set_root(self, element):
364
 
                self.set_language(element)
365
 
                
366
 
        def _preprocess_element(self, element):
367
 
                if not self.loaded:
368
 
                        if not element.tag == "snippets":
369
 
                                self.load_error("Root element should be `snippets' instead " \
370
 
                                                "of `%s'" % element.tag)
371
 
                                return False
372
 
                        else:
373
 
                                self._set_root(element)
374
 
                                self.loaded = True
375
 
                elif element.tag != 'snippet' and not self.insnippet:
376
 
                        self.load_error("Element should be `snippet' instead of `%s'" \
377
 
                                        % element.tag)
378
 
                        return False
379
 
                else:
380
 
                        self.insnippet = True
381
 
 
382
 
                return True
383
 
 
384
 
        def _process_element(self, element):
385
 
                if element.tag == 'snippet':
386
 
                        self._add_snippet(element)
387
 
                        self.insnippet = False                  
388
 
 
389
 
                return True
390
 
 
391
 
        def ensure(self):
392
 
                if not self.ok or self.loaded:
393
 
                        return
394
 
                
395
 
                self.load()
396
 
 
397
 
        def parse_xml(self, readsize=16384):
398
 
                if not self.path:
399
 
                        return
400
 
                        
401
 
                elements = []
402
 
 
403
 
                builder = SnippetsTreeBuilder( \
404
 
                                lambda node: elements.append((node, True)), \
405
 
                                lambda node: elements.append((node, False)))
406
 
 
407
 
                parser = et.XMLTreeBuilder(target=builder)
408
 
                self.insnippet = False
409
 
                
410
 
                try:
411
 
                        f = open(self.path, "r")
412
 
                        
413
 
                        while True:
414
 
                                data = f.read(readsize)
415
 
                                
416
 
                                if not data:
417
 
                                        break
418
 
                                
419
 
                                parser.feed(data)
420
 
                                
421
 
                                for element in elements:
422
 
                                        yield element
423
 
                                
424
 
                                del elements[:]
425
 
                        
426
 
                        f.close()
427
 
                except IOError:
428
 
                        self.ok = False
429
 
 
430
 
        def load(self):
431
 
                if not self.ok:
432
 
                        return
433
 
 
434
 
                snippets_debug("Loading library (" + str(self.language) + "): " + \
435
 
                                self.path)
436
 
                
437
 
                self.loaded = False
438
 
                self.ok = False
439
 
                self.loading_elements = []
440
 
                
441
 
                for element in self.parse_xml():
442
 
                        if element[1]:
443
 
                                if not self._preprocess_element(element[0]):
444
 
                                        del self.loading_elements[:]
445
 
                                        return
446
 
                        else:
447
 
                                if not self._process_element(element[0]):
448
 
                                        del self.loading_elements[:]
449
 
                                        return
450
 
 
451
 
                for element in self.loading_elements:
452
 
                        snippet = SnippetsLibrary().add_snippet(self, element)
453
 
                
454
 
                del self.loading_elements[:]
455
 
                self.ok = True
456
 
 
457
 
        # This function will get the language for a file by just inspecting the
458
 
        # root element of the file. This is provided so that a cache can be built
459
 
        # for which file contains which language.
460
 
        # It returns the name of the language
461
 
        def ensure_language(self):
462
 
                if not self.loaded:
463
 
                        self.ok = False
464
 
                        
465
 
                        for element in self.parse_xml(256):
466
 
                                if element[1]:
467
 
                                        if element[0].tag == 'snippets':
468
 
                                                self.set_language(element[0])
469
 
                                                self.ok = True
470
 
 
471
 
                                        break
472
 
        
473
 
        def unload(self):
474
 
                snippets_debug("Unloading library (" + str(self.language) + "): " + \
475
 
                                self.path)
476
 
                self.language = None
477
 
                self.loaded = False
478
 
                self.ok = True
479
 
 
480
 
class SnippetsUserFile(SnippetsSystemFile):
481
 
        def __init__(self, path=None):
482
 
                SnippetsSystemFile.__init__(self, path)
483
 
                self.tainted = False
484
 
                self.need_id = False
485
 
                
486
 
        def _set_root(self, element):
487
 
                SnippetsSystemFile._set_root(self, element)
488
 
                self.root = element
489
 
                        
490
 
        def add_prop(self, node, tag, data):
491
 
                if data[tag]:
492
 
                        prop = et.SubElement(node, tag)
493
 
                        prop.text = data[tag]
494
 
                
495
 
                        return prop
496
 
                else:
497
 
                        return None
498
 
 
499
 
        def new_snippet(self, properties=None):
500
 
                if (not self.ok) or self.root == None:
501
 
                        return None
502
 
                
503
 
                element = et.SubElement(self.root, 'snippet')
504
 
                
505
 
                if properties:
506
 
                        for prop in properties:
507
 
                                sub = et.SubElement(element, prop)
508
 
                                sub.text = properties[prop]
509
 
                
510
 
                self.tainted = True
511
 
                
512
 
                return SnippetsLibrary().add_snippet(self, element)
513
 
        
514
 
        def set_language(self, element):
515
 
                SnippetsSystemFile.set_language(self, element)
516
 
                
517
 
                filename = os.path.basename(self.path).lower()
518
 
                
519
 
                if not self.language and filename == "global.xml":
520
 
                        self.modifier = True
521
 
                elif self.language and filename == self.language + ".xml":
522
 
                        self.modifier = True
523
 
                else:
524
 
                        self.modifier = False
525
 
        
526
 
        def create_root(self, language):
527
 
                if self.loaded:
528
 
                        snippets_debug('Not creating root, already loaded')
529
 
                        return
530
 
                
531
 
                if language:
532
 
                        root = et.Element('snippets', {'language': language})
533
 
                        self.path = os.path.join(SnippetsLibrary().userdir, language.lower() + '.xml')
534
 
                else:
535
 
                        root = et.Element('snippets')
536
 
                        self.path = os.path.join(SnippetsLibrary().userdir, 'global.xml')
537
 
                
538
 
                self._set_root(root)
539
 
                self.loaded = True
540
 
                self.ok = True
541
 
                self.tainted = True
542
 
                self.save()
543
 
        
544
 
        def remove(self, element):
545
 
                try:
546
 
                        self.root.remove(element)
547
 
                        self.tainted = True
548
 
                except:
549
 
                        return
550
 
                
551
 
                try:
552
 
                        first = self.root[0]
553
 
                except:
554
 
                        # No more elements, this library is useless now
555
 
                        SnippetsLibrary().remove_library(self)
556
 
        
557
 
        def save(self):
558
 
                if not self.ok or self.root == None or not self.tainted:
559
 
                        return
560
 
 
561
 
                path = os.path.dirname(self.path)
562
 
                
563
 
                try:
564
 
                        if not os.path.isdir(path):
565
 
                                os.makedirs(path, 0755)
566
 
                except OSError:
567
 
                        # TODO: this is bad...
568
 
                        sys.stderr.write("Error in making dirs\n")
569
 
 
570
 
                try:
571
 
                        write_xml(self.root, self.path, ('text', 'accelerator'))
572
 
                        self.tainted = False
573
 
                except IOError:
574
 
                        # Couldn't save, what to do
575
 
                        sys.stderr.write("Could not save user snippets file to " + \
576
 
                                        self.path + "\n")
577
 
        
578
 
        def unload(self):
579
 
                SnippetsSystemFile.unload(self)
580
 
                self.root = None
581
 
 
582
 
class SnippetsLibraryImpl:
583
 
        def __init__(self):
584
 
                self._accelerator_activated_cb = None
585
 
                self.loaded = False
586
 
                self.check_buffer = gtk.TextBuffer()
587
 
                        
588
 
        def set_dirs(self, userdir, systemdirs):
589
 
                self.userdir = userdir
590
 
                self.systemdirs = systemdirs
591
 
                
592
 
                self.libraries = {}
593
 
                self.containers = {}
594
 
                self.overridden = {}
595
 
                self.loaded_ids = []
596
 
 
597
 
                self.loaded = False
598
 
        
599
 
        def set_accelerator_callback(self, cb):
600
 
                self._accelerator_activated_cb = cb
601
 
        
602
 
        def accelerator_activated(self, group, obj, keyval, mod):
603
 
                if self._accelerator_activated_cb:
604
 
                        self._accelerator_activated_cb(group, obj, keyval, mod)
605
 
 
606
 
        def add_snippet(self, library, element):
607
 
                container = self.container(library.language)
608
 
                overrided = self.overrided(library, element)
609
 
                
610
 
                if overrided:
611
 
                        overrided.set_library(library)
612
 
                        snippets_debug('Snippet is overriden: ' + overrided['description'])
613
 
                        return None
614
 
                
615
 
                snippet = SnippetData(element, library)
616
 
                
617
 
                if snippet.id in self.loaded_ids:
618
 
                        snippets_debug('Not added snippet ' + str(library.language) + \
619
 
                                        '::' + snippet['description'] + ' (duplicate)')
620
 
                        return None
621
 
 
622
 
                snippet = container.append(snippet)
623
 
                snippets_debug('Added snippet ' + str(library.language) + '::' + \
624
 
                                snippet['description'])
625
 
                
626
 
                if snippet and snippet.override:
627
 
                        self.add_override(snippet)
628
 
                
629
 
                if snippet.id:
630
 
                        self.loaded_ids.append(snippet.id)
631
 
 
632
 
                return snippet
633
 
        
634
 
        def container(self, language):
635
 
                language = self.normalize_language(language)
636
 
                
637
 
                if not language in self.containers:
638
 
                        self.containers[language] = LanguageContainer(language)
639
 
                
640
 
                return self.containers[language]
641
 
        
642
 
        def get_user_library(self, language):
643
 
                target = None
644
 
                
645
 
                if language in self.libraries:
646
 
                        for library in self.libraries[language]:
647
 
                                if isinstance(library, SnippetsUserFile) and library.modifier:
648
 
                                        target = library
649
 
                                elif not isinstance(library, SnippetsUserFile):
650
 
                                        break
651
 
                
652
 
                if not target:
653
 
                        # Create a new user file then
654
 
                        snippets_debug('Creating a new user file for language ' + \
655
 
                                        str(language))
656
 
                        target = SnippetsUserFile()
657
 
                        target.create_root(language)
658
 
                        self.add_library(target)
659
 
        
660
 
                return target
661
 
        
662
 
        def new_snippet(self, language, properties=None):
663
 
                language = self.normalize_language(language)
664
 
                library = self.get_user_library(language)
665
 
 
666
 
                return library.new_snippet(properties)
667
 
        
668
 
        def revert_snippet(self, snippet):
669
 
                # This will revert the snippet to the one it overrides
670
 
                if not snippet.can_modify() or not snippet.override in self.overridden:
671
 
                        # It can't be reverted, shouldn't happen, but oh..
672
 
                        return
673
 
                
674
 
                # The snippet in self.overriden only contains the property contents and
675
 
                # the library it belongs to
676
 
                revertto = self.overridden[snippet.override]
677
 
                del self.overridden[snippet.override]
678
 
                
679
 
                if revertto:
680
 
                        snippet.revert(revertto)
681
 
                
682
 
                        if revertto.id:
683
 
                                self.loaded_ids.append(revertto.id)
684
 
        
685
 
        def remove_snippet(self, snippet):
686
 
                if not snippet.can_modify() or snippet.is_override():
687
 
                        return
688
 
                
689
 
                # Remove from the library
690
 
                userlib = snippet.library()
691
 
                userlib.remove(snippet.node)
692
 
                
693
 
                # Remove from the container
694
 
                container = self.containers[userlib.language]
695
 
                container.remove(snippet)
696
 
        
697
 
        def overrided(self, library, element):
698
 
                id = NamespacedId(library.language, element.attrib.get('id')).id
699
 
                
700
 
                if id in self.overridden:
701
 
                        snippet = SnippetData(element, None)
702
 
                        snippet.set_node(None)
703
 
                        
704
 
                        self.overridden[id] = snippet
705
 
                        return snippet
706
 
                else:
707
 
                        return None
708
 
        
709
 
        def add_override(self, snippet):
710
 
                snippets_debug('Add override:', snippet.override)
711
 
                
712
 
                if not snippet.override in self.overridden:
713
 
                        self.overridden[snippet.override] = None
714
 
        
715
 
        def add_library(self, library):
716
 
                library.ensure_language()
717
 
                
718
 
                if not library.ok:
719
 
                        snippets_debug('Library in wrong format, ignoring')
720
 
                        return
721
 
                
722
 
                snippets_debug('Adding library (' + str(library.language) + '): ' + \
723
 
                                library.path)
724
 
 
725
 
                if library.language in self.libraries:
726
 
                        # Make sure all the user files are before the system files
727
 
                        if isinstance(library, SnippetsUserFile):
728
 
                                self.libraries[library.language].insert(0, library)
729
 
                        else:
730
 
                                self.libraries[library.language].append(library)
731
 
                else:
732
 
                        self.libraries[library.language] = [library]
733
 
        
734
 
        def remove_library(self, library):
735
 
                if not library.ok:
736
 
                        return
737
 
                
738
 
                if library.path and os.path.isfile(library.path):
739
 
                        os.unlink(library.path)
740
 
                
741
 
                try:
742
 
                        self.libraries[library.language].remove(library)
743
 
                except KeyError:
744
 
                        True
745
 
                        
746
 
                container = self.containers[library.language]
747
 
                        
748
 
                for snippet in list(container.snippets):
749
 
                        if snippet.library() == library:
750
 
                                container.remove(snippet)
751
 
        
752
 
        def _add_user_library(self, path):
753
 
                library = SnippetsUserFile(path)
754
 
                self.add_library(library)
755
 
                
756
 
        def _add_system_library(self, path):
757
 
                library = SnippetsSystemFile(path)
758
 
                self.add_library(library)
759
 
 
760
 
        def find_libraries(self, path, searched, addcb):
761
 
                snippets_debug("Finding in: " + path)
762
 
                
763
 
                if not os.path.isdir(path):
764
 
                        return searched
765
 
 
766
 
                files = os.listdir(path)
767
 
                searched.append(path)
768
 
                
769
 
                for f in files:
770
 
                        f = os.path.realpath(os.path.join(path, f))
771
 
 
772
 
                        # Determine what language this file provides snippets for
773
 
                        if os.path.isfile(f):
774
 
                                addcb(f)
775
 
                        elif os.path.isdir(f) and not f in searched:
776
 
                                # Recursion rules!
777
 
                                searched = self.find_libraries(self, f, searched, addcb)
778
 
                
779
 
                return searched
780
 
        
781
 
        def normalize_language(self, language):
782
 
                if language:
783
 
                        return language.lower()
784
 
                
785
 
                return language
786
 
        
787
 
        def remove_container(self, language):
788
 
                for snippet in self.containers[language].snippets:
789
 
                        if snippet.id in self.loaded_ids:
790
 
                                self.loaded_ids.remove(snippet.id)
791
 
 
792
 
                        if snippet.override in self.overridden:
793
 
                                del self.overridden[snippet.override]
794
 
 
795
 
                del self.containers[language]
796
 
                
797
 
        def get_accel_group(self, language):
798
 
                language = self.normalize_language(language)
799
 
                
800
 
                container = self.container(language)
801
 
                return container.accel_group
802
 
                
803
 
        def save(self, language):
804
 
                language = self.normalize_language(language)
805
 
                
806
 
                if language in self.libraries:
807
 
                        for library in self.libraries[language]:
808
 
                                if isinstance(library, SnippetsUserFile):
809
 
                                        library.save()
810
 
                                else:
811
 
                                        break
812
 
        
813
 
        def ref(self, language):
814
 
                language = self.normalize_language(language)
815
 
 
816
 
                snippets_debug('Ref:', language)
817
 
                self.container(language).ref()
818
 
        
819
 
        def unref(self, language):
820
 
                language = self.normalize_language(language)
821
 
                
822
 
                snippets_debug('Unref:', language)
823
 
                
824
 
                if language in self.containers:
825
 
                        if not self.containers[language].unref() and \
826
 
                                        language in self.libraries:
827
 
 
828
 
                                for library in self.libraries[language]:
829
 
                                        library.unload()
830
 
                                
831
 
                                self.remove_container(language)
832
 
 
833
 
        def ensure(self, language):
834
 
                language = self.normalize_language(language)
835
 
 
836
 
                # Ensure language as well as the global snippets (None)
837
 
                for lang in (None, language):
838
 
                        if lang in self.libraries:
839
 
                                # Ensure the container exists
840
 
                                self.container(lang)
841
 
 
842
 
                                for library in self.libraries[lang]:
843
 
                                        library.ensure()
844
 
 
845
 
        def ensure_files(self):
846
 
                if self.loaded:
847
 
                        return
848
 
 
849
 
                searched = []
850
 
                searched = self.find_libraries(self.userdir, searched, \
851
 
                                self._add_user_library)
852
 
                
853
 
                for d in self.systemdirs:
854
 
                        searched = self.find_libraries(d, searched, \
855
 
                                        self._add_system_library)
856
 
 
857
 
                self.loaded = True
858
 
 
859
 
        def valid_accelerator(self, keyval, mod):
860
 
                mod &= gtk.accelerator_get_default_mod_mask()
861
 
        
862
 
                return (mod and (gdk.keyval_to_unicode(keyval) or \
863
 
                                keyval in range(gtk.keysyms.F1, gtk.keysyms.F12 + 1)))
864
 
        
865
 
        def valid_tab_trigger(self, trigger):
866
 
                if not trigger:
867
 
                        return True
868
 
 
869
 
                if trigger.isdigit():
870
 
                        return False
871
 
 
872
 
                self.check_buffer.set_text(trigger)
873
 
 
874
 
                start, end = self.check_buffer.get_bounds()
875
 
                text = self.check_buffer.get_text(start, end)
876
 
                                
877
 
                s = start.copy()
878
 
                e = end.copy()
879
 
                
880
 
                end.backward_word_start()
881
 
                start.forward_word_end()
882
 
                
883
 
                return (s.equal(end) and e.equal(start)) or (len(text) == 1 and not (text.isalnum() or text.isspace()))
884
 
 
885
 
        # Snippet getters
886
 
        # ===============
887
 
        
888
 
        # Get snippets for a given language
889
 
        def get_snippets(self, language=None):
890
 
                self.ensure_files()
891
 
                language = self.normalize_language(language)
892
 
                
893
 
                if not language in self.libraries:
894
 
                        return []
895
 
                
896
 
                snippets = []
897
 
                self.ensure(language)
898
 
                
899
 
                return list(self.containers[language].snippets)
900
 
 
901
 
        # Get snippets for a given accelerator
902
 
        def from_accelerator(self, accelerator, language=None):
903
 
                self.ensure_files()
904
 
                
905
 
                result = []             
906
 
                language = self.normalize_language(language)
907
 
                        
908
 
                if not language in self.containers:
909
 
                        return []
910
 
 
911
 
                self.ensure(language)
912
 
                result = self.containers[language].from_prop('accelerator', accelerator)
913
 
                
914
 
                if len(result) == 0 and language and None in self.containers:
915
 
                        result = self.containers[None].from_prop('accelerator', accelerator)
916
 
                
917
 
                return result
918
 
 
919
 
        # Get snippets for a given tag
920
 
        def from_tag(self, tag, language=None):
921
 
                self.ensure_files()
922
 
                
923
 
                result = []
924
 
                language = self.normalize_language(language)
925
 
                        
926
 
                if not language in self.containers:
927
 
                        return []
928
 
 
929
 
                self.ensure(language)
930
 
                result = self.containers[language].from_prop('tag', tag)
931
 
                
932
 
                if len(result) == 0 and language and None in self.containers:
933
 
                        result = self.containers[None].from_prop('tag', tag)
934
 
                
935
 
                return result
936
 
 
937
 
class SnippetsLibrary:
938
 
        __instance = None
939
 
        
940
 
        def __init__(self):
941
 
                if not SnippetsLibrary.__instance:
942
 
                        SnippetsLibrary.__instance = SnippetsLibraryImpl()
943
 
        
944
 
                self.__dict__['_SnippetsLibrary__instance'] = SnippetsLibrary.__instance
945
 
                
946
 
        def __getattr__(self, attr):
947
 
                return getattr(self.__instance, attr)
948
 
        
949
 
        def __setattr__(self, attr, value):
950
 
                return setattr(self.__instance, attr, value)