/stdhome

# init.py

#

# Copyright (C) 2013 Tim Marston <tim@edm.am>

#

# This file is part of stdhome (hereafter referred to as "this program").

# See http://ed.am/dev/stdhome for more information.

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import sys, os, re, getopt, shutil, subprocess

from .command import Command

import stdhome.the as the

from stdhome.deployment import Deployment

class InitCommand( Command ):

	def __init__( self ):

		self.quiet = False

		self.vcs = the.config.default_vcs

	def print_help( self ):

		print("Usage: " + the.program.name + " init [URL] [--repo=REPO]")

		print()

		#      01234567890123456789012345678901234567890123456789012345678901234567890123456789

		print("Initialise a local repository.")

		print()

		print("If an URL is given, the local reposity is a checkout of it (i.e., you can")

		print("receive updates from it and changes you commit will be sent to it).  The URL")

		print("can take the form of a simple hostname, such as \"example.com\", or it can be a")

		print("fully-qualified bazaar URL.  (Actually, in the first case, where it is a simple")

		print("hostname, it is internally expanded to scp://HOSTNAME/~/.stdhome/REPO).")

		print()

		print("Options:")

		print("      --quiet      do not report changes to the home directory")

		print("  -r, --repo=REPO  select the repo to check-out or create (defaults to 'home')")

		print("  -v, --verbose    display information about what is being done")

		print("      --vcs=VCS    select the version control system")

		print("      --help       display help and exit")

		exit( 0 )

	def parse_command_line( self ):

		opts, args = getopt.gnu_getopt(

			sys.argv[ 1: ], "r:v",

			[ "quiet", "repo=", "vcs=", "verbose", "help" ] )

		for opt, optarg in opts:

			if opt == "--quiet":

				self.quiet = True

			elif opt in [ '--repo', '-r' ]:

				if not re.search( '^[-a-zA-z0-9.]+$', optarg ):

					raise the.program.FatalError(

						'invalid repository name: ' + optarg )

				the.repo = optarg

			elif opt == "--vcs":

				self.vcs = optarg

			elif opt in [ '--verbose', '-v' ]:

				the.verbose += 1

			elif opt == "--help":

				self.print_help()

		# discard first argument (the command)

		args.pop( 0 )

		# URL argument

		self.url = args[ 0 ].strip() if len( args ) else None

		# check remaining arguments

		if len( args ) > 1:

			raise the.program.UsageError( 'too many arguments' )

	def run( self ):

		# set up repo and check it *doesn't* already exists

		the.repo.check_dir_exists( False )

		the.repo.set_vcs( self.vcs )

		# ensure our top-level directory exists

		if not os.path.exists( the.full_dir ):

			os.mkdir( the.full_dir )

		# checkout a remote repo, or create an empty local one?

		if self.url:

			# expand url if it's a simple hostname

			if re.search( '^[0-9a-zA-z.]+$', self.url ):

				self.url = the.repo.vcs.expand_repo_url( self.url )

			# initialise deployment (with an empty repo)

			deployment = Deployment()

			deployment.copy_in()

			# perform bzr checkout

			if the.verbose >= 1: print("checking out %s" % the.repo.dir)

			try:

				the.repo.vcs.checkout( self.url )

			except Exception:

				# attempt to clean-up repo dir

				if os.path.exists( the.repo.full_dir ):

					try:

						shutil.rmtree( the.repo.full_dir )

					except OSError:

						pass

				raise

			message = ''

			# check for deployment conclicts

			conflicts = deployment.get_conflicts()

			if conflicts:

				message += 'deployment conflicts:\n  %s' % \

						   '\n  '.join( conflicts )

			# stop if there are conflicts

			if message:

				raise the.program.FatalError(

					'there were conflicts...\n' + message )

			# perform deployment

			deployment.copy_out( self.quiet )

		else:

			# perform bzr init

			if the.verbose >= 1: print('initialising %s' % the.repo.dir)

			try:

				the.repo.vcs.init()

			except Exception:

				# attempt to clean-up repo dir, and die

				if os.path.exists( the.repo.full_dir ):

					try:

						shutil.rmtree( the.repo.full_dir )

					except OSError:

						pass

				raise