# Copyright (c) 2013, Frappe Technologies Pvt. Ltd. and contributors
# For license information, please see license.txt

from collections import defaultdict

import frappe
from frappe import _
from frappe.utils import date_diff, flt, getdate, today

from erpnext.stock.report.stock_analytics.stock_analytics import get_period, get_period_date_ranges


def execute(filters=None):
	columns, data = [], []

	if not filters.get("age"):
		filters["age"] = 0

	data = get_data(filters)
	columns = get_columns(filters)
	chart_data = get_chart_data(data, filters)
	return columns, data, None, chart_data


def get_data(filters):
	query_filters = {"docstatus": ("<", 2)}

	fields = [
		"name",
		"status",
		"sales_order",
		"production_item",
		"qty",
		"creation",
		"produced_qty",
		"planned_start_date",
		"planned_end_date",
		"actual_start_date",
		"actual_end_date",
		"lead_time",
	]

	for field in ["sales_order", "production_item"]:
		if filters.get(field):
			query_filters[field] = ("in", filters.get(field))

	for field in ["status", "company"]:
		if filters.get(field):
			query_filters[field] = filters.get(field)

	if filters.get("based_on") == "Planned Date":
		query_filters["planned_start_date"] = (">=", filters.get("from_date"))
		query_filters["planned_end_date"] = ("<=", filters.get("to_date"))
	elif filters.get("based_on") == "Actual Date":
		query_filters["actual_start_date"] = (">=", filters.get("from_date"))
		query_filters["actual_end_date"] = ("<=", filters.get("to_date"))
	else:
		query_filters["creation"] = ("between", [filters.get("from_date"), filters.get("to_date")])

	data = frappe.get_all(
		"Work Order", fields=fields, filters=query_filters, order_by="planned_start_date asc"
	)

	res = []
	for d in data:
		start_date = d.actual_start_date or d.planned_start_date
		d.age = 0

		if d.status != "Completed":
			d.age = date_diff(today(), start_date)

		if filters.get("age") <= d.age:
			res.append(d)

	return res


def get_chart_data(data, filters):
	if filters.get("charts_based_on") == "Status":
		return get_chart_based_on_status(data)
	elif filters.get("charts_based_on") == "Age":
		return get_chart_based_on_age(data)
	else:
		return get_chart_based_on_qty(data, filters)


def get_chart_based_on_status(data):
	labels = frappe.get_meta("Work Order").get_options("status").split("\n")
	if "" in labels:
		labels.remove("")

	status_wise_data = defaultdict(int)

	for d in data:
		status_wise_data[d.status] += 1

	labels = [_(label) for label in labels]
	values = [status_wise_data[label] for label in labels]

	chart = {
		"data": {"labels": labels, "datasets": [{"name": "Qty Wise Chart", "values": values}]},
		"type": "donut",
		"height": 300,
	}

	return chart


def get_chart_based_on_age(data):
	labels = [_("0-30 Days"), _("30-60 Days"), _("60-90 Days"), _("90 Above")]

	age_wise_data = {"0-30 Days": 0, "30-60 Days": 0, "60-90 Days": 0, "90 Above": 0}

	for d in data:
		if d.age > 0 and d.age <= 30:
			age_wise_data["0-30 Days"] += 1
		elif d.age > 30 and d.age <= 60:
			age_wise_data["30-60 Days"] += 1
		elif d.age > 60 and d.age <= 90:
			age_wise_data["60-90 Days"] += 1
		else:
			age_wise_data["90 Above"] += 1

	values = [
		age_wise_data["0-30 Days"],
		age_wise_data["30-60 Days"],
		age_wise_data["60-90 Days"],
		age_wise_data["90 Above"],
	]

	chart = {
		"data": {"labels": labels, "datasets": [{"name": "Qty Wise Chart", "values": values}]},
		"type": "donut",
		"height": 300,
	}

	return chart


def get_chart_based_on_qty(data, filters):
	labels, periodic_data = prepare_chart_data(data, filters)

	pending, completed = [], []
	datasets = []

	for d in labels:
		pending.append(periodic_data.get("Pending").get(d))
		completed.append(periodic_data.get("Completed").get(d))

	datasets.append({"name": _("Pending"), "values": pending})
	datasets.append({"name": _("Completed"), "values": completed})

	chart = {
		"data": {"labels": labels, "datasets": datasets},
		"type": "bar",
		"barOptions": {"stacked": 1},
	}

	return chart


def prepare_chart_data(data, filters):
	labels = []

	periodic_data = {"Pending": {}, "Completed": {}}

	filters.range = "Monthly"

	ranges = get_period_date_ranges(filters)
	for from_date, end_date in ranges:
		period = get_period(end_date, filters)
		if period not in labels:
			labels.append(period)

		if period not in periodic_data["Pending"]:
			periodic_data["Pending"][period] = 0

		if period not in periodic_data["Completed"]:
			periodic_data["Completed"][period] = 0

		for d in data:
			if getdate(d.planned_start_date) >= from_date and getdate(d.planned_start_date) <= end_date:
				periodic_data["Pending"][period] += flt(d.qty) - flt(d.produced_qty)
				periodic_data["Completed"][period] += flt(d.produced_qty)

	return labels, periodic_data


def get_columns(filters):
	columns = [
		{
			"label": _("Id"),
			"fieldname": "name",
			"fieldtype": "Link",
			"options": "Work Order",
			"width": 100,
		},
	]

	if not filters.get("status"):
		columns.append(
			{"label": _("Status"), "fieldname": "status", "width": 100},
		)

	columns.extend(
		[
			{
				"label": _("Production Item"),
				"fieldname": "production_item",
				"fieldtype": "Link",
				"options": "Item",
				"width": 130,
			},
			{"label": _("Produce Qty"), "fieldname": "qty", "fieldtype": "Float", "width": 110},
			{"label": _("Produced Qty"), "fieldname": "produced_qty", "fieldtype": "Float", "width": 110},
			{
				"label": _("Sales Order"),
				"fieldname": "sales_order",
				"fieldtype": "Link",
				"options": "Sales Order",
				"width": 90,
			},
			{
				"label": _("Created On"),
				"fieldname": "creation",
				"fieldtype": "Date",
				"width": 150,
			},
			{
				"label": _("Planned Start Date"),
				"fieldname": "planned_start_date",
				"fieldtype": "Date",
				"width": 150,
			},
			{
				"label": _("Planned End Date"),
				"fieldname": "planned_end_date",
				"fieldtype": "Date",
				"width": 150,
			},
		]
	)

	if filters.get("status") != "Not Started":
		columns.extend(
			[
				{
					"label": _("Actual Start Date"),
					"fieldname": "actual_start_date",
					"fieldtype": "Date",
					"width": 100,
				},
				{
					"label": _("Actual End Date"),
					"fieldname": "actual_end_date",
					"fieldtype": "Date",
					"width": 100,
				},
				{"label": _("Age"), "fieldname": "age", "fieldtype": "Float", "width": 110},
			]
		)

	if filters.get("status") == "Completed":
		columns.extend(
			[
				{
					"label": _("Lead Time (in mins)"),
					"fieldname": "lead_time",
					"fieldtype": "Float",
					"width": 110,
				},
			]
		)

	return columns