/stdhome

# stage_status.py

#

# Copyright (C) 2014 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, re, getopt

from .command import Command

import stdhome.the as the

from stdhome.deployment import Deployment

from stdhome.walker.status import StatusWalker

class StageStatusCommand( Command ):

	def print_help( self ):

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

		print()

		#      01234567890123456789012345678901234567890123456789012345678901234567890123456789

		print("Show the state of files in the local repository (including modifications and")

		print("conflicts).")

		print()

		Command.print_stage_commands_notice()

		print()

		print("This lists files in your local repository which have been modified and which")

		print("have not been committed.")

		print()

		print("Options:")

		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("      --help       display help and exit")

		exit( 0 )

	def parse_command_line( self ):

		opts, args = getopt.gnu_getopt(

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

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

		for opt, optarg in opts:

			if 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 in [ '--verbose', '-v' ]:

				the.verbose += 1

			elif opt == "--help":

				self.print_help()

		# discard first argument (the command)

		args.pop( 0 )

		# remaining arguments

		self.files = args

	def run( self ):

		# set up repo and check it exists

		the.repo.check_dir_exists()

		# determine files

		files = self.expand_files( self.files )

		message = ''

		# initialise deployment and check if it's ongoing

		deployment = Deployment()

		if deployment.is_ongoing():

			print("deployment ongoing")

			# check for conflicts in repo

			files = the.repo.vcs.get_conflicts()

			if files:

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

						   ( the.repo.name, '\n  '.join( files ) )

			# get deployment conflicts

			conflicts = deployment.get_conflicts()

			if conflicts:

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

						   '\n  '.join( conflicts )

		else:

			# repo status

			files = the.repo.vcs.status()

			if len( files ) > 0: message = str( files )

		# show status

		if message: print(message.rstrip())