22
22
import sys, os, re, getopt, shutil, subprocess
23
from command import Command
24
import stdhome.the as the
25
from stdhome.deployment import Deployment
28
class InitCommand( Command ):
24
from deployment import Deployment
25
from subprocess import Popen
31
31
def __init__( self ):
33
self.vcs = the.config.default_vcs
36
35
def print_help( self ):
46
45
print "hostname, it is internally expanded to scp://HOSTNAME/~/.stdhome/REPO)."
49
print " --quiet do not report changes to the home directory"
50
48
print " -r, --repo=REPO select the repo to check-out or create (defaults to 'home')"
51
49
print " -v, --verbose display information about what is being done"
52
print " --vcs=VCS select the version control system"
53
50
print " --help display help and exit"
57
54
def parse_command_line( self ):
58
55
opts, args = getopt.gnu_getopt(
59
56
sys.argv[ 1: ], "r:v",
60
[ "quiet", "repo=", "vcs=", "verbose", "help" ] )
57
[ "repo=", "verbose", "help" ] )
61
58
for opt, optarg in opts:
64
elif opt in [ '--repo', '-r' ]:
65
if not re.search( '^[-a-zA-z0-9.]+$', optarg ):
59
if opt in [ '--repo', '-r' ]:
60
if not re.match( '^[-a-zA-z0-9.]+$', optarg ):
66
61
raise the.program.FatalError(
67
62
'invalid repository name: ' + optarg )
71
64
elif opt in [ '--verbose', '-v' ]:
73
66
elif opt == "--help":
76
69
# discard first argument (the command)
80
73
self.url = args[ 0 ].strip() if len( args ) else None
82
# check remaining arguments
83
76
if len( args ) > 1:
84
77
raise the.program.UsageError( 'too many arguments' )
89
82
# set up repo and check it *doesn't* already exists
83
the.set_repo( self.repo )
90
84
the.repo.check_dir_exists( False )
91
the.repo.set_vcs( self.vcs )
93
86
# ensure our top-level directory exists
94
87
if not os.path.exists( the.full_dir ):
100
93
# expand url if it's a simple hostname
101
if re.search( '^[0-9a-zA-z.]+$', self.url ):
102
self.url = the.repo.vcs.expand_repo_url( self.url )
94
if re.match( '^[0-9a-zA-z.]+$', self.url ):
95
self.url = 'sftp://%s/%s/%s' % \
96
( self.url, the.dir, the.repo.name )
104
98
# initialise deployment (with an empty repo)
105
99
deployment = Deployment()
106
100
deployment.copy_in()
108
102
# perform bzr checkout
109
if the.verbose >= 1: print "checking out %s" % the.repo.dir
103
if the.verbose: print "checking out %s" % the.repo.dir
111
105
the.repo.vcs.checkout( self.url )
106
except Exception as e:
114
108
# attempt to clean-up repo dir
115
if os.path.exists( the.repo.full_dir ):
117
shutil.rmtree( the.repo.full_dir )
125
# check for deployment conclicts
126
conflicts = deployment.get_conflicts()
128
message += 'deployment conflicts:\n %s' % \
129
'\n '.join( conflicts )
131
# stop if there are conflicts
133
raise the.program.FatalError(
134
'there were conflicts...\n' + message )
110
shutil.rmtree( the.repo.full_dir )
136
116
# perform deployment
137
deployment.copy_out( self.quiet )
118
deployment.copy_out()
119
except deployment.Conflict as e:
120
raise the.program.FatalError( e.msg )
141
124
# perform bzr init
142
if the.verbose >= 1: print 'initialising %s' % the.repo.dir
125
if the.verbose: print 'initialising %s' % the.repo.dir
144
127
the.repo.vcs.init()
128
except Exception as e:
147
130
# attempt to clean-up repo dir, and die
148
if os.path.exists( the.repo.full_dir ):
150
shutil.rmtree( the.repo.full_dir )
132
shutil.rmtree( the.repo.full_dir )