# Copyright (c) 2021, Frappe Technologies Pvt. Ltd. and Contributors
# MIT License. See license.txt

from __future__ import unicode_literals
import frappe
from frappe import _

@frappe.whitelist()
def get_all_nodes(parent, parent_name, method, company):
	'''Recursively gets all data from nodes'''
	method = frappe.get_attr(method)

	if method not in frappe.whitelisted:
		frappe.throw(_('Not Permitted'), frappe.PermissionError)

	data = method(parent, company)
	result = [dict(parent=parent, parent_name=parent_name, data=data)]

	nodes_to_expand = [{'id': d.get('id'), 'name': d.get('name')} for d in data if d.get('expandable')]

	while nodes_to_expand:
		parent = nodes_to_expand.pop(0)
		data = method(parent.get('id'), company)
		result.append(dict(parent=parent.get('id'), parent_name=parent.get('name'), data=data))
		for d in data:
			if d.get('expandable'):
				nodes_to_expand.append({'id': d.get('id'), 'name': d.get('name')})

	return result