/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/commands.py

  • Committer: mbp at sourcefrog
  • Date: 2005-03-29 08:01:30 UTC
  • Revision ID: mbp@sourcefrog.net-20050329080130-89b78af279b196b3
match ignore patterns like ./config.h

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2004, 2005 by Martin Pool
 
2
# Copyright (C) 2005 by Canonical Ltd
 
3
 
 
4
 
 
5
# This program is free software; you can redistribute it and/or modify
 
6
# it under the terms of the GNU General Public License as published by
 
7
# the Free Software Foundation; either version 2 of the License, or
 
8
# (at your option) any later version.
 
9
 
 
10
# This program is distributed in the hope that it will be useful,
 
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
13
# GNU General Public License for more details.
 
14
 
 
15
# You should have received a copy of the GNU General Public License
 
16
# along with this program; if not, write to the Free Software
 
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
18
 
 
19
"""Bazaar-NG -- a free distributed version-control tool
 
20
 
 
21
**WARNING: THIS IS AN UNSTABLE DEVELOPMENT VERSION**
 
22
 
 
23
Current limitation include:
 
24
 
 
25
* Metadata format is not stable yet -- you may need to
 
26
  discard history in the future.
 
27
 
 
28
* Insufficient error handling.
 
29
 
 
30
* Many commands unimplemented or partially implemented.
 
31
 
 
32
* Space-inefficient storage.
 
33
 
 
34
* No merge operators yet.
 
35
 
 
36
Interesting commands::
 
37
 
 
38
  bzr help [COMMAND]
 
39
       Show help screen
 
40
  bzr version
 
41
       Show software version/licence/non-warranty.
 
42
  bzr init
 
43
       Start versioning the current directory
 
44
  bzr add FILE...
 
45
       Make files versioned.
 
46
  bzr log
 
47
       Show revision history.
 
48
  bzr diff
 
49
       Show changes from last revision to working copy.
 
50
  bzr commit -m 'MESSAGE'
 
51
       Store current state as new revision.
 
52
  bzr export REVNO DESTINATION
 
53
       Export the branch state at a previous version.
 
54
  bzr status
 
55
       Show summary of pending changes.
 
56
  bzr remove FILE...
 
57
       Make a file not versioned.
 
58
  bzr info
 
59
       Show statistics about this branch.
 
60
"""
 
61
 
 
62
 
 
63
 
 
64
 
 
65
import sys, os, random, time, sha, sets, types, re, shutil, tempfile
 
66
import traceback, socket, fnmatch, difflib
 
67
from os import path
 
68
from sets import Set
 
69
from pprint import pprint
 
70
from stat import *
 
71
from glob import glob
 
72
 
 
73
import bzrlib
 
74
from bzrlib.store import ImmutableStore
 
75
from bzrlib.trace import mutter, note, log_error
 
76
from bzrlib.errors import bailout, BzrError
 
77
from bzrlib.osutils import quotefn, pumpfile, isdir, isfile
 
78
from bzrlib.tree import RevisionTree, EmptyTree, WorkingTree, Tree
 
79
from bzrlib.revision import Revision
 
80
from bzrlib import Branch, Inventory, InventoryEntry, ScratchBranch, BZRDIR, \
 
81
     format_date
 
82
 
 
83
BZR_DIFF_FORMAT = "## Bazaar-NG diff, format 0 ##\n"
 
84
BZR_PATCHNAME_FORMAT = 'cset:sha1:%s'
 
85
 
 
86
## standard representation
 
87
NONE_STRING = '(none)'
 
88
EMPTY = 'empty'
 
89
 
 
90
 
 
91
## TODO: Perhaps a different version of inventory commands that
 
92
## returns iterators...
 
93
 
 
94
## TODO: Perhaps an AtomicFile class that writes to a temporary file and then renames.
 
95
 
 
96
## TODO: Some kind of locking on branches.  Perhaps there should be a
 
97
## parameter to the branch object saying whether we want a read or
 
98
## write lock; release it from destructor.  Perhaps don't even need a
 
99
## read lock to look at immutable objects?
 
100
 
 
101
## TODO: Perhaps make UUIDs predictable in test mode to make it easier
 
102
## to compare output?
 
103
 
 
104
## TODO: Some kind of global code to generate the right Branch object
 
105
## to work on.  Almost, but not quite all, commands need one, and it
 
106
## can be taken either from their parameters or their working
 
107
## directory.
 
108
 
 
109
## TODO: rename command, needed soon: check destination doesn't exist
 
110
## either in working copy or tree; move working copy; update
 
111
## inventory; write out
 
112
 
 
113
## TODO: move command; check destination is a directory and will not
 
114
## clash; move it.
 
115
 
 
116
## TODO: command to show renames, one per line, as to->from
 
117
 
 
118
 
 
119
 
 
120
 
 
121
def cmd_status(all=False):
 
122
    """Display status summary.
 
123
 
 
124
    For each file there is a single line giving its file state and name.
 
125
    The name is that in the current revision unless it is deleted or
 
126
    missing, in which case the old name is shown.
 
127
 
 
128
    :todo: Don't show unchanged files unless ``--all`` is given?
 
129
    """
 
130
    Branch('.').show_status(show_all=all)
 
131
 
 
132
 
 
133
 
 
134
######################################################################
 
135
# examining history
 
136
def cmd_get_revision(revision_id):
 
137
    Branch('.').get_revision(revision_id).write_xml(sys.stdout)
 
138
 
 
139
 
 
140
def cmd_get_file_text(text_id):
 
141
    """Get contents of a file by hash."""
 
142
    sf = Branch('.').text_store[text_id]
 
143
    pumpfile(sf, sys.stdout)
 
144
 
 
145
 
 
146
 
 
147
######################################################################
 
148
# commands
 
149
    
 
150
 
 
151
def cmd_revno():
 
152
    """Show number of revisions on this branch"""
 
153
    print Branch('.').revno()
 
154
    
 
155
 
 
156
    
 
157
def cmd_add(file_list, verbose=False):
 
158
    """Add specified files or directories.
 
159
 
 
160
    In non-recursive mode, all the named items are added, regardless
 
161
    of whether they were previously ignored.  A warning is given if
 
162
    any of the named files are already versioned.
 
163
 
 
164
    In recursive mode (the default), files are treated the same way
 
165
    but the behaviour for directories is different.  Directories that
 
166
    are already versioned do not give a warning.  All directories,
 
167
    whether already versioned or not, are searched for files or
 
168
    subdirectories that are neither versioned or ignored, and these
 
169
    are added.  This search proceeds recursively into versioned
 
170
    directories.
 
171
 
 
172
    Therefore simply saying 'bzr add .' will version all files that
 
173
    are currently unknown.
 
174
    """
 
175
    bzrlib.add.smart_add(file_list, verbose)
 
176
    
 
177
 
 
178
def cmd_relpath(filename):
 
179
    """Show path of file relative to root"""
 
180
    print Branch(filename).relpath(filename)
 
181
 
 
182
 
 
183
def cmd_inventory(revision=None):
 
184
    """Show inventory of the current working copy."""
 
185
    ## TODO: Also optionally show a previous inventory
 
186
    ## TODO: Format options
 
187
    b = Branch('.')
 
188
    if revision == None:
 
189
        inv = b.read_working_inventory()
 
190
    else:
 
191
        inv = b.get_revision_inventory(b.lookup_revision(revision))
 
192
        
 
193
    for path, entry in inv.iter_entries():
 
194
        print '%-50s %s' % (entry.file_id, path)
 
195
 
 
196
 
 
197
 
 
198
def cmd_info():
 
199
    """info: Show statistical information for this branch
 
200
 
 
201
usage: bzr info"""
 
202
    import info
 
203
    info.show_info(Branch('.'))        
 
204
    
 
205
 
 
206
 
 
207
def cmd_remove(file_list, verbose=False):
 
208
    b = Branch(file_list[0])
 
209
    b.remove([b.relpath(f) for f in file_list], verbose=verbose)
 
210
 
 
211
 
 
212
 
 
213
def cmd_file_id(filename):
 
214
    b = Branch(filename)
 
215
    i = b.inventory.path2id(b.relpath(filename))
 
216
    if i is None:
 
217
        bailout("%s is not a versioned file" % filename)
 
218
    else:
 
219
        print i
 
220
 
 
221
 
 
222
def cmd_find_filename(fileid):
 
223
    n = find_filename(fileid)
 
224
    if n is None:
 
225
        bailout("%s is not a live file id" % fileid)
 
226
    else:
 
227
        print n
 
228
 
 
229
 
 
230
def cmd_revision_history():
 
231
    for patchid in Branch('.').revision_history():
 
232
        print patchid
 
233
 
 
234
 
 
235
 
 
236
def cmd_init():
 
237
    # TODO: Check we're not already in a working directory?  At the
 
238
    # moment you'll get an ugly error.
 
239
    
 
240
    # TODO: What if we're in a subdirectory of a branch?  Would like
 
241
    # to allow that, but then the parent may need to understand that
 
242
    # the children have disappeared, or should they be versioned in
 
243
    # both?
 
244
 
 
245
    # TODO: Take an argument/option for branch name.
 
246
    Branch('.', init=True)
 
247
 
 
248
 
 
249
def cmd_diff(revision=None):
 
250
    """bzr diff: Show differences in working tree.
 
251
    
 
252
usage: bzr diff [-r REV]
 
253
 
 
254
--revision REV
 
255
    Show changes since REV, rather than predecessor.
 
256
 
 
257
TODO: Given two revision arguments, show the difference between them.
 
258
 
 
259
TODO: Allow diff across branches.
 
260
 
 
261
TODO: Option to use external diff command; could be GNU diff, wdiff,
 
262
or a graphical diff.
 
263
 
 
264
TODO: Diff selected files.
 
265
"""
 
266
 
 
267
    ## TODO: Shouldn't be in the cmd function.
 
268
 
 
269
    b = Branch('.')
 
270
 
 
271
    if revision == None:
 
272
        old_tree = b.basis_tree()
 
273
    else:
 
274
        old_tree = b.revision_tree(b.lookup_revision(revision))
 
275
        
 
276
    new_tree = b.working_tree()
 
277
    old_inv = old_tree.inventory
 
278
    new_inv = new_tree.inventory
 
279
 
 
280
    # TODO: Options to control putting on a prefix or suffix, perhaps as a format string
 
281
    old_label = ''
 
282
    new_label = ''
 
283
 
 
284
    DEVNULL = '/dev/null'
 
285
    # Windows users, don't panic about this filename -- it is a
 
286
    # special signal to GNU patch that the file should be created or
 
287
    # deleted respectively.
 
288
 
 
289
    # TODO: Generation of pseudo-diffs for added/deleted files could
 
290
    # be usefully made into a much faster special case.
 
291
 
 
292
    # TODO: Better to return them in sorted order I think.
 
293
    
 
294
    for file_state, fid, old_name, new_name, kind in bzrlib.diff_trees(old_tree, new_tree):
 
295
        d = None
 
296
 
 
297
        # Don't show this by default; maybe do it if an option is passed
 
298
        # idlabel = '      {%s}' % fid
 
299
        idlabel = ''
 
300
 
 
301
        # FIXME: Something about the diff format makes patch unhappy
 
302
        # with newly-added files.
 
303
 
 
304
        def diffit(*a, **kw):
 
305
            sys.stdout.writelines(difflib.unified_diff(*a, **kw))
 
306
            print
 
307
        
 
308
        if file_state in ['.', '?', 'I']:
 
309
            continue
 
310
        elif file_state == 'A':
 
311
            print '*** added %s %r' % (kind, new_name)
 
312
            if kind == 'file':
 
313
                diffit([],
 
314
                       new_tree.get_file(fid).readlines(),
 
315
                       fromfile=DEVNULL,
 
316
                       tofile=new_label + new_name + idlabel)
 
317
        elif file_state == 'D':
 
318
            assert isinstance(old_name, types.StringTypes)
 
319
            print '*** deleted %s %r' % (kind, old_name)
 
320
            if kind == 'file':
 
321
                diffit(old_tree.get_file(fid).readlines(), [],
 
322
                       fromfile=old_label + old_name + idlabel,
 
323
                       tofile=DEVNULL)
 
324
        elif file_state in ['M', 'R']:
 
325
            if file_state == 'M':
 
326
                assert kind == 'file'
 
327
                assert old_name == new_name
 
328
                print '*** modified %s %r' % (kind, new_name)
 
329
            elif file_state == 'R':
 
330
                print '*** renamed %s %r => %r' % (kind, old_name, new_name)
 
331
 
 
332
            if kind == 'file':
 
333
                diffit(old_tree.get_file(fid).readlines(),
 
334
                       new_tree.get_file(fid).readlines(),
 
335
                       fromfile=old_label + old_name + idlabel,
 
336
                       tofile=new_label + new_name)
 
337
        else:
 
338
            bailout("can't represent state %s {%s}" % (file_state, fid))
 
339
 
 
340
 
 
341
 
 
342
def cmd_deleted(show_ids=False):
 
343
    """List files deleted in the working tree.
 
344
 
 
345
TODO: Show files deleted since a previous revision, or between two revisions.
 
346
    """
 
347
    b = Branch('.')
 
348
    old = b.basis_tree()
 
349
    new = b.working_tree()
 
350
 
 
351
    for path, ie in old.inventory.iter_entries():
 
352
        if not new.has_id(ie.file_id):
 
353
            if show_ids:
 
354
                print '%-50s %s' % (path, ie.file_id)
 
355
            else:
 
356
                print path
 
357
                
 
358
    
 
359
 
 
360
 
 
361
def cmd_root(filename=None):
 
362
    """Print the branch root."""
 
363
    print bzrlib.branch.find_branch_root(filename)
 
364
    
 
365
 
 
366
def cmd_log(timezone='original'):
 
367
    """Show log of this branch.
 
368
 
 
369
    :todo: Options for utc; to show ids; to limit range; etc.
 
370
    """
 
371
    Branch('.').write_log(show_timezone=timezone)
 
372
 
 
373
 
 
374
def cmd_ls(revision=None, verbose=False):
 
375
    """List files in a tree.
 
376
 
 
377
    :todo: Take a revision or remote path and list that tree instead.
 
378
    """
 
379
    b = Branch('.')
 
380
    if revision == None:
 
381
        tree = b.working_tree()
 
382
    else:
 
383
        tree = b.revision_tree(b.lookup_revision(revision))
 
384
        
 
385
    for fp, fc, kind, fid in tree.list_files():
 
386
        if verbose:
 
387
            if kind == 'directory':
 
388
                kindch = '/'
 
389
            elif kind == 'file':
 
390
                kindch = ''
 
391
            else:
 
392
                kindch = '???'
 
393
                
 
394
            print '%-8s %s%s' % (fc, fp, kindch)
 
395
        else:
 
396
            print fp
 
397
    
 
398
    
 
399
 
 
400
def cmd_unknowns():
 
401
    """List unknown files"""
 
402
    for f in Branch('.').unknowns():
 
403
        print quotefn(f)
 
404
 
 
405
 
 
406
 
 
407
def cmd_ignored(verbose=True):
 
408
    """List ignored files and the patterns that matched them.
 
409
      """
 
410
    tree = Branch('.').working_tree()
 
411
    for path, file_class, kind, id in tree.list_files():
 
412
        if file_class != 'I':
 
413
            continue
 
414
        ## XXX: Slightly inefficient since this was already calculated
 
415
        pat = tree.is_ignored(path)
 
416
        print '%-50s %s' % (path, pat)
 
417
 
 
418
 
 
419
def cmd_lookup_revision(revno):
 
420
    try:
 
421
        revno = int(revno)
 
422
    except ValueError:
 
423
        bailout("usage: lookup-revision REVNO",
 
424
                ["REVNO is a non-negative revision number for this branch"])
 
425
 
 
426
    print Branch('.').lookup_revision(revno) or NONE_STRING
 
427
 
 
428
 
 
429
 
 
430
def cmd_export(revno, dest):
 
431
    """Export past revision to destination directory."""
 
432
    b = Branch('.')
 
433
    rh = b.lookup_revision(int(revno))
 
434
    t = b.revision_tree(rh)
 
435
    t.export(dest)
 
436
 
 
437
 
 
438
 
 
439
######################################################################
 
440
# internal/test commands
 
441
 
 
442
 
 
443
def cmd_uuid():
 
444
    """Print a newly-generated UUID."""
 
445
    print bzrlib.osutils.uuid()
 
446
 
 
447
 
 
448
 
 
449
def cmd_local_time_offset():
 
450
    print bzrlib.osutils.local_time_offset()
 
451
 
 
452
 
 
453
 
 
454
def cmd_commit(message=None, verbose=False):
 
455
    """Commit changes to a new revision.
 
456
 
 
457
--message MESSAGE
 
458
    Description of changes in this revision; free form text.
 
459
    It is recommended that the first line be a single-sentence
 
460
    summary.
 
461
--verbose
 
462
    Show status of changed files,
 
463
 
 
464
TODO: Commit only selected files.
 
465
 
 
466
TODO: Run hooks on tree to-be-committed, and after commit.
 
467
 
 
468
TODO: Strict commit that fails if there are unknown or deleted files.
 
469
"""
 
470
 
 
471
    if not message:
 
472
        bailout("please specify a commit message")
 
473
    Branch('.').commit(message, verbose=verbose)
 
474
 
 
475
 
 
476
def cmd_check(dir='.'):
 
477
    """check: Consistency check of branch history.
 
478
 
 
479
usage: bzr check [-v] [BRANCH]
 
480
 
 
481
options:
 
482
  --verbose, -v         Show progress of checking.
 
483
 
 
484
This command checks various invariants about the branch storage to
 
485
detect data corruption or bzr bugs.
 
486
"""
 
487
    import bzrlib.check
 
488
    bzrlib.check.check(Branch(dir, find_root=False))
 
489
 
 
490
 
 
491
def cmd_is(pred, *rest):
 
492
    """Test whether PREDICATE is true."""
 
493
    try:
 
494
        cmd_handler = globals()['assert_' + pred.replace('-', '_')]
 
495
    except KeyError:
 
496
        bailout("unknown predicate: %s" % quotefn(pred))
 
497
        
 
498
    try:
 
499
        cmd_handler(*rest)
 
500
    except BzrCheckError:
 
501
        # by default we don't print the message so that this can
 
502
        # be used from shell scripts without producing noise
 
503
        sys.exit(1)
 
504
 
 
505
 
 
506
def cmd_username():
 
507
    print bzrlib.osutils.username()
 
508
 
 
509
 
 
510
def cmd_user_email():
 
511
    print bzrlib.osutils.user_email()
 
512
 
 
513
 
 
514
def cmd_gen_revision_id():
 
515
    import time
 
516
    print bzrlib.branch._gen_revision_id(time.time())
 
517
 
 
518
 
 
519
def cmd_selftest(verbose=False):
 
520
    """Run internal test suite"""
 
521
    ## -v, if present, is seen by doctest; the argument is just here
 
522
    ## so our parser doesn't complain
 
523
 
 
524
    ## TODO: --verbose option
 
525
 
 
526
    failures, tests = 0, 0
 
527
    
 
528
    import doctest, bzrlib.store, bzrlib.tests
 
529
    bzrlib.trace.verbose = False
 
530
 
 
531
    for m in bzrlib.store, bzrlib.inventory, bzrlib.branch, bzrlib.osutils, \
 
532
        bzrlib.tree, bzrlib.tests, bzrlib.commands, bzrlib.add:
 
533
        mf, mt = doctest.testmod(m)
 
534
        failures += mf
 
535
        tests += mt
 
536
        print '%-40s %3d tests' % (m.__name__, mt),
 
537
        if mf:
 
538
            print '%3d FAILED!' % mf
 
539
        else:
 
540
            print
 
541
 
 
542
    print '%-40s %3d tests' % ('total', tests),
 
543
    if failures:
 
544
        print '%3d FAILED!' % failures
 
545
    else:
 
546
        print
 
547
 
 
548
 
 
549
 
 
550
# deprecated
 
551
cmd_doctest = cmd_selftest
 
552
 
 
553
 
 
554
######################################################################
 
555
# help
 
556
 
 
557
 
 
558
def cmd_help(topic=None):
 
559
    if topic == None:
 
560
        print __doc__
 
561
        return
 
562
 
 
563
    # otherwise, maybe the name of a command?
 
564
    try:
 
565
        cmdfn = globals()['cmd_' + topic.replace('-', '_')]
 
566
    except KeyError:
 
567
        bailout("no help for %r" % topic)
 
568
 
 
569
    doc = cmdfn.__doc__
 
570
    if doc == None:
 
571
        bailout("sorry, no detailed help yet for %r" % topic)
 
572
 
 
573
    print doc
 
574
        
 
575
 
 
576
 
 
577
 
 
578
def cmd_version():
 
579
    print "bzr (bazaar-ng) %s" % bzrlib.__version__
 
580
    print bzrlib.__copyright__
 
581
    print "http://bazaar-ng.org/"
 
582
    print
 
583
    print \
 
584
"""bzr comes with ABSOLUTELY NO WARRANTY.  bzr is free software, and
 
585
you may use, modify and redistribute it under the terms of the GNU 
 
586
General Public License version 2 or later."""
 
587
 
 
588
 
 
589
def cmd_rocks():
 
590
    """Statement of optimism."""
 
591
    print "it sure does!"
 
592
 
 
593
 
 
594
 
 
595
######################################################################
 
596
# main routine
 
597
 
 
598
 
 
599
# list of all available options; the rhs can be either None for an
 
600
# option that takes no argument, or a constructor function that checks
 
601
# the type.
 
602
OPTIONS = {
 
603
    'all':                    None,
 
604
    'help':                   None,
 
605
    'message':                unicode,
 
606
    'profile':                None,
 
607
    'revision':               int,
 
608
    'show-ids':               None,
 
609
    'timezone':               str,
 
610
    'verbose':                None,
 
611
    'version':                None,
 
612
    }
 
613
 
 
614
SHORT_OPTIONS = {
 
615
    'm':                      'message',
 
616
    'r':                      'revision',
 
617
    'v':                      'verbose',
 
618
}
 
619
 
 
620
# List of options that apply to particular commands; commands not
 
621
# listed take none.
 
622
cmd_options = {
 
623
    'add':                    ['verbose'],
 
624
    'commit':                 ['message', 'verbose'],
 
625
    'deleted':                ['show-ids'],
 
626
    'diff':                   ['revision'],
 
627
    'inventory':              ['revision'],
 
628
    'log':                    ['timezone'],
 
629
    'ls':                     ['revision', 'verbose'],
 
630
    'remove':                 ['verbose'],
 
631
    'status':                 ['all'],
 
632
    }
 
633
 
 
634
 
 
635
cmd_args = {
 
636
    'add':                    ['file+'],
 
637
    'commit':                 [],
 
638
    'diff':                   [],
 
639
    'export':                 ['revno', 'dest'],
 
640
    'file-id':                ['filename'],
 
641
    'get-file-text':          ['text_id'],
 
642
    'get-inventory':          ['inventory_id'],
 
643
    'get-revision':           ['revision_id'],
 
644
    'get-revision-inventory': ['revision_id'],
 
645
    'help':                   ['topic?'],
 
646
    'init':                   [],
 
647
    'log':                    [],
 
648
    'lookup-revision':        ['revno'],
 
649
    'relpath':                ['filename'],
 
650
    'remove':                 ['file+'],
 
651
    'root':                   ['filename?'],
 
652
    'status':                 [],
 
653
    }
 
654
 
 
655
 
 
656
def parse_args(argv):
 
657
    """Parse command line.
 
658
    
 
659
    Arguments and options are parsed at this level before being passed
 
660
    down to specific command handlers.  This routine knows, from a
 
661
    lookup table, something about the available options, what optargs
 
662
    they take, and which commands will accept them.
 
663
 
 
664
    >>> parse_args('--help'.split())
 
665
    ([], {'help': True})
 
666
    >>> parse_args('--version'.split())
 
667
    ([], {'version': True})
 
668
    >>> parse_args('status --all'.split())
 
669
    (['status'], {'all': True})
 
670
    >>> parse_args('commit --message=biter'.split())
 
671
    (['commit'], {'message': u'biter'})
 
672
    """
 
673
    args = []
 
674
    opts = {}
 
675
 
 
676
    # TODO: Maybe handle '--' to end options?
 
677
 
 
678
    while argv:
 
679
        a = argv.pop(0)
 
680
        if a[0] == '-':
 
681
            optarg = None
 
682
            if a[1] == '-':
 
683
                mutter("  got option %r" % a)
 
684
                if '=' in a:
 
685
                    optname, optarg = a[2:].split('=', 1)
 
686
                else:
 
687
                    optname = a[2:]
 
688
                if optname not in OPTIONS:
 
689
                    bailout('unknown long option %r' % a)
 
690
            else:
 
691
                shortopt = a[1:]
 
692
                if shortopt not in SHORT_OPTIONS:
 
693
                    bailout('unknown short option %r' % a)
 
694
                optname = SHORT_OPTIONS[shortopt]
 
695
            
 
696
            if optname in opts:
 
697
                # XXX: Do we ever want to support this, e.g. for -r?
 
698
                bailout('repeated option %r' % a)
 
699
                
 
700
            optargfn = OPTIONS[optname]
 
701
            if optargfn:
 
702
                if optarg == None:
 
703
                    if not argv:
 
704
                        bailout('option %r needs an argument' % a)
 
705
                    else:
 
706
                        optarg = argv.pop(0)
 
707
                opts[optname] = optargfn(optarg)
 
708
            else:
 
709
                if optarg != None:
 
710
                    bailout('option %r takes no argument' % optname)
 
711
                opts[optname] = True
 
712
        else:
 
713
            args.append(a)
 
714
 
 
715
    return args, opts
 
716
 
 
717
 
 
718
 
 
719
def _match_args(cmd, args):
 
720
    """Check non-option arguments match required pattern.
 
721
 
 
722
    >>> _match_args('status', ['asdasdsadasd'])
 
723
    Traceback (most recent call last):
 
724
    ...
 
725
    BzrError: ("extra arguments to command status: ['asdasdsadasd']", [])
 
726
    >>> _match_args('add', ['asdasdsadasd'])
 
727
    {'file_list': ['asdasdsadasd']}
 
728
    >>> _match_args('add', 'abc def gj'.split())
 
729
    {'file_list': ['abc', 'def', 'gj']}
 
730
    """
 
731
    # match argument pattern
 
732
    argform = cmd_args.get(cmd, [])
 
733
    argdict = {}
 
734
    # TODO: Need a way to express 'cp SRC... DEST', where it matches
 
735
    # all but one.
 
736
 
 
737
    # step through args and argform, allowing appropriate 0-many matches
 
738
    for ap in argform:
 
739
        argname = ap[:-1]
 
740
        if ap[-1] == '?':
 
741
            if args:
 
742
                argdict[argname] = args.pop(0)
 
743
        elif ap[-1] == '*':
 
744
            assert 0
 
745
        elif ap[-1] == '+':
 
746
            if not args:
 
747
                bailout("command %r needs one or more %s"
 
748
                        % (cmd, argname.upper()))
 
749
            else:
 
750
                argdict[argname + '_list'] = args[:]
 
751
                args = []
 
752
        else:
 
753
            # just a plain arg
 
754
            argname = ap
 
755
            if not args:
 
756
                bailout("command %r requires argument %s"
 
757
                        % (cmd, argname.upper()))
 
758
            else:
 
759
                argdict[argname] = args.pop(0)
 
760
            
 
761
    if args:
 
762
        bailout("extra arguments to command %s: %r"
 
763
                % (cmd, args))
 
764
 
 
765
    return argdict
 
766
 
 
767
 
 
768
 
 
769
def run_bzr(argv):
 
770
    """Execute a command.
 
771
 
 
772
    This is similar to main(), but without all the trappings for
 
773
    logging and error handling.
 
774
    """
 
775
    try:
 
776
        args, opts = parse_args(argv[1:])
 
777
        if 'help' in opts:
 
778
            # TODO: pass down other arguments in case they asked for
 
779
            # help on a command name?
 
780
            cmd_help()
 
781
            return 0
 
782
        elif 'version' in opts:
 
783
            cmd_version()
 
784
            return 0
 
785
        cmd = args.pop(0)
 
786
    except IndexError:
 
787
        log_error('usage: bzr COMMAND\n')
 
788
        log_error('  try "bzr help"\n')
 
789
        return 1
 
790
 
 
791
    try:
 
792
        cmd_handler = globals()['cmd_' + cmd.replace('-', '_')]
 
793
    except KeyError:
 
794
        bailout("unknown command " + `cmd`)
 
795
 
 
796
    # global option
 
797
    if 'profile' in opts:
 
798
        profile = True
 
799
        del opts['profile']
 
800
    else:
 
801
        profile = False
 
802
 
 
803
    # check options are reasonable
 
804
    allowed = cmd_options.get(cmd, [])
 
805
    for oname in opts:
 
806
        if oname not in allowed:
 
807
            bailout("option %r is not allowed for command %r"
 
808
                    % (oname, cmd))
 
809
 
 
810
    # mix arguments and options into one dictionary
 
811
    cmdargs = _match_args(cmd, args)
 
812
    for k, v in opts.items():
 
813
        cmdargs[k.replace('-', '_')] = v
 
814
 
 
815
    if profile:
 
816
        import hotshot
 
817
        prof = hotshot.Profile('.bzr.profile')
 
818
        ret = prof.runcall(cmd_handler, **cmdargs) or 0
 
819
        prof.close()
 
820
 
 
821
        import hotshot.stats
 
822
        stats = hotshot.stats.load('.bzr.profile')
 
823
        #stats.strip_dirs()
 
824
        stats.sort_stats('cumulative', 'calls')
 
825
        stats.print_stats(20)
 
826
    else:
 
827
        return cmd_handler(**cmdargs) or 0
 
828
 
 
829
 
 
830
 
 
831
def main(argv):
 
832
    ## TODO: Handle command-line options; probably know what options are valid for
 
833
    ## each command
 
834
 
 
835
    ## TODO: If the arguments are wrong, give a usage message rather
 
836
    ## than just a backtrace.
 
837
 
 
838
    bzrlib.trace.create_tracefile(argv)
 
839
    
 
840
    try:
 
841
        ret = run_bzr(argv)
 
842
        return ret
 
843
    except BzrError, e:
 
844
        log_error('bzr: error: ' + e.args[0] + '\n')
 
845
        if len(e.args) > 1:
 
846
            for h in e.args[1]:
 
847
                log_error('  ' + h + '\n')
 
848
        return 1
 
849
    except Exception, e:
 
850
        log_error('bzr: exception: %s\n' % e)
 
851
        log_error('    see .bzr.log for details\n')
 
852
        traceback.print_exc(None, bzrlib.trace._tracefile)
 
853
        traceback.print_exc(None, sys.stderr)
 
854
        return 1
 
855
 
 
856
    # TODO: Maybe nicer handling of IOError?
 
857
 
 
858
 
 
859
 
 
860
if __name__ == '__main__':
 
861
    sys.exit(main(sys.argv))
 
862
    ##import profile
 
863
    ##profile.run('main(sys.argv)')
 
864