#coding=utf-8
#!/usr/bin/python
import sys
sys.path.append('..') 
from base.spider import Spider
import json
import requests
import time
import re

class Spider(Spider):  # 元类 默认的元类 type
	def getName(self):
		return "阿里云盘"
	def init(self,extend=""):
		print("============{0}============".format(extend))
		pass
	def homeContent(self,filter):
		result = {}
		return result
	def homeVideoContent(self):
		result = {}
		return result
	def categoryContent(self,tid,pg,filter,extend):
		result = {}
		return result
	def searchContent(self,key,quick):
		result = {}
		return result
	def isVideoFormat(self,url):
		pass
	def manualVideoCheck(self):
		pass
	def playerContent(self,flag,id,vipFlags):
		if flag == 'AliYun':
			return self.originContent(flag,id,vipFlags)
		elif flag == 'AliYun原画':
			return self.fhdContent(flag,id,vipFlags)
		else:
			return 	{}		
	def fhdContent(self,flag,id,vipFlags):
		if not self.login():
			return {}
		ids = id.split('+')
		shareId = ids[0]
		shareToken = ids[1]
		fileId = ids[2]
		category = ids[3]
		subtitle = ids[4]
		url = self.getDownloadUrl(shareId,shareToken,fileId,category)

		noRsp = requests.get(url,headers=self.header, allow_redirects=False,verify = False)
		realUrl = ''
		if 'Location' in noRsp.headers:
			realUrl = noRsp.headers['Location']
		if 'location' in noRsp.headers and len(realUrl) == 0 :
			realUrl = noRsp.headers['location']
		newHeader = {
			"user-agent":"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.54 Safari/537.36",
			"referer":"https://www.aliyundrive.com/",
		}
		subtitleUrl = self.subtitleContent(id)
		result = {
			'parse':'0',
			'playUrl':'',
			'url':realUrl,
			'header':newHeader,
			'subt':subtitleUrl
		}
		return result
	def subtitleContent(self,id):
		ids = id.split('+')
		shareId = ids[0]
		shareToken = ids[1]
		fileId = ids[2]
		category = ids[3]
		subtitle = ids[4]
		if len(subtitle) == 0:
			return ""

		customHeader = self.header.copy()
		customHeader['x-share-token'] = shareToken
		customHeader['authorization'] = self.authorization

		jo = {
			"expire_sec": 600,
			"share_id": shareId,
			"file_id": subtitle,
            "image_url_process": "image/resize,w_1920/format,jpeg",
            "image_thumbnail_process": "image/resize,w_1920/format,jpeg",
            "get_streams_url": True
            # ,
            # "drive_id": "183237630"
		}

		downloadUrl = 'https://api.aliyundrive.com/v2/file/get_share_link_download_url'
		resultJo = requests.post(downloadUrl,json = jo,headers=customHeader).json()
		print(resultJo)
		noRsp = requests.get(resultJo['download_url'],headers=self.header, allow_redirects=False,verify = False)
		realUrl = ''
		if 'Location' in noRsp.headers:
			realUrl = noRsp.headers['Location']
		if 'location' in noRsp.headers and len(realUrl) == 0 :
			realUrl = noRsp.headers['location']
		return realUrl

	def originContent(self,flag,id,vipFlags):
		if not self.login():
			return {}
		ids = id.split('+')
		shareId = ids[0]
		shareToken = ids[1]
		fileId = ids[2]
		subtitle = ids[4]
		url = '{0}?do=push_agent&api=python&type=m3u8&share_id={1}&file_id={2}'.format(self.localProxyUrl,shareId,fileId)
		subtitleUrl = self.subtitleContent(id)
		newHeader = {
			"user-agent":"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.54 Safari/537.36",
			"referer":"https://www.aliyundrive.com/",
		}
		result = {
			'parse':'0',
			'playUrl':'',
			'url':url,
			'header':newHeader,
			'subt':subtitleUrl
		}
		return result

	def detailContent(self,array):
		tid = array[0]
		m = re.search('www.aliyundrive.com\\/s\\/([^\\/]+)(\\/folder\\/([^\\/]+))?', tid)
		col = m.groups()
		shareId = col[0]
		fileId = col[2]

		infoUrl = 'https://api.aliyundrive.com/adrive/v3/share_link/get_share_by_anonymous'

		infoForm = {'share_id':shareId}
		infoRsp = requests.post(infoUrl,json = infoForm,headers=self.header)
		infoJo = json.loads(infoRsp.text)

		infoJa = []
		if 'file_infos' in infoJo:
			infoJa = infoJo['file_infos']
		if len(infoJa) <= 0 :
			return ''
		fileInfo = {}
		
		fileInfo = infoJa[0]

		if fileId == None or len(fileId) <= 0:
			fileId = fileInfo['file_id']

		vodList = {
			'vod_id':tid,
			'vod_name':infoJo['share_name'],
			'vod_pic':infoJo['avatar'],
			'vod_content':tid,
			'vod_play_from':'AliYun$$$AliYun原画'
		}
		fileType = fileInfo['type']
		if fileType != 'folder':
			if fileType != 'file' or fileInfo['category'] != video:
				return ''
			fileId = 'root'

		shareToken = self.getToken(shareId,'')
		hashMap = {}
		self.listFiles(hashMap,shareId,shareToken,fileId)

		sortedMap = sorted(hashMap.items(), key=lambda x: x[0])
		arrayList = []
		playList = []

		for sm in sortedMap:
			arrayList.append(sm[0]+'$'+sm[1])
		playList.append('#'.join(arrayList))
		playList.append('#'.join(arrayList))
		vodList['vod_play_url'] = '$$$'.join(playList)

		result = {
			'list':[vodList]
		}
		return result
		
	authorization = ''
	timeoutTick = 0
	localTime = 0
	expiresIn = 0
	shareTokenMap = {}
	expiresMap = {}
	localMedia = {}
	header = {
		"Referer":"https://www.aliyundrive.com/",
		"User-Agent":"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.54 Safari/537.36"
	}
	localProxyUrl = 'http://127.0.0.1:UndCover/proxy'

	def redirectResponse(tUrl):
		rsp = requests.get(tUrl, allow_redirects=False,verify = False)
		if 'Location' in rsp.headers:
			return redirectResponse(rsp.headers['Location'])
		else:
			return rsp

	def getDownloadUrl(self,shareId,token,fileId,category):
		lShareId = shareId
		lFileId = fileId
		params = {
			"share_id": lShareId,
			"category": "live_transcoding",
			"file_id": lFileId,
			"template_id": ""
		}
		customHeader = self.header.copy()
		customHeader['x-share-token'] = token
		customHeader['authorization'] = self.authorization
		url = 'https://api.aliyundrive.com/v2/file/get_share_link_video_preview_play_info'
		if category == 'video':
			rsp = requests.post(url,json = params,headers=customHeader)
			rspJo = json.loads(rsp.text)
			lShareId = rspJo['share_id']
			lFileId = rspJo['file_id']
		jo = {

		}
		if category == 'video':
			jo['share_id'] = lShareId
			jo['file_id'] = lFileId
			jo['expire_sec'] = 600
		if category == 'audio':
			jo['share_id'] = lShareId
			jo['file_id'] = lFileId
			jo['get_audio_play_info'] = True
		downloadUrl = 'https://api.aliyundrive.com/v2/file/get_share_link_download_url'
		downloadRsp = requests.post(downloadUrl,json = jo,headers=customHeader)
		resultJo = json.loads(downloadRsp.text)
		return resultJo['download_url']

	def getMediaSlice(self,shareId,token,fileId):
		params = {
			"share_id": shareId,
			"category": "live_transcoding",
			"file_id": fileId,
			"template_id": ""
		}
		customHeader = self.header.copy()
		customHeader['x-share-token'] = token
		customHeader['authorization'] = self.authorization
		url = 'https://api.aliyundrive.com/v2/file/get_share_link_video_preview_play_info'

		rsp = requests.post(url,json = params,headers=customHeader)
		rspJo = json.loads(rsp.text)

		quality = ['FHD','HD','SD']
		videoList = rspJo['video_preview_play_info']['live_transcoding_task_list']
		highUrl = ''
		for q in quality:
			if len(highUrl) > 0:
				break
			for video in videoList:
				if(video['template_id'] == q):
					highUrl = video['url']
					break
			if len(highUrl) == 0:
				highUrl = videoList[0]['url']

		noRsp = requests.get(highUrl,headers=self.header, allow_redirects=False,verify = False)
		m3u8Url = ''
		if 'Location' in noRsp.headers:
			m3u8Url = noRsp.headers['Location']
		if 'location' in noRsp.headers and len(m3u8Url) == 0 :
			m3u8Url = noRsp.headers['location']
		m3u8Rsp = requests.get(m3u8Url,headers=self.header)
		m3u8Content = m3u8Rsp.text

		tmpArray = m3u8Url.split('/')[0:-1]
		host = '/'.join(tmpArray) + '/'

		m3u8List = []
		mediaMap = {}
		slices = m3u8Content.split("\n")
		count = 0
		for slice in slices:
			tmpSlice = slice
			if 'x-oss-expires' in tmpSlice:
				count = count + 1
				mediaMap[str(count)] = host+tmpSlice

				tmpSlice = "{0}?do=push_agent&api=python&type=media&share_id={1}&file_id={2}&media_id={3}".format(self.localProxyUrl,shareId,fileId,count)
			m3u8List.append(tmpSlice)

		self.localMedia[fileId] = mediaMap
		
		return '\n'.join(m3u8List)

	def proxyMedia(self,map):
		shareId = map['share_id']
		fileId = map['file_id']
		mediaId = map['media_id']
		shareToken = self.getToken(shareId,'')

		refresh = False
		url = ''
		ts = 0
		if fileId in self.localMedia:
			fileMap = self.localMedia[fileId]
			if mediaId in fileMap:
				url = fileMap[mediaId]
		if len(url) > 0:
			ts = int(self.regStr(url,"x-oss-expires=(\\d+)&"))

		self.localTime = int(time.time())

		if ts - self.localTime <= 60:
			self.getMediaSlice(shareId,shareToken,fileId)
			url = self.localMedia[fileId][mediaId]
			
		action = {
			'url':url,
			'header':self.header,
			'param':'',
			'type':'stream',
			'after':''
		}
		return [200, "video/MP2T", action, ""]

	def proxyM3U8(self,map):
		shareId = map['share_id']
		fileId = map['file_id']

		shareToken = self.getToken(shareId,'')
		content = self.getMediaSlice(shareId,shareToken,fileId)

		action = {
			'url':'',
			'header':'',
			'param':'',
			'type':'string',
			'after':''
		}

		return [200, "application/octet-stream", action, content]

	def localProxy(self,param):
		if not self.login():
			return {}
		typ = param['type']
		if typ == "m3u8":
			return self.proxyM3U8(param)
		if typ == "media":
			return self.proxyMedia(param)
		return None

	def getToken(self,shareId,sharePwd):
		self.localTime = int(time.time())
		shareToken = ''
		if shareId in self.shareTokenMap:
			shareToken = self.shareTokenMap[shareId]
			# todo
			expire = self.expiresMap[shareId]
			if len(shareToken) > 0 and expire - self.localTime > 600:
				return shareToken
		params = {
			'share_id':shareId,
			'share_pwd':sharePwd
		}
		url = 'https://api.aliyundrive.com/v2/share_link/get_share_token'
		rsp = requests.post(url,json = params,headers=self.header)
		jo = json.loads(rsp.text)
		newShareToken = jo['share_token']
		self.expiresMap[shareId] = self.localTime + int(jo['expires_in'])
		self.shareTokenMap[shareId] = newShareToken

		# print(self.expiresMap)
		# print(self.shareTokenMap)

		return newShareToken

	def listFiles(self,map,shareId,shareToken,fileId,subtitle={}):
		url = 'https://api.aliyundrive.com/adrive/v3/file/list'
		newHeader = self.header.copy()
		newHeader['x-share-token'] = shareToken
		params = {
			'image_thumbnail_process':'image/resize,w_160/format,jpeg',
			'image_url_process':'image/resize,w_1920/format,jpeg',
			'limit':200,
			'order_by':'updated_at',
			'order_direction':'DESC',
			'parent_file_id':fileId,
			'share_id':shareId,
			'video_thumbnail_process':'video/snapshot,t_1000,f_jpg,ar_auto,w_300'
		}
		maker = ''
		arrayList = []
		for i in range(1,51):
			if i >= 2 and len(maker) == 0:
				break
			params['marker'] = maker			
			rsp = requests.post(url,json = params,headers=newHeader)
			jo = json.loads(rsp.text)
			ja = jo['items']
			for jt in ja:
				if jt['type'] == 'folder':
					arrayList.append(jt['file_id'])
				else:
					if 'video' in jt['mime_type'] or 'video' in jt['category']:
						repStr = jt['name'].replace("#", "_").replace("$", "_").replace(jt['file_extension'],'')[0:-1]
						map[repStr] = shareId + "+" + shareToken + "+" + jt['file_id'] + "+" + jt['category'] + "+"
					elif 'others' == jt['category'] and ('srt' == jt['file_extension'] or 'ass' == jt['file_extension']):
						repStr = jt['name'].replace("#", "_").replace("$", "_").replace(jt['file_extension'],'')[0:-1]
						subtitle[repStr] = jt['file_id']
			maker = jo['next_marker']
			i = i + 1

		for item in arrayList:
			self.listFiles(map,shareId,shareToken,item,subtitle)
		for key in map.keys():
			for subKey in subtitle.keys():
				if key in subKey and map[key][-1] == "+":
					map[key]=map[key]+subtitle[subKey]
					break

	def login(self):
		self.localTime = int(time.time())
		url = 'https://api.aliyundrive.com/token/refresh'
		if len(self.authorization) == 0 or self.timeoutTick - self.localTime <= 600:
			form = {
				'refresh_token':'ab0b9a7555e84175bbc6f8e60310ae49'				
			}
			rsp = requests.post(url,json = form,headers=self.header)
			jo = json.loads(rsp.text)
			if rsp.status_code == 200:
				self.authorization = jo['token_type'] + ' ' + jo['access_token']
				self.expiresIn = int(jo['expires_in'])
				self.timeoutTick = self.localTime + self.expiresIn
				return True
			return False
		else:
			return True

			# print(self.authorization)
			# print(self.timeoutTick)
			# print(self.localTime)
			# print(self.expiresIn)