##############################################################################
#
# Copyright (c) 2005-2006 TINY SPRL. (http://tiny.be) All Rights Reserved.
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# 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 2
# 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, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
##############################################################################

import time
import wizard
import ir
import pooler
from osv.osv import except_osv
import netsvc

invoice_form = """<?xml version="1.0"?>
<form string="Create invoices">
	<separator colspan="4" string="Create invoices" />
	<field name="journal_id"/>
	<newline/>
	<field name="group"/>
	<newline/>
	<field name="type"/>
</form>
"""

invoice_fields = {
	'journal_id' : {
		'string':'Destination Journal',
		'type':'many2one',
		'relation':'account.journal',
		'required':True
	},
	'group' : {'string':'Group by partner', 'type':'boolean'}
}

def make_default(val):
	def fct(obj, cr, uid):
		return val
	return fct

def _get_type(self, cr, uid, data, context):
	picking_obj=pooler.get_pool(cr.dbname).get('stock.picking')
	pick=picking_obj.browse(cr, uid, [data['id']])[0]
	if pick.loc_move_id:
		usage=pick.loc_move_id.usage
	else:
		usage=pick.move_lines[0].location_id.usage
	if pick.type=='out' and usage=='supplier':
		type='in_refund'
	elif pick.type=='out' and usage=='customer':
		type='out_invoice'
	elif pick.type=='in' and usage=='supplier':
		type='in_invoice'
	elif pick.type=='in' and usage=='customer':
		type='out_refund'
	else:
		type='out_invoice'
	invoice_fields['type']={'string':'Type', 'type':'selection', 'default':make_default(type),
		'selection':[
			('out_invoice','Customer Invoice'),
			('in_invoice','Supplier Invoice'),
			('out_refund','Customer Refund'),
			('in_refund','Supplier Refund'),
			], 'required':True}
	return {}

def _create_invoice(self, cr, uid, data, context):
	pool = pooler.get_pool(cr.dbname)
	picking_obj = pooler.get_pool(cr.dbname).get('stock.picking')
	res = picking_obj.action_invoice_create(cr, uid, data['ids'],journal_id=data['form']['journal_id'],group=data['form']['group'], type=data['form']['type'], context= context)
	for pick_id, inv_id in res.items():
		pool.get('stock.picking').write(cr, uid, [pick_id], {'invoice_state':'invoiced'})
		invoice = pool.get('account.invoice').browse(cr, uid, inv_id, context)
		picking = pool.get('stock.picking').browse(cr, uid, pick_id, context)
		if not picking.carrier_id:
			continue
		grid_id = pool.get('delivery.carrier').grid_get(cr, uid, [picking.carrier_id.id], picking.address_id.id, context)
		price = pool.get('delivery.grid').get_price_from_picking(cr, uid, grid_id, 
			invoice.amount_untaxed, picking.weight,
			picking.volume, context) or 0.0
		a =  picking.carrier_id.product_id.product_tmpl_id.property_account_income
		if not a:
			a = picking.carrier_id.product_id.categ_id.property_account_income_categ
		account_id =  a[0]
		pool.get('account.invoice.line').create(cr, uid, {
			'name':picking.carrier_id.name,
			'invoice_id': inv_id,
			'uos_id': picking.carrier_id.product_id.uos_id.id,
			'product_id':picking.carrier_id.product_id.id,
			'account_id':account_id,
			'price_unit': price,
			'quantity': 1,
			'invoice_line_tax_id': [(6,0,[x.id for x in (picking.carrier_id.product_id.taxes_id or [])  ])],
			'note': '',
		})
	return res

def _action_open_window(self, cr, uid, data, context):
	res = _create_invoice(self, cr, uid, data, context)
	iids = res.values()
	form = data['form']
	return {
		'domain': "[('id','in', ["+','.join(map(str,iids))+"])]",
		'name': 'Invoices',
		'view_type': 'form',
		'view_mode': 'tree,form',
		'res_model': 'account.invoice',
		'view_id': False,
		'context': "{'type':'out_invoice'}",
		'type': 'ir.actions.act_window'
	}

class make_invoice_onshipping(wizard.interface):
	states = {
		'init' : {
			'actions' : [_get_type],
			'result' : {'type' : 'form',
				    'arch' : invoice_form,
				    'fields' : invoice_fields,
				    'state' : [('end', 'Cancel'),('create_invoice', 'Create invoice') ]}
		},
		'create_invoice' : {
			'actions' : [],
			'result' : {'type' : 'action', 'action': _action_open_window, 'state' : 'end'}
		},
	}
make_invoice_onshipping("delivery.invoice_onshipping")