#!/usr/bin/env python
#
# mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
# Copyright (C) 2013 by philemon <mmgen-py@yandex.com>
#
# 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/>.
"""
mmgen-txcreate: Create a BTC transaction, sending to specified addresses
"""

import sys
#from hashlib import sha256

from mmgen.Opts import *
from mmgen.license import *
from mmgen.config import *
from mmgen.tx import *
from mmgen.utils import check_opts, msg, msg_r, user_confirm
from decimal import Decimal

prog_name = sys.argv[0].split("/")[-1]

help_data = {
	'prog_name': prog_name,
	'desc':    "Create a BTC transaction, sending to specified addresses",
	'usage':   "[opts] <address:amount>[,...] <transaction fee> <change address>",
	'options': """
-h, --help                Print this help message
-d, --outdir           d  Specify an alternate directory 'd' for output
-e, --echo-passphrase     Print passphrase to screen when typing it
-i, --info                Display unspent outputs and exit
-q, --quiet               Suppress warnings; overwrite files without
                          prompting

Outputs to spend are chosen by the user via a menu.

Ages of transactions are approximate based on an average block discovery
interval of %s minutes.
""" % mins_per_block
}

short_opts = "hd:eiq"
long_opts  = "help","outdir=","echo_passphrase","info","quiet"

opts,cmd_args = process_opts(sys.argv,help_data,short_opts,long_opts)

# Exits on invalid input
check_opts(opts, ('outdir',))

if debug:
	print "Processed options:     %s" % repr(opts)
	print "Cmd args:              %s" % repr(cmd_args)

if len(cmd_args) == 3:
	rcpt_arg,tx_fee,change_addr = cmd_args
	check_address(change_addr)
elif len(cmd_args) == 2:
	rcpt_arg,tx_fee = cmd_args
	change_addr = ""
elif len(cmd_args) == 0 and 'info' in opts:
	pass
else: usage(help_data)

if not 'info' in opts:
	tx_out = make_tx_out(rcpt_arg)
	for i in tx_out.keys():   check_address(i)
	for i in tx_out.values(): check_btc_amt(i)
	tx_fee   = check_btc_amt(tx_fee)

# Begin execution
c = connect_to_bitcoind()

if not 'quiet' in opts and not 'info' in opts: do_license_msg()

# Begin test
# import mmgen.rpc
# us = eval(get_data_from_file("listunspent.json"))
# End test

us = c.listunspent()

if not us:
	msg_r("""
No spendable outputs found!  Import addresses with balances into your
watch-only wallet using 'mmgen-addrimport' and then re-run this program.
""")
	sys.exit(2)

# write_to_file("listunspent.json",repr(us))
# sys.exit()
unspent = sort_and_view(us)

total = trim_exponent(sum([i.amount for i in unspent]))

msg("Total unspent: %s BTC (%s outputs)" % (total, len(unspent)))
if 'info' in opts: sys.exit(0)

send_amt = sum(tx_out.values())
msg("Total amount to spend: %s BTC" % send_amt)

while True:
	sel_nums = select_outputs(unspent,
			"Enter a range or space-separated list of outputs to spend: ")
	msg("Selected outputs: %s" % " ".join(str(i) for i in sel_nums))
	sel_unspent = [unspent[i-1] for i in sel_nums]

	lbls = set([verify_mmgen_label(
				i.account,return_str=True,check_label_len=True)
					for i in sel_unspent])
	lbls.discard("")

	if lbls and len(lbls) < len(sel_unspent):
		msg(txmsg['mixed_inputs'] % ", ".join(sorted(lbls)))
		if not user_confirm("Accept?"):
			continue

	total_in = trim_exponent(sum([o.amount for o in sel_unspent]))
	change   = trim_exponent(total_in - (send_amt + tx_fee))

	if change >= 0:
		prompt = "Transaction produces %s BTC in change.  OK?" % change
		if user_confirm(prompt,default_yes=True):
			break
	else:
		msg(txmsg['not_enough_btc'] % change)


if change > 0 and not change_addr:
	msg(txmsg['throwaway_change'] % (change, total_in-tx_fee))
	sys.exit(2)

tx_in = [{"txid":i.txid, "vout":i.vout} for i in sel_unspent]
for i in tx_out.keys(): tx_out[i] = float(tx_out[i])
if change: tx_out[change_addr] = float(change)
tx_hex = c.createrawtransaction(tx_in,tx_out)

prompt = "Transaction successfully created\nView decoded transaction?"
if user_confirm(prompt,default_yes=False):
	view_tx_data(c,[i.__dict__ for i in sel_unspent],tx_hex)

prompt = "Save transaction?"
if user_confirm(prompt,default_yes=True):
	print_tx_to_file(tx_hex,sel_unspent,send_amt,opts)