sugar-toolkit-gtk3/sugar/activity/bundlebuilder.py

#!/usr/bin/python

# Copyright (C) 2006, Red Hat, Inc.
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2 of the License, or (at your option) any later version.
#
# This library 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
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the
# Free Software Foundation, Inc., 59 Temple Place - Suite 330,
# Boston, MA 02111-1307, USA.

import sys
import os
import zipfile
import shutil

from sugar.activity.bundle import Bundle

class _SvnFileList(list):
	def __init__(self):
		f = os.popen('svn list -R')
		for line in f.readlines():
			filename = line.strip()
			if os.path.isfile(filename):
				self.append(filename)
		f.close()

class _GitFileList(list):
	def __init__(self):
		f = os.popen('git-ls-files')
		for line in f.readlines():
			filename = line.strip()
			if not filename.startswith('.'):
				self.append(filename)
		f.close()

def _extract_bundle(source_file, dest_dir):
		if not os.path.exists(dest_dir):
			os.mkdir(dest_dir)

		zf = zipfile.ZipFile(source_file)

		for i, name in enumerate(zf.namelist()):
			path = os.path.join(dest_dir, name)
			
			if not os.path.exists(os.path.dirname(path)):
				os.makedirs(os.path.dirname(path))

			outfile = open(path, 'wb')
			outfile.write(zf.read(name))
			outfile.flush()
			outfile.close()

def _get_source_path():
	return os.getcwd()

def _get_activities_path():
	path = os.path.expanduser('~/Activities')
	if not os.path.isdir(path):
		os.mkdir(path)
	return path

def _get_bundle_dir():
	bundle_name = os.path.basename(_get_source_path())
	return bundle_name + '.activity'	

def _get_install_dir(prefix):
	return os.path.join(prefix, 'share/activities')

def _get_bundle_path():
	return os.path.join(_get_activities_path(), _get_bundle_dir())

def _get_package_name():
	bundle = Bundle(_get_source_path())
	zipname = '%s-%d.xo' % (bundle.get_name(), bundle.get_activity_version())
	return zipname

def _delete_backups(arg, dirname, names):
	for name in names:
		if name.endswith('~') or name.endswith('pyc'):
			os.remove(os.path.join(dirname, name))

def cmd_help():
	print 'Usage: \n\
setup.py dev     - setup for development \n\
setup.py dist    - create a bundle package \n\
setup.py install - install the bundle \n\
setup.py clean   - clean the directory \n\
setup.py help    - print this message \n\
'

def cmd_dev():
	bundle_path = get_bundle_path()
	try:
		os.symlink(_get_source_path(), bundle_path)
	except OSError:
		if os.path.islink(bundle_path):
			print 'ERROR - The bundle has been already setup for development.'
		else:
			print 'ERROR - A bundle with the same name is already installed.'	

def cmd_dist():
	if os.path.isdir('.git'):
		file_list = _GitFileList()
	elif os.path.isdir('.svn'):
		file_list = _SvnFileList()
	else:
		print 'ERROR - The command works only with git or svn repositories.'

	zipname = _get_package_name()
	bundle_zip = zipfile.ZipFile(zipname, 'w', zipfile.ZIP_DEFLATED)
	
	for filename in file_list:
		arcname = os.path.join(_get_bundle_dir(), filename)
		bundle_zip.write(filename, arcname)

	bundle_zip.close()

def cmd_install(prefix):
	cmd_dist()
	cmd_uninstall(prefix)
	_extract_bundle(_get_package_name(), _get_install_dir(prefix))

def cmd_uninstall(prefix):
	path = os.path.join(_get_install_dir(prefix), _get_bundle_dir())
	if os.path.isdir(path):
		shutil.rmtree(path)

def cmd_clean():
	os.path.walk('.', delete_backups, None)

def start():
	if len(sys.argv) < 2:
		cmd_help()
	elif sys.argv[1] == 'build':
		pass
	elif sys.argv[1] == 'dev':
		cmd_dev()
	elif sys.argv[1] == 'dist':
		cmd_dist()
	elif sys.argv[1] == 'install' and len(sys.argv) == 3:
		cmd_install(sys.argv[2])
	elif sys.argv[1] == 'uninstall' and len(sys.argv) == 3:
		cmd_uninstall(sys.argv[2])
	elif sys.argv[1] == 'clean':
		cmd_clean()
	else:
		cmd_help()