#!/usr/bin/env python2.7

# Copyright (c) 2013, Web Notes Technologies Pvt. Ltd. and Contributors
# MIT License. See license.txt

from __future__ import unicode_literals
import os
import subprocess
import frappe
import json
from frappe.utils import cint

site_arg_optional = ['serve', 'build', 'watch', 'celery', 'resize_images', 'doctor', 'dump_queue_status', 'purge_all_tasks']

def get_site(parsed_args):
	if not parsed_args.get("site") and os.path.exists(os.path.join(parsed_args["sites_path"], "currentsite.txt")):
		with open(os.path.join(parsed_args["sites_path"], "currentsite.txt"), "r") as sitefile:
			parsed_args["site"] = sitefile.read().strip()
			return  parsed_args["site"]
	return parsed_args.get("site")

def main():
	parsed_args = frappe._dict(vars(setup_parser()))
	fn = get_function(parsed_args)
	if parsed_args.get("sites_path"):
		parsed_args["sites_path"] = parsed_args["sites_path"][0]
	else:
		parsed_args["sites_path"] = os.environ.get("SITES_PATH", ".")
	sites_path = parsed_args.get("sites_path")

	if not parsed_args.get("make_app"):
		if parsed_args.get("site")=="all":
			for site in get_sites(parsed_args["sites_path"]):
				print "\nRunning", fn, "for", site
				print "-"*50
				args = parsed_args.copy()
				args["site"] = site
				frappe.init(site, sites_path=sites_path)
				ret = run(fn, args)
				if ret:
					# if there's a return value, it's an error, so quit
					return ret
		else:
			site = get_site(parsed_args)
			if fn not in site_arg_optional and not site:
				print 'site argument required'
				return 1
			elif site:
				frappe.init(site, sites_path=sites_path)
			else:
				# site argument optional
				frappe.init("", sites_path=sites_path)
			return run(fn, parsed_args)
	else:
		return run(fn, parsed_args)

def cmd(fn):
	def new_fn(*args, **kwargs):
		import inspect
		fnargs, varargs, varkw, defaults = inspect.getargspec(fn)
		new_kwargs = {}
		for i, a in enumerate(fnargs):
			# should not pass an argument more than once
			if i >= len(args) and a in kwargs:
				new_kwargs[a] = kwargs.get(a)

		return fn(*args, **new_kwargs)

	return new_fn


def run(fn, args):
	import cProfile, pstats, StringIO

	use_profiler = args.get("profile") and fn!="serve"
	if use_profiler:
		pr = cProfile.Profile()
		pr.enable()

	if isinstance(args.get(fn), (list, tuple)):
		out = globals().get(fn)(*args.get(fn), **args)
	else:
		out = globals().get(fn)(**args)

	if use_profiler:
		pr.disable()
		s = StringIO.StringIO()
		ps = pstats.Stats(pr, stream=s).sort_stats('tottime', 'ncalls')
		ps.print_stats()
		print s.getvalue()

	return out

def get_function(args):
	for fn, val in args.items():
		if (val or isinstance(val, list)) and globals().get(fn):
			return fn

def get_sites(sites_path=None):
	import os
	if not sites_path:
		sites_path = '.'
	return [site for site in os.listdir(sites_path)
			if os.path.isdir(os.path.join(sites_path, site))
				and not site in ('assets',)]

def setup_parser():
	import argparse
	parser = argparse.ArgumentParser(description="Run frappe utility functions")

	setup_install(parser)
	setup_utilities(parser)
	setup_translation(parser)
	setup_test(parser)

	parser.add_argument("site", nargs="?")

	# common
	parser.add_argument("-f", "--force", default=False, action="store_true",
		help="Force execution where applicable (look for [-f] in help)")
	parser.add_argument("--all", default=False, action="store_true",
		help="Get all (where applicable)")
	parser.add_argument("--verbose", default=False, action="store_true",
		help="Show verbose output (where applicable)")
	parser.add_argument("--quiet", default=False, action="store_true",
		help="Do not show verbose output (where applicable)")
	parser.add_argument("--args", metavar="pass arguments", nargs="*",
		help="pass arguments to the method")

	return parser.parse_args()

def setup_install(parser):
	parser.add_argument("--make_app", metavar=("DESTINATION", "APP-NAME"), nargs=2,
		help="Make a new application with boilerplate")
	parser.add_argument("--install", metavar="DB-NAME", nargs=1,
		help="Install a new db")
	parser.add_argument("--admin_password", metavar="ADMIN-PASSWD", help="Administrator password for site")
	parser.add_argument("--root_password", metavar="ROOT-PASSWD",
		help="MariaDB root password")
	parser.add_argument("--sites_path", metavar="SITES_PATH", nargs=1,
		help="path to directory with sites")
	parser.add_argument("--install_app", metavar="APP-NAME", nargs=1,
		help="Install a new app")
	parser.add_argument("--add_to_installed_apps", metavar="APP-NAME", nargs="*",
		help="Add these app(s) to Installed Apps")
	parser.add_argument("--remove_from_installed_apps", metavar="APP-NAME", nargs="*",
		help="Remove these app(s) from Installed Apps")
	parser.add_argument("--reinstall", default=False, action="store_true",
		help="Install a fresh app in db_name specified in conf.py")
	parser.add_argument("--restore", metavar=("DB-NAME", "SQL-FILE"), nargs=2,
		help="Restore from an sql file")
	parser.add_argument("--with_scheduler_enabled", default=False, action="store_true",
		help="Enable scheduler on restore")
	parser.add_argument("--add_system_manager", nargs="+",
		metavar=("EMAIL", "[FIRST-NAME] [LAST-NAME]"), help="Add a user with all roles")

def setup_test(parser):
	parser.add_argument("--run_tests", default=False, action="store_true",
		help="Run tests options [-d doctype], [-m module]")
	parser.add_argument("--app", metavar="APP-NAME", nargs=1,
		help="Run command for specified app")
	parser.add_argument("-d", "--doctype", metavar="DOCTYPE", nargs=1,
		help="Run command for specified doctype")
	parser.add_argument("-m", "--module", metavar="MODULE", nargs=1,
		help="Run command for specified module")
	parser.add_argument("--tests", metavar="TEST FUNCTION", nargs="*",
		help="Run one or more specific test functions")
	parser.add_argument("--serve_test", action="store_true", help="Run development server for testing")
	parser.add_argument("--driver", nargs="?", help="Run selenium using given driver")


def setup_utilities(parser):
	# serving
	parser.add_argument("--port", type=int, help="port for development server")
	parser.add_argument("--use", action="store_true", help="Set current site for development.")

	# update
	parser.add_argument("-u", "--update", nargs="*", metavar=("REMOTE", "BRANCH"),
		help="Perform git pull, run patches, sync schema and rebuild files/translations")
	parser.add_argument("--reload_gunicorn", default=False, action="store_true", help="reload gunicorn on update")
	parser.add_argument("--patch", nargs=1, metavar="PATCH-MODULE",
		help="Run a particular patch [-f]")
	parser.add_argument("-l", "--latest", default=False, action="store_true",
		help="Run patches, sync schema and rebuild files/translations")
	parser.add_argument("--sync_all", default=False, action="store_true",
		help="Reload all doctypes, pages, etc. using txt files [-f]")
	parser.add_argument("--update_all_sites", nargs="*", metavar=("REMOTE", "BRANCH"),
		help="Perform git pull, run patches, sync schema and rebuild files/translations")

	parser.add_argument("--reload_doc", nargs=3,
		metavar=('"MODULE"', '"DOCTYPE"', '"DOCNAME"'))

	# build
	parser.add_argument("-b", "--build", default=False, action="store_true",
		help="Minify + concatenate JS and CSS files, build translations")
	parser.add_argument("--make_copy", default=False, action="store_true",
		help="Make copy of assets instead of symlinks")
	parser.add_argument("-w", "--watch", default=False, action="store_true",
		help="Watch and concatenate JS and CSS files as and when they change")

	# misc
	parser.add_argument("--backup", default=False, action="store_true",
		help="Take backup of database in backup folder [--with_files]")
	parser.add_argument("--move", default=False, action="store_true",
		help="Move site to different directory defined by --dest_dir")
	parser.add_argument("--dest_dir", nargs=1, metavar="DEST-DIR",
		help="Move site to different directory")
	parser.add_argument("--with_files", default=False, action="store_true",
		help="Also take backup of files")
	parser.add_argument("--domain", nargs="*",
		help="Get or set domain in Website Settings")
	parser.add_argument("--make_conf", nargs="*", metavar=("DB-NAME", "DB-PASSWORD"),
		help="Create new conf.py file")
	parser.add_argument("--make_custom_server_script", nargs=1, metavar="DOCTYPE",
		help="Create new custome server script")
	parser.add_argument("--init_list", nargs=1, metavar="DOCTYPE",
		help="Create new list.js and list.html files")
	parser.add_argument("--set_admin_password", metavar='ADMIN-PASSWORD', nargs="*",
		help="Set administrator password")
	parser.add_argument("--request", metavar='URL-ARGS', nargs=1, help="Run request as admin")
	parser.add_argument("--mysql", action="store_true", help="get mysql shell for a site")
	parser.add_argument("--serve", action="store_true", help="Run development server")
	parser.add_argument("--profile", action="store_true", help="enable profiling in development server")
	parser.add_argument("--smtp", action="store_true", help="Run smtp debug server",
		dest="smtp_debug_server")
	parser.add_argument("--python", action="store_true", help="get python shell for a site")
	parser.add_argument("--ipython", action="store_true", help="get ipython shell for a site")
	parser.add_argument("--execute", help="execute a function", nargs=1, metavar="FUNCTION")
	parser.add_argument("--get_site_status", action="store_true", help="Get site details")
	parser.add_argument("--update_site_config", nargs=1,
		metavar="site-CONFIG-JSON",
		help="Update site_config.json for a given site")
	parser.add_argument("--resize_images", nargs=1, metavar="PATH", help="resize images to a max width of 500px")


	# clear
	parser.add_argument("--clear_web", default=False, action="store_true",
		help="Clear website cache")
	parser.add_argument("--build_website", default=False, action="store_true",
		help="Sync statics and clear cache")
	parser.add_argument("--sync_statics", default=False, action="store_true",
		help="Sync files from templates/statics to Web Pages")
	parser.add_argument("--clear_cache", default=False, action="store_true",
		help="Clear cache, doctype cache and defaults")
	parser.add_argument("--reset_perms", default=False, action="store_true",
		help="Reset permissions for all doctypes")
	parser.add_argument("--clear_all_sessions", default=False, action="store_true",
		help="Clear sessions of all users (logs them out)")

	# scheduler
	parser.add_argument("--run_scheduler", default=False, action="store_true",
		help="Trigger scheduler")
	parser.add_argument("--celery", nargs="*", help="Run Celery Commands")
	parser.add_argument("--run_scheduler_event", nargs=1,
		metavar="all | daily | weekly | monthly",
		help="Run a scheduler event")
	parser.add_argument("--enable_scheduler", default=False, action="store_true",
		help="Enable scheduler")
	parser.add_argument("--disable_scheduler", default=False, action="store_true",
		help="Disable scheduler")


	# replace
	parser.add_argument("--replace", nargs=3,
		metavar=("SEARCH-REGEX", "REPLACE-BY", "FILE-EXTN"),
		help="Multi-file search-replace [-f]")

	# import/export
	parser.add_argument("--export_doc", nargs=2, metavar=('"DOCTYPE"', '"DOCNAME"'))
	parser.add_argument("--export_doclist", nargs=3, metavar=("DOCTYPE", "NAME", "PATH"),
		help="""Export doclist as json to the given path, use '-' as name for Singles.""")
	parser.add_argument("--export_csv", nargs=2, metavar=("DOCTYPE", "PATH"),
		help="""Dump DocType as csv""")
	parser.add_argument("--export_fixtures", default=False, action="store_true",
		help="""Export fixtures""")
	parser.add_argument("--import_doc", nargs=1, metavar="PATH",
		help="""Import (insert/update) doclist. If the argument is a directory, all files ending with .json are imported""")

	# doctor
	parser.add_argument("--doctor", default=False, action="store_true",
		help="Print diagnostic information for task queues")
	parser.add_argument("--purge_all_tasks", default=False, action="store_true",
		help="Purge any pending periodic tasks of 'all' event. Doesn't purge hourly, daily and weekly")
	parser.add_argument("--dump_queue_status", default=False, action="store_true",
		help="Dump detailed diagnostic infomation for task queues in JSON format")

def setup_translation(parser):
	parser.add_argument("--build_message_files", default=False, action="store_true",
		help="Build message files for translation.")
	parser.add_argument("--get_untranslated", nargs=2, metavar=("LANG-CODE", "TARGET-FILE-PATH"),
		help="""Get untranslated strings for lang.""")
	parser.add_argument("--update_translations", nargs=3,
		metavar=("LANG-CODE", "UNTRANSLATED-FILE-PATH", "TRANSLATED-FILE-PATH"),
		help="""Update translated strings.""")

# methods
@cmd
def make_app(destination, app_name):
	from frappe.utils.boilerplate import make_boilerplate
	make_boilerplate(destination, app_name)

@cmd
def use(sites_path):
	with open(os.path.join(sites_path,  "currentsite.txt"), "w") as sitefile:
		sitefile.write(frappe.local.site)

# install
def _install(db_name, root_login="root", root_password=None, source_sql=None,
		admin_password = 'admin', force=False, site_config=None, reinstall=False, quiet=False, install_apps=None):

	from frappe.installer import install_db, install_app, make_site_dirs
	import frappe.utils.scheduler

	verbose = not quiet

	# enable scheduler post install?
	enable_scheduler = _is_scheduler_enabled()

	install_db(root_login=root_login, root_password=root_password, db_name=db_name, source_sql=source_sql,
		admin_password = admin_password, verbose=verbose, force=force, site_config=site_config, reinstall=reinstall)
	make_site_dirs()
	install_app("frappe", verbose=verbose, set_as_patched=not source_sql)

	if frappe.conf.get("install_apps"):
		for app in frappe.conf.install_apps:
			install_app(app, verbose=verbose, set_as_patched=not source_sql)

	if install_apps:
		for app in install_apps:
			install_app(app, verbose=verbose, set_as_patched=not source_sql)

	frappe.utils.scheduler.toggle_scheduler(enable_scheduler)
	scheduler_status = "disabled" if frappe.utils.scheduler.is_scheduler_disabled() else "enabled"
	print "*** Scheduler is", scheduler_status, "***"

@cmd
def install(db_name, root_login="root", root_password=None, source_sql=None,
		admin_password=None, force=False, site_config=None, reinstall=False, quiet=False, install_apps=None):
	_install(db_name, root_login, root_password, source_sql, admin_password, force, site_config, reinstall, quiet, install_apps)
	frappe.destroy()

def _is_scheduler_enabled():
	enable_scheduler = False
	try:
		frappe.connect()
		enable_scheduler = cint(frappe.db.get_default("enable_scheduler"))
	except:
		pass
	finally:
		frappe.db.close()

	return enable_scheduler

@cmd
def install_app(app_name, quiet=False):
	verbose = not quiet
	from frappe.installer import install_app
	frappe.connect()
	install_app(app_name, verbose=verbose)
	frappe.destroy()

@cmd
def add_to_installed_apps(*apps):
	from frappe.installer import add_to_installed_apps
	frappe.connect()
	all_apps = frappe.get_all_apps(with_frappe=True)
	for app in apps:
		if app in all_apps:
			add_to_installed_apps(app, rebuild_website=False)
	frappe.destroy()

@cmd
def remove_from_installed_apps(*apps):
	from frappe.installer import remove_from_installed_apps
	frappe.connect()
	for app in apps:
		remove_from_installed_apps(app)
	frappe.destroy()

@cmd
def reinstall(quiet=False):
	verbose = not quiet
	try:
		frappe.connect()
		installed = frappe.get_installed_apps()
		frappe.clear_cache()
	except:
		installed = []
	finally:
		frappe.db.close()

	install(db_name=frappe.conf.db_name, verbose=verbose, force=True, reinstall=True, install_apps=installed)

@cmd
def restore(db_name, source_sql, force=False, quiet=False, with_scheduler_enabled=False):
	import frappe.utils.scheduler
	_install(db_name, source_sql=source_sql, quiet=quiet, force=force)

	try:
		frappe.connect()
		frappe.utils.scheduler.toggle_scheduler(with_scheduler_enabled)
		scheduler_status = "disabled" if frappe.utils.scheduler.is_scheduler_disabled() else "enabled"
		print "*** Scheduler is", scheduler_status, "***"
	finally:
		frappe.destroy()

@cmd
def add_system_manager(email, first_name=None, last_name=None):
	import frappe.utils.user
	frappe.connect()
	frappe.utils.user.add_system_manager(email, first_name, last_name)
	frappe.db.commit()
	frappe.destroy()

# utilities

@cmd
def update(remote=None, branch=None, reload_gunicorn=False):
	pull(remote=remote, branch=branch)

	# maybe there are new framework changes, any consequences?
	reload(frappe)
	build()
	latest()
	if reload_gunicorn:
		subprocess.check_output("killall -HUP gunicorn".split())

@cmd
def latest(rebuild_website=True, quiet=False):
	import frappe.modules.patch_handler
	import frappe.model.sync
	from frappe.utils.fixtures import sync_fixtures
	import frappe.translate
	from frappe.desk.notifications import clear_notifications

	verbose = not quiet

	frappe.connect()

	try:
		# run patches
		frappe.modules.patch_handler.run_all()
		# sync
		frappe.model.sync.sync_all(verbose=verbose)
		frappe.translate.clear_cache()
		sync_fixtures()

		clear_notifications()

		if rebuild_website:
			build_website()
	finally:
		frappe.destroy()

@cmd
def sync_all(force=False, quiet=False):
	import frappe.model.sync
	verbose = not quiet
	frappe.connect()
	frappe.model.sync.sync_all(force=force, verbose=verbose)
	frappe.destroy()

@cmd
def patch(patch_module, force=False):
	import frappe.modules.patch_handler
	frappe.connect()
	frappe.modules.patch_handler.run_single(patch_module, force=force)
	frappe.destroy()

@cmd
def update_all_sites(remote=None, branch=None, quiet=False):
	verbose = not quiet
	pull(remote, branch)

	# maybe there are new framework changes, any consequences?
	reload(frappe)

	build()
	for site in get_sites():
		frappe.init(site)
		latest(verbose=verbose)

@cmd
def reload_doc(module, doctype, docname, force=False):
	frappe.connect()
	frappe.reload_doc(module, doctype, docname, force=force)
	frappe.db.commit()
	frappe.destroy()

@cmd
def build(make_copy=False, verbose=False):
	import frappe.build
	import frappe
	frappe.build.bundle(False, make_copy=make_copy, verbose=verbose)

@cmd
def watch():
	import frappe.build
	frappe.build.watch(True)

@cmd
def backup(with_files=False, backup_path_db=None, backup_path_files=None, quiet=False):
	from frappe.utils.backups import scheduled_backup
	verbose = not quiet
	frappe.connect()
	odb = scheduled_backup(ignore_files=not with_files, backup_path_db=backup_path_db, backup_path_files=backup_path_files)
	if verbose:
		from frappe.utils import now
		print "database backup taken -", odb.backup_path_db, "- on", now()
		if with_files:
			print "files backup taken -", odb.backup_path_files, "- on", now()
	frappe.destroy()

@cmd
def move(dest_dir=None, site=None):
	import os
	if not dest_dir:
		raise Exception, "--dest_dir is required for --move"
	if not os.path.isdir(dest_dir):
		raise Exception, "destination is not a directory or does not exist"

	old_path = frappe.utils.get_site()
	new_path = os.path.join(dest_dir, site)

	# check if site dump of same name already exists
	site_dump_exists = True
	count = 0
	while site_dump_exists:
		final_new_path = new_path + (count and str(count) or "")
		site_dump_exists = os.path.exists(final_new_path)
		count = int(count or 0) + 1

	os.rename(old_path, final_new_path)
	frappe.destroy()
	return os.path.basename(final_new_path)

@cmd
def domain(host_url=None):
	frappe.connect()
	if host_url:
		frappe.db.set_value("Website Settings", None, "subdomain", host_url)
		frappe.db.commit()
	else:
		print frappe.db.get_value("Website Settings", None, "subdomain")
	frappe.destroy()

@cmd
def make_conf(db_name=None, db_password=None, site_config=None):
	from frappe.install_lib.install import make_conf
	make_conf(db_name=db_name, db_password=db_password, site_config=site_config)

@cmd
def make_custom_server_script(doctype):
	from frappe.custom.doctype.custom_script.custom_script import make_custom_server_script_file
	frappe.connect()
	make_custom_server_script_file(doctype)
	frappe.destroy()

@cmd
def init_list(doctype):
	import frappe.core.doctype.doctype.doctype
	frappe.core.doctype.doctype.doctype.init_list(doctype)

# clear
@cmd
def clear_cache():
	import frappe.sessions
	from frappe.desk.notifications import clear_notifications
	frappe.connect()
	frappe.clear_cache()
	clear_notifications()
	frappe.destroy()

@cmd
def clear_web():
	import frappe.website.render
	frappe.connect()
	frappe.website.render.clear_cache()
	frappe.destroy()

@cmd
def clear_all_sessions():
	import frappe.sessions
	frappe.connect()
	frappe.sessions.clear_all_sessions()
	frappe.db.commit()
	frappe.destroy()

@cmd
def build_website(verbose=False):
	from frappe.website import render, statics
	frappe.connect()
	render.clear_cache()
	statics.sync(verbose=verbose).start(True)
	frappe.db.commit()
	frappe.destroy()

@cmd
def sync_statics(force=False):
	from frappe.website import statics
	frappe.connect()
	statics.sync_statics(rebuild = force)
	frappe.db.commit()
	frappe.destroy()

@cmd
def reset_perms():
	from frappe.permissions import reset_perms
	frappe.connect()
	for d in frappe.db.sql_list("""select name from `tabDocType`
		where ifnull(istable, 0)=0 and ifnull(custom, 0)=0"""):
			frappe.clear_cache(doctype=d)
			reset_perms(d)
	frappe.destroy()

@cmd
def execute(method, args=None):
	frappe.connect()
	if args:
		ret = frappe.get_attr(method)(*args)
	else:
		ret = frappe.get_attr(method)()

	if frappe.db:
		frappe.db.commit()
		frappe.destroy()
	if ret:
		print ret

@cmd
def celery(arg):
	import frappe
	import commands, os
	python = commands.getoutput('which python')
	os.execv(python, [python, "-m", "frappe.celery_app"] + arg.split())
	frappe.destroy()

@cmd
def run_scheduler_event(event, force=False):
	import frappe.utils.scheduler
	frappe.connect()
	frappe.utils.scheduler.trigger(frappe.local.site, event, now=force)
	frappe.destroy()

@cmd
def enable_scheduler():
	import frappe.utils.scheduler
	frappe.connect()
	frappe.utils.scheduler.enable_scheduler()
	frappe.db.commit()
	print "Enabled"
	frappe.destroy()

@cmd
def disable_scheduler():
	import frappe.utils.scheduler
	frappe.connect()
	frappe.utils.scheduler.disable_scheduler()
	frappe.db.commit()
	print "Disabled"
	frappe.destroy()

# replace
@cmd
def replace(search_regex, replacement, extn, force=False):
	print search_regex, replacement, extn
	replace_code('.', search_regex, replacement, extn, force=force)

# import/export
@cmd
def export_doc(doctype, docname):
	import frappe.modules
	frappe.connect()
	frappe.modules.export_doc(doctype, docname)
	frappe.destroy()

@cmd
def export_doclist(doctype, name, path):
	from frappe.core.page.data_import_tool import data_import_tool
	frappe.connect()
	data_import_tool.export_json(doctype, name, path)
	frappe.destroy()

@cmd
def export_csv(doctype, path):
	from frappe.core.page.data_import_tool import data_import_tool
	frappe.connect()
	data_import_tool.export_csv(doctype, path)
	frappe.destroy()

@cmd
def export_fixtures():
	from frappe.utils.fixtures import export_fixtures
	frappe.connect()
	export_fixtures()
	frappe.destroy()

@cmd
def import_doc(path, force=False):
	from frappe.core.page.data_import_tool import data_import_tool
	frappe.connect()
	data_import_tool.import_doc(path, overwrite=force)
	frappe.destroy()

# translation
@cmd
def build_message_files():
	import frappe.translate
	frappe.connect()
	frappe.translate.rebuild_all_translation_files()
	frappe.destroy()

@cmd
def get_untranslated(lang, untranslated_file, all=None):
	import frappe.translate
	frappe.connect()
	frappe.translate.get_untranslated(lang, untranslated_file, get_all=all)
	frappe.destroy()

@cmd
def update_translations(lang, untranslated_file, translated_file):
	import frappe.translate
	frappe.connect()
	frappe.translate.update_translations(lang, untranslated_file, translated_file)
	frappe.destroy()

# git
@cmd
def git(param):
	if isinstance(param, (list, tuple)):
		param = " ".join(param)
	import os
	os.system("""cd lib && git %s""" % param)
	os.system("""cd app && git %s""" % param)

def get_remote_and_branch(remote=None, branch=None):
	if not (remote and branch):
		if not frappe.conf.branch:
			raise Exception("Please specify remote and branch")

		remote = remote or "origin"
		branch = branch or frappe.conf.branch
		frappe.destroy()

	return remote, branch

@cmd
def pull(remote=None, branch=None):
	remote, branch = get_remote_and_branch(remote, branch)
	git(("pull", remote, branch))

@cmd
def push(remote=None, branch=None):
	remote, branch = get_remote_and_branch(remote, branch)
	git(("push", remote, branch))

@cmd
def status():
	git("status")

@cmd
def commit(message):
	git("""commit -a -m "%s" """ % message.replace('"', '\"'))

@cmd
def checkout(branch):
	git(("checkout", branch))

@cmd
def set_admin_password(admin_password=None):
	import frappe
	import getpass

	while not admin_password:
		admin_password = getpass.getpass("Administrator's password: ")

	frappe.connect()
	frappe.db.sql("""update __Auth set `password`=password(%s)
		where user='Administrator'""", (admin_password,))
	frappe.db.commit()
	frappe.destroy()

@cmd
def mysql():
	import frappe
	import commands, os
	msq = commands.getoutput('which mysql')
	os.execv(msq, [msq, '-u', frappe.conf.db_name, '-p'+frappe.conf.db_password, frappe.conf.db_name, '-h', frappe.conf.db_host or "localhost", "-A"])
	frappe.destroy()

@cmd
def python(site):
	import frappe
	import commands, os
	python = commands.getoutput('which python')
	if site:
		os.environ["site"] = site
	os.environ["PYTHONSTARTUP"] = os.path.join(os.path.dirname(frappe.__file__), "pythonrc.py")
	os.execv(python, [python])
	frappe.destroy()

@cmd
def ipython(site):
	import frappe
	frappe.connect(site=site)
	import IPython
	IPython.embed()

@cmd
def smtp_debug_server():
	import commands, os
	python = commands.getoutput('which python')
	os.execv(python, [python, '-m', "smtpd", "-n", "-c", "DebuggingServer", "localhost:25"])

@cmd
def run_tests(app=None, module=None, doctype=None, verbose=False, tests=(), driver=None, force=False):
	import frappe.test_runner
	from frappe.utils import sel

	# sel.start(verbose, driver)

	ret = 1
	try:
		ret = frappe.test_runner.main(app and app[0], module and module[0], doctype and doctype[0], verbose,
			tests=tests, force=force)
		if len(ret.failures) == 0 and len(ret.errors) == 0:
			ret = 0
	finally:
		pass
		# sel.close()

	return ret

@cmd
def serve(port=None, profile=False, sites_path='.', site=None):
	if not port: port = 8000

	import frappe.app
	frappe.app.serve(port=port, profile=profile, site=frappe.local.site, sites_path=sites_path)

@cmd
def serve_test(port=None, profile=False, sites_path='.', site=None):
	from frappe.utils import sel
	if not port: port = sel.port
	serve(port)

@cmd
def request(args):
	import frappe.handler
	import frappe.api
	frappe.connect()
	if "?" in args:
		frappe.local.form_dict = frappe._dict([a.split("=") for a in args.split("?")[-1].split("&")])
	else:
		frappe.local.form_dict = frappe._dict()

	if args.startswith("/api/method"):
		frappe.local.form_dict.cmd = args.split("?")[0].split("/")[-1]

	frappe.handler.execute_cmd(frappe.form_dict.cmd)

	print frappe.response
	frappe.destroy()

@cmd
def resize_images(path):
	import frappe.utils.image
	frappe.utils.image.resize_images(path)

def replace_code(start, txt1, txt2, extn, search=None, force=False):
	"""replace all txt1 by txt2 in files with extension (extn)"""
	import frappe.utils
	import os, re
	esc = frappe.utils.make_esc('[]')
	if not search: search = esc(txt1)
	for wt in os.walk(start, followlinks=1):
		for fn in wt[2]:
			if fn.split('.')[-1]==extn:
				fpath = os.path.join(wt[0], fn)
				with open(fpath, 'r') as f:
					content = f.read()

				if re.search(search, content):
					res = search_replace_with_prompt(fpath, txt1, txt2, force)
					if res == 'skip':
						return 'skip'


def search_replace_with_prompt(fpath, txt1, txt2, force=False):
	""" Search and replace all txt1 by txt2 in the file with confirmation"""
	from termcolor import colored
	with open(fpath, 'r') as f:
		content = f.readlines()

	tmp = []
	for c in content:
		if c.find(txt1) != -1:
			print fpath
			print  colored(txt1, 'red').join(c[:-1].split(txt1))
			a = ''
			if force:
				c = c.replace(txt1, txt2)
			else:
				while a.lower() not in ['y', 'n', 'skip']:
					a = raw_input('Do you want to Change [y/n/skip]?')
				if a.lower() == 'y':
					c = c.replace(txt1, txt2)
				elif a.lower() == 'skip':
					return 'skip'
		tmp.append(c)

	with open(fpath, 'w') as f:
		f.write(''.join(tmp))
	print colored('Updated', 'green')

@cmd
def get_site_status(verbose=False):
	import frappe
	import frappe.utils
	from frappe.utils.user import get_system_managers
	from frappe.core.doctype.user.user import get_total_users, get_active_users, \
		get_website_users, get_active_website_users

	import json
	frappe.connect()
	ret = {
		'last_backup_on': frappe.local.conf.last_backup_on,
		'active_users': get_active_users(),
		'total_users': get_total_users(),
		'active_website_users': get_active_website_users(),
		'website_users': get_website_users(),
		'system_managers': "\n".join(get_system_managers()),
		'default_company': frappe.db.get_default("company"),
		'disk_usage': frappe.utils.get_disk_usage(),
		'working_directory': frappe.local.site_path
	}

	# country, timezone, industry
	for key in ["country", "time_zone", "industry"]:
		ret[key] = frappe.db.get_default(key)

	# basic usage/progress analytics
	for doctype in ("Company", "Customer", "Item", "Quotation", "Sales Invoice",
		"Journal Entry", "Stock Ledger Entry"):
			key = doctype.lower().replace(" ", "_") + "_exists"
			ret[key] = 1 if frappe.db.count(doctype) else 0

	frappe.destroy()

	if verbose:
		print json.dumps(ret, indent=1, sort_keys=True)

	return ret

@cmd
def update_site_config(site_config, verbose=False):
	import json

	if isinstance(site_config, basestring):
		site_config = json.loads(site_config)

	config = frappe.get_site_config()
	config.update(site_config)
	site_config_path = os.path.join(frappe.local.site_path, "site_config.json")

	with open(site_config_path, "w") as f:
		json.dump(config, f, indent=1, sort_keys=True)

	frappe.destroy()

@cmd
def doctor():
	from frappe.utils.doctor import doctor as _doctor
	return _doctor()

@cmd
def purge_all_tasks():
	from frappe.utils.doctor import purge_pending_tasks
	count = purge_pending_tasks()
	print "Purged {} tasks".format(count)

@cmd
def dump_queue_status():
	from frappe.utils.doctor import dump_queue_status as _dump_queue_status
	print json.dumps(_dump_queue_status(), indent=1)

if __name__=="__main__":
	out = main()
	if out and out==1:
		exit(1)