#! /usr/bin/python

# To change this template, choose Tools | Templates
# and open the template in the editor.

__author__="bjschuma"
__date__ ="$Feb 20, 2010 2:19:10 PM$"

from bt.message import *
from session import alias
from session import manager
from session import vars



def runCmd(input):
	write("Running command: "+input,2)

	# Find command
	split = input.split(' ',1)
	cmd = split[0]
	args=None
	if len(split) > 1:
		args = split[1]

	if (cmd in manager.enabled) == True:
		if args == None:
			return manager.enabled[cmd].start()
		else:
			return manager.enabled[cmd].start(args.split())
	else:
		return input


# Check if we are storing in a variable
def varCheck(cmd):
	split = cmd.split('=', 1)
	if len(split)==2 and len(split[0].split())==1:
		var = split[0].strip()
		if not var[0]=="$":
			var = "$" + var
		write("Using variable: "+var, 2)
		return ( var, split[1].strip() )
	else:
		return (None, cmd)


# Replace variables in the command with their real values
def varReplace(cmd):
	for key in vars.keys():
		v = "`"+key+"`"
		if cmd.find(v) > -1:
			new = str(vars[key])
			write(key + " => " + new, 2)
			cmd = cmd.replace(v, new)
	return cmd


# Replace aliases with their real values
def aliasReplace(cmd):
	split = cmd.split()
	out = ""
	for index,word in enumerate(split):
		if index > 0:
			out += " "
		if alias.has(word) == True:
			write(word + " => " + alias[word], 2)
			out += aliasReplace(alias[word])
		else:
			out += word
	return out


def run(string):
	split = string.split(";")
	ans = []
	for cmd in split:
		(var,cmd) = varCheck(cmd)
		cmd = varReplace(cmd)
		cmd = aliasReplace(cmd)

		if var == None:
			ans += [ runCmd(cmd) ]
		else:
			disable()
			vars[var] = runCmd(cmd)
			enable()

	if len(ans) == 1:
		return ans[0]
	return ans


def call(string):
	# disable text printing
	disable()
	result = run(string)
	# enable text printing
	enable()
	return result