# -*- coding: utf-8 -*-
# Copyright (c) 2020, Frappe Technologies Pvt. Ltd. and contributors
# For license information, please see license.txt

from __future__ import unicode_literals
import frappe, requests, json
from frappe import _
from frappe.utils import get_site_url, get_url_to_form, get_link_to_form
from frappe.model.document import Document
from frappe.utils.file_manager import get_file, get_file_path
from six.moves.urllib.parse import urlencode

class LinkedInSettings(Document):
	@frappe.whitelist()
	def get_authorization_url(self):
		params = urlencode({
			"response_type":"code",
			"client_id": self.consumer_key,
			"redirect_uri": "{0}/api/method/verp.crm.doctype.linkedin_settings.linkedin_settings.callback?".format(frappe.utils.get_url()),
			"scope": "r_emailaddress w_organization_social r_basicprofile r_liteprofile r_organization_social rw_organization_admin w_member_social"
		})

		url = "https://www.linkedin.com/oauth/v2/authorization?{}".format(params)

		return url

	def get_access_token(self, code):
		url = "https://www.linkedin.com/oauth/v2/accessToken"
		body = {
			"grant_type": "authorization_code",
			"code": code,
			"client_id": self.consumer_key,
			"client_secret": self.get_password(fieldname="consumer_secret"),
			"redirect_uri": "{0}/api/method/verp.crm.doctype.linkedin_settings.linkedin_settings.callback?".format(frappe.utils.get_url()),
		}
		headers = {
			"Content-Type": "application/x-www-form-urlencoded"
		}

		response = self.http_post(url=url, data=body, headers=headers)
		response = frappe.parse_json(response.content.decode())
		self.db_set("access_token", response["access_token"])

	def get_member_profile(self):
		headers = {
			"Authorization": "Bearer {}".format(self.access_token)
		}
		url = "https://api.linkedin.com/v2/me"
		response = requests.get(url=url, headers=headers)
		response = frappe.parse_json(response.content.decode())

		frappe.db.set_value(self.doctype, self.name, {
			"person_urn": response["id"],
			"account_name": response["vanityName"],
			"session_status": "Active"
		})
		frappe.local.response["type"] = "redirect"
		frappe.local.response["location"] = get_url_to_form("LinkedIn Settings","LinkedIn Settings")

	def post(self, text, media=None):
		if not media:
			return self.post_text(text)
		else:
			media_id = self.upload_image(media)

			if media_id:
				return self.post_text(text, media_id=media_id)
			else:
				frappe.log_error("Failed to upload media.","LinkedIn Upload Error")


	def upload_image(self, media):
		media = get_file_path(media)
		register_url = "https://api.linkedin.com/v2/assets?action=registerUpload"
		body = {
			"registerUploadRequest": {
				"recipes": ["urn:li:digitalmediaRecipe:feedshare-image"],
				"owner": "urn:li:organization:{0}".format(self.company_id),
				"serviceRelationships": [{
					"relationshipType": "OWNER",
					"identifier": "urn:li:userGeneratedContent"
				}]
			}
		}
		headers = {
			"Authorization": "Bearer {}".format(self.access_token)
		}
		response = self.http_post(url=register_url, body=body, headers=headers)

		if response.status_code == 200:
			response = response.json()
			asset = response["value"]["asset"]
			upload_url = response["value"]["uploadMechanism"]["com.linkedin.digitalmedia.uploading.MediaUploadHttpRequest"]["uploadUrl"]
			headers['Content-Type']='image/jpeg'
			response = self.http_post(upload_url, headers=headers, data=open(media,"rb"))
			if response.status_code < 200 and response.status_code > 299:
				frappe.throw(_("Error While Uploading Image"), title="{0} {1}".format(response.status_code, response.reason))
				return None
			return asset

		return None

	def post_text(self, text, media_id=None):
		url = "https://api.linkedin.com/v2/shares"
		headers = {
			"X-Restli-Protocol-Version": "2.0.0",
			"Authorization": "Bearer {}".format(self.access_token),
			"Content-Type": "application/json; charset=UTF-8"
		}
		body = {
			"distribution": {
				"linkedInDistributionTarget": {}
			},
			"owner":"urn:li:organization:{0}".format(self.company_id),
			"subject": "Test Share Subject",
			"text": {
				"text": text
			}
		}

		if media_id:
			body["content"]= {
				"contentEntities": [{
					"entity": media_id
				}],
				"shareMediaCategory": "IMAGE"
			}

		response = self.http_post(url=url, headers=headers, body=body)
		return response

	def http_post(self, url, headers=None, body=None, data=None):
		try:
			response = requests.post(
				url = url,
				json = body,
				data = data,
				headers = headers
			)
			if response.status_code not in [201,200]:
				raise

		except Exception as e:
			content = json.loads(response.content)

			if response.status_code == 401:
				self.db_set("session_status", "Expired")
				frappe.db.commit()
				frappe.throw(content["message"], title="LinkedIn Error - Unauthorized")
			elif response.status_code == 403:
				frappe.msgprint(_("You Didn't have permission to access this API"))
				frappe.throw(content["message"], title="LinkedIn Error - Access Denied")
			else:
				frappe.throw(response.reason, title=response.status_code)

		return response

@frappe.whitelist(allow_guest=True)
def callback(code=None, error=None, error_description=None):
	if not error:
		linkedin_settings = frappe.get_doc("LinkedIn Settings")
		linkedin_settings.get_access_token(code)
		linkedin_settings.get_member_profile()
		frappe.db.commit()
	else:
		frappe.local.response["type"] = "redirect"
		frappe.local.response["location"] = get_url_to_form("LinkedIn Settings","LinkedIn Settings")
