23
23
from command import Command
24
24
import stdhome.the as the
25
25
from stdhome.deployment import Deployment
26
from stdhome.subprocess import Popen
28
29
class InitCommand( Command ):
33
self.vcs = the.config.default_vcs
36
32
def print_help( self ):
37
33
print "Usage: " + the.program.name + " init [URL] [--repo=REPO]"
46
42
print "hostname, it is internally expanded to scp://HOSTNAME/~/.stdhome/REPO)."
49
print " --quiet do not report changes to the home directory"
50
45
print " -r, --repo=REPO select the repo to check-out or create (defaults to 'home')"
51
46
print " -v, --verbose display information about what is being done"
52
print " --vcs=VCS select the version control system"
53
47
print " --help display help and exit"
57
51
def parse_command_line( self ):
58
52
opts, args = getopt.gnu_getopt(
59
53
sys.argv[ 1: ], "r:v",
60
[ "quiet", "repo=", "vcs=", "verbose", "help" ] )
54
[ "repo=", "verbose", "help" ] )
61
55
for opt, optarg in opts:
64
elif opt in [ '--repo', '-r' ]:
65
if not re.search( '^[-a-zA-z0-9.]+$', optarg ):
56
if opt in [ '--repo', '-r' ]:
57
if not re.match( '^[-a-zA-z0-9.]+$', optarg ):
66
58
raise the.program.FatalError(
67
59
'invalid repository name: ' + optarg )
71
61
elif opt in [ '--verbose', '-v' ]:
73
63
elif opt == "--help":
76
66
# discard first argument (the command)
80
70
self.url = args[ 0 ].strip() if len( args ) else None
82
# check remaining arguments
83
73
if len( args ) > 1:
84
74
raise the.program.UsageError( 'too many arguments' )
89
79
# set up repo and check it *doesn't* already exists
90
80
the.repo.check_dir_exists( False )
91
the.repo.set_vcs( self.vcs )
93
82
# ensure our top-level directory exists
94
83
if not os.path.exists( the.full_dir ):
100
89
# 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 )
90
if re.match( '^[0-9a-zA-z.]+$', self.url ):
91
self.url = 'sftp://%s/%s/%s' % \
92
( self.url, the.dir, the.repo.name )
104
94
# initialise deployment (with an empty repo)
105
95
deployment = Deployment()
106
96
deployment.copy_in()
108
98
# perform bzr checkout
109
if the.verbose >= 1: print "checking out %s" % the.repo.dir
99
if the.verbose: print "checking out %s" % the.repo.dir
111
101
the.repo.vcs.checkout( self.url )
102
except Exception as e:
114
104
# 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 )
106
shutil.rmtree( the.repo.full_dir )
136
112
# perform deployment
137
deployment.copy_out( self.quiet )
114
deployment.copy_out()
115
except deployment.Conflict as e:
116
raise the.program.FatalError( e.msg )
141
120
# perform bzr init
142
if the.verbose >= 1: print 'initialising %s' % the.repo.dir
121
if the.verbose: print 'initialising %s' % the.repo.dir
144
123
the.repo.vcs.init()
124
except Exception as e:
147
126
# attempt to clean-up repo dir, and die
148
if os.path.exists( the.repo.full_dir ):
150
shutil.rmtree( the.repo.full_dir )
128
shutil.rmtree( the.repo.full_dir )