From cd8320b24c9b03c6f8f55332f83997d28ae80c49 Mon Sep 17 00:00:00 2001
From: Shariq Ansari <sharique.rik@gmail.com>
Date: Mon, 11 Nov 2024 12:25:40 +0530
Subject: [PATCH] fix: frappecloud billing api helper

---
 .../frappe_providers/frappecloud_billing.py   | 83 +++++++++++++++++++
 1 file changed, 83 insertions(+)
 create mode 100644 frappe/integrations/frappe_providers/frappecloud_billing.py

diff --git a/frappe/integrations/frappe_providers/frappecloud_billing.py b/frappe/integrations/frappe_providers/frappecloud_billing.py
new file mode 100644
index 0000000000..b9b32fd327
--- /dev/null
+++ b/frappe/integrations/frappe_providers/frappecloud_billing.py
@@ -0,0 +1,83 @@
+import requests
+
+import frappe
+from frappe import _
+
+
+def get_base_url():
+	url = frappe.conf.get("saas_billing_base_url") or "https://frappecloud.com"
+	if frappe.conf.developer_mode and frappe.conf.get("saas_billing_base_url"):
+		url = frappe.conf.get("saas_billing_base_url")
+	return url
+
+
+def get_site_name():
+	site_name = frappe.local.site
+	if frappe.conf.developer_mode and frappe.conf.get("saas_billing_site_name"):
+		site_name = frappe.conf.get("saas_billing_site_name")
+	return site_name
+
+
+def get_headers():
+	# check if user is system manager
+	if frappe.get_roles(frappe.session.user).count("System Manager") == 0:
+		frappe.throw(_("You are not allowed to access this resource"))
+
+	# check if communication secret is set
+	if not frappe.conf.get("fc_communication_secret"):
+		frappe.throw(_("Communication secret not set"))
+
+	return {
+		"X-Site-Token": frappe.conf.get("fc_communication_secret"),
+		"X-Site": get_site_name(),
+	}
+
+
+@frappe.whitelist()
+def get_token_and_base_url():
+	request = requests.post(
+		f"{get_base_url()}/api/method/press.saas.api.auth.generate_access_token",
+		headers=get_headers(),
+	)
+	if request.status_code == 200:
+		return {
+			"base_url": get_base_url(),
+			"token": request.json()["message"],
+		}
+	else:
+		frappe.throw(_("Failed to generate access token"))
+
+
+@frappe.whitelist()
+def is_access_token_valid(token):
+	headers = {"Content-Type": "application/json"}
+	request = requests.post(
+		f"{get_base_url()}/api/method/press.saas.api.auth.is_access_token_valid",
+		headers,
+		json={token},
+	)
+	return request.json()["message"]
+
+
+@frappe.whitelist()
+def current_site_info():
+	request = requests.post(f"{get_base_url()}/api/method/press.saas.api.site.info", headers=get_headers())
+	if request.status_code == 200:
+		return request.json().get("message")
+	else:
+		frappe.throw(_("Failed to get site info"))
+
+
+@frappe.whitelist()
+def api(method, data=None):
+	if data is None:
+		data = {}
+	request = requests.post(
+		f"{get_base_url()}/api/method/press.saas.api.{method}",
+		headers=get_headers(),
+		json=data,
+	)
+	if request.status_code == 200:
+		return request.json().get("message")
+	else:
+		frappe.throw(_("Failed while calling API {0}", method))