from __future__ import unicode_literals

import json

import frappe
from frappe.custom.doctype.custom_field.custom_field import create_custom_fields


def execute():
	company = frappe.get_all('Company', filters = {'country': 'India'})
	if not company:
		return

	# move hidden einvoice fields to a different section
	custom_fields = {
		'Sales Invoice': [
			dict(fieldname='einvoice_section', label='E-Invoice Fields', fieldtype='Section Break', insert_after='gst_vehicle_type',
			print_hide=1, hidden=1),

			dict(fieldname='ack_no', label='Ack. No.', fieldtype='Data', read_only=1, hidden=1, insert_after='einvoice_section',
				no_copy=1, print_hide=1),

			dict(fieldname='ack_date', label='Ack. Date', fieldtype='Data', read_only=1, hidden=1, insert_after='ack_no', no_copy=1, print_hide=1),

			dict(fieldname='irn_cancel_date', label='Cancel Date', fieldtype='Data', read_only=1, hidden=1, insert_after='ack_date',
				no_copy=1, print_hide=1),

			dict(fieldname='signed_einvoice', label='Signed E-Invoice', fieldtype='Code', options='JSON', hidden=1, insert_after='irn_cancel_date',
				no_copy=1, print_hide=1, read_only=1),

			dict(fieldname='signed_qr_code', label='Signed QRCode', fieldtype='Code', options='JSON', hidden=1, insert_after='signed_einvoice',
				no_copy=1, print_hide=1, read_only=1),

			dict(fieldname='qrcode_image', label='QRCode', fieldtype='Attach Image', hidden=1, insert_after='signed_qr_code',
				no_copy=1, print_hide=1, read_only=1),

			dict(fieldname='einvoice_status', label='E-Invoice Status', fieldtype='Select', insert_after='qrcode_image',
				options='\nPending\nGenerated\nCancelled\nFailed', default=None, hidden=1, no_copy=1, print_hide=1, read_only=1),

			dict(fieldname='failure_description', label='E-Invoice Failure Description', fieldtype='Code', options='JSON',
				hidden=1, insert_after='einvoice_status', no_copy=1, print_hide=1, read_only=1)
		]
	}
	create_custom_fields(custom_fields, update=True)

	if frappe.db.exists('E Invoice Settings') and frappe.db.get_single_value('E Invoice Settings', 'enable'):
		frappe.db.sql('''
			UPDATE `tabSales Invoice` SET einvoice_status = 'Pending'
			WHERE
				posting_date >= '2021-04-01'
				AND ifnull(irn, '') = ''
				AND ifnull(`billing_address_gstin`, '') != ifnull(`company_gstin`, '')
				AND ifnull(gst_category, '') in ('Registered Regular', 'SEZ', 'Overseas', 'Deemed Export')
		''')

		# set appropriate statuses
		frappe.db.sql('''UPDATE `tabSales Invoice` SET einvoice_status = 'Generated'
			WHERE ifnull(irn, '') != '' AND ifnull(irn_cancelled, 0) = 0''')

		frappe.db.sql('''UPDATE `tabSales Invoice` SET einvoice_status = 'Cancelled'
			WHERE ifnull(irn_cancelled, 0) = 1''')

	# set correct acknowledgement in e-invoices
	einvoices = frappe.get_all('Sales Invoice', {'irn': ['is', 'set']}, ['name', 'signed_einvoice'])

	if einvoices:
		for inv in einvoices:
			signed_einvoice = inv.get('signed_einvoice')
			if signed_einvoice:
				signed_einvoice = json.loads(signed_einvoice)
				frappe.db.set_value('Sales Invoice', inv.get('name'), 'ack_no', signed_einvoice.get('AckNo'), update_modified=False)
				frappe.db.set_value('Sales Invoice', inv.get('name'), 'ack_date', signed_einvoice.get('AckDt'), update_modified=False)