__init__.py 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582
  1. # -*- coding: utf-8 -*-
  2. import os
  3. import pygame
  4. import platform
  5. #import commands
  6. import glob
  7. import json
  8. import gobject
  9. import sqlite3
  10. #from beeprint import pp
  11. from libs.roundrects import aa_round_rect
  12. ## local UI import
  13. from UI.constants import Width,Height,ICON_TYPES,RESTARTUI
  14. from UI.page import Page,PageSelector
  15. from UI.label import Label
  16. from UI.util_funcs import midRect,FileExists,ArmSystem,reconstruct_broken_string
  17. from UI.keys_def import CurKeys, IsKeyStartOrA, IsKeyMenuOrB
  18. from UI.scroller import ListScroller
  19. from UI.icon_pool import MyIconPool
  20. from UI.icon_item import IconItem
  21. from UI.multilabel import MultiLabel
  22. from UI.skin_manager import MySkinManager
  23. from UI.lang_manager import MyLangManager
  24. from UI.info_page_list_item import InfoPageListItem
  25. from UI.info_page_selector import InfoPageSelector
  26. from UI.yes_cancel_confirm_page import YesCancelConfirmPage
  27. import config
  28. class RPCStack:
  29. def __init__(self):
  30. self.stack = list()
  31. def Push(self,data):
  32. if data not in self.stack:
  33. self.stack.append(data)
  34. return True
  35. return False
  36. def Pop(self):
  37. if len(self.stack)<=0:
  38. return None,False
  39. return self.stack.pop(),True
  40. def Last(self):
  41. idx = len(self.stack) -1
  42. if idx < 0:
  43. return None
  44. else:
  45. return self.stack[ idx ]
  46. def Length(self):
  47. return len(self.stack)
  48. class GameConfirmInstallPopupUpPage(Page):
  49. pass
  50. class Aria2DownloadProcessPage(Page):
  51. _FootMsg = ["Nav.","","Pause","Back","Cancel"]
  52. _DownloaderTimer = -1
  53. _Value = 0
  54. _GID = None
  55. _PngSize = {}
  56. _FileNameLabel = None
  57. _SizeLabel = None
  58. _URLColor = MySkinManager.GiveColor('URL')
  59. _TextColor = MySkinManager.GiveColor('Text')
  60. def __init__(self):
  61. Page.__init__(self)
  62. self._Icons = {}
  63. self._CanvasHWND = None
  64. def Init(self):
  65. self._PosX = self._Index * self._Screen._Width
  66. self._Width = self._Screen._Width
  67. self._Height = self._Screen._Height
  68. self._CanvasHWND = self._Screen._CanvasHWND
  69. bgpng = IconItem()
  70. bgpng._ImgSurf = MyIconPool.GiveIconSurface("rom_download")
  71. bgpng._MyType = ICON_TYPES["STAT"]
  72. bgpng._Parent = self
  73. bgpng.Adjust(0,0,self._PngSize["bg"][0],self._PngSize["bg"][1],0)
  74. self._Icons["bg"] = bgpng
  75. self._FileNameLabel = Label()
  76. self._FileNameLabel.SetCanvasHWND(self._CanvasHWND)
  77. self._FileNameLabel.Init("", MyLangManager.TrFont("varela12"))
  78. self._SizeLabel = Label()
  79. self._SizeLabel.SetCanvasHWND(self._CanvasHWND)
  80. self._SizeLabel.Init("0/0Kb",MyLangManager.TrFont("varela12"))
  81. self._SizeLabel.SetColor( self._URLColor )
  82. def GObjectUpdateProcessInterval(self):
  83. if self._Screen.CurPage() == self and self._GID is not None:
  84. self._Value = config.RPC.tellStatus(self._GID)
  85. downloaded = 0
  86. total = 0
  87. if self._Value["status"] == "waiting":
  88. self._FileNameLabel.SetText( "waiting to download..." )
  89. if self._Value["status"] == "paused":
  90. self._FileNameLabel.SetText( "download paused..." )
  91. if self._Value["status"] == "error":
  92. self._FileNameLabel.SetText("download errors,cancel it please")
  93. if self._Value["status"] == "active":
  94. downloaded = self._Value["completedLength"]
  95. total = self._Value["totalLength"]
  96. downloaded = downloaded/1000.0/1000.0
  97. total = total/1000.0/1000.0
  98. self._SizeLabel.SetText( "%.2f" % downloaded+"/"+ "%.2f" % total +"Mb")
  99. print("Progress: %d%%" % (self._Value))
  100. self._Screen.Draw()
  101. self._Screen.SwapAndShow()
  102. return True
  103. else:
  104. return False
  105. def CheckDownload(self,aria2_gid):
  106. self._GID = aria2_gid
  107. self._DownloaderTimer = gobject.timeout_add(123, self.GObjectUpdateProcessInterval)
  108. def KeyDown(self,event):
  109. if IsKeyMenuOrB(event.key):
  110. gobject.source_remove(self._DownloaderTimer)
  111. self._DownloaderTimer = -1
  112. #if self._Downloader != None:
  113. # try:
  114. # self._Downloader.stop()
  115. # except:
  116. # print("user canceled ")
  117. self.ReturnToUpLevelPage()
  118. self._Screen.Draw()
  119. self._Screen.SwapAndShow()
  120. def Draw(self):
  121. self.ClearCanvas()
  122. self._Icons["bg"].NewCoord(self._Width/2,self._Height/2-20)
  123. self._Icons["bg"].Draw()
  124. percent = self._Value
  125. if percent < 10:
  126. percent = 10
  127. rect_ = midRect(self._Width/2,self._Height/2+33,170,17, Width,Height)
  128. aa_round_rect(self._CanvasHWND,rect_,MySkinManager.GiveColor('TitleBg'),5,0,MySkinManager.GiveColor('TitleBg'))
  129. rect2 = midRect(self._Width/2,self._Height/2+33,int(170*(percent/100.0)),17, Width,Height)
  130. rect2.left = rect_.left
  131. rect2.top = rect_.top
  132. aa_round_rect(self._CanvasHWND,rect2,MySkinManager.GiveColor('Front'),5,0,MySkinManager.GiveColor('Front'))
  133. rect3 = midRect(self._Width/2,self._Height/2+53,self._FileNameLabel._Width, self._FileNameLabel._Height,Width,Height)
  134. rect4 = midRect(self._Width/2,self._Height/2+70,self._SizeLabel._Width, self._SizeLabel._Height,Width,Height)
  135. self._FileNameLabel.NewCoord(rect3.left,rect3.top)
  136. self._SizeLabel.NewCoord(rect4.left, rect4.top)
  137. self._FileNameLabel.Draw()
  138. self._SizeLabel.Draw()
  139. def dict_factory(cursor, row):
  140. d = {}
  141. for idx, col in enumerate(cursor.description):
  142. d[col[0]] = row[idx]
  143. return d
  144. class GameStorePage(Page):
  145. _FootMsg = ["Nav","Update store","Up","Back","Select"]
  146. _MyList = []
  147. _ListFont12 = MyLangManager.TrFont("notosanscjk12")
  148. _ListFont15 = MyLangManager.TrFont("varela15")
  149. _AList = {}
  150. _Scrolled = 0
  151. _BGwidth = 320
  152. _BGheight = 240-24-20
  153. _DrawOnce = False
  154. _Scroller = None
  155. _InfoPage = None
  156. _Downloading = None
  157. _aria2_db = "aria2tasks.db"
  158. def __init__(self):
  159. Page.__init__(self)
  160. self._Icons = {}
  161. self._MyStack = RPCStack()
  162. #title path type
  163. repos = [
  164. {"title":"github.com/cuu/gamestore","file":"https://raw.githubusercontent.com/cuu/gamestore/master/index.json","type":"dir"}
  165. ]
  166. self._MyStack.Push(repos)
  167. def SyncSqlite(self):
  168. try:
  169. conn = sqlite3.connect(self._aria2_db)
  170. conn.row_factory = dict_factory
  171. c = conn.cursor()
  172. ret = c.execute("SELECT * FROM tasks").fetchall()
  173. conn.close()
  174. return ret
  175. except Exception as ex:
  176. print(ex)
  177. return None
  178. return None
  179. def SyncList(self):
  180. self._MyList = []
  181. start_x = 0
  182. start_y = 0
  183. last_height = 0
  184. repos = []
  185. stk = self._MyStack.Last()
  186. stk_lev = self._MyStack.Length()
  187. repos.extend(stk)
  188. if stk_lev == 1: # on top
  189. sqlite3_menu= self.SyncSqlite()
  190. if sqlite3_menu != None and len(sqlite3_menu) > 0:
  191. #print(sqlite3_menu)
  192. repos.extend(sqlite3_menu )
  193. print(repos)
  194. for i,u in enumerate( repos ):
  195. #print(i,u)
  196. li = InfoPageListItem()
  197. li._Parent = self
  198. li._PosX = start_x
  199. li._PosY = start_y + last_height
  200. li._Width = Width
  201. li._Fonts["normal"] = self._ListFont15
  202. li._Active = False
  203. li._ReadOnly = True
  204. li._Value = u
  205. li.Init( u["title"] )
  206. if stk_lev >1:
  207. li._ReadOnly = False
  208. elif stk_lev == 1:
  209. if "status" in u:
  210. if u["status"] == "complete":
  211. li._ReadOnly = False
  212. if i == 0:
  213. li._ReadOnly = False
  214. last_height += li._Height
  215. self._MyList.append(li)
  216. def Init(self):
  217. if self._Screen != None:
  218. if self._Screen._CanvasHWND != None and self._CanvasHWND == None:
  219. self._HWND = self._Screen._CanvasHWND
  220. self._CanvasHWND = pygame.Surface( (self._Screen._Width,self._BGheight) )
  221. self._PosX = self._Index*self._Screen._Width
  222. self._Width = self._Screen._Width ## equal to screen width
  223. self._Height = self._Screen._Height
  224. done = IconItem()
  225. done._ImgSurf = MyIconPool.GiveIconSurface("done")
  226. done._MyType = ICON_TYPES["STAT"]
  227. done._Parent = self
  228. self._Icons["done"] = done
  229. ps = InfoPageSelector()
  230. ps._Parent = self
  231. self._Ps = ps
  232. self._PsIndex = 0
  233. self.SyncList()
  234. self._Scroller = ListScroller()
  235. self._Scroller._Parent = self
  236. self._Scroller._PosX = self._Width - 10
  237. self._Scroller._PosY = 2
  238. self._Scroller.Init()
  239. self._Scroller.SetCanvasHWND(self._HWND)
  240. self._remove_page = YesCancelConfirmPage()
  241. self._remove_page._Screen = self._Screen
  242. self._remove_page._StartOrA_Event = self.RemoveGame
  243. self._remove_page._Name ="Are you sure?"
  244. self._remove_page.Init()
  245. def RemoveGame(self):
  246. if self._PsIndex > len(self._MyList) -1:
  247. return
  248. cur_li = self._MyList[self._PsIndex]
  249. #if cur_li._Active == True:
  250. # return
  251. print("Remove cur_li._Value",cur_li._Value)
  252. if "gid" in cur_li._Value:
  253. try:
  254. gid = cur_li._Value["gid"]
  255. conn = sqlite3.connect(self._aria2_db)
  256. conn.row_factory = dict_factory
  257. c = conn.cursor()
  258. ret = c.execute("SELECT * FROM tasks WHERE gid='%s'" % gid ).fetchone()
  259. if ret != None:
  260. remote_file_url = ret["file"]
  261. menu_file = remote_file_url.split("master")[1]
  262. local_menu_file = "%s/aria2download%s" % (os.path.expanduser('~'),menu_file )
  263. try:
  264. if os.path.exists(local_menu_file):
  265. os.remove(local_menu_file)
  266. if os.path.exists(local_menu_file+".aria2"):
  267. os.remove(local_menu_file+".aria2")
  268. except Exception as ex:
  269. print(ex)
  270. c.execute("DELETE FROM tasks WHERE gid = '%s'" % gid )
  271. conn.commit()
  272. conn.close()
  273. except Exception as ex:
  274. print(ex)
  275. def Click(self):
  276. if self._PsIndex > len(self._MyList) -1:
  277. return
  278. cur_li = self._MyList[self._PsIndex]
  279. #if cur_li._Active == True:
  280. # return
  281. print("cur_li._Value",cur_li._Value)
  282. if cur_li._Value["type"] == "dir":
  283. remote_file_url = cur_li._Value["file"]
  284. menu_file = remote_file_url.split("master")[1] #assume master branch
  285. local_menu_file = "%s/aria2download%s" % (os.path.expanduser('~'),menu_file )
  286. print(local_menu_file)
  287. if FileExists( local_menu_file ) == False:
  288. print(local_menu_file, "non-existed")
  289. gid,ret = config.RPC.urlDownloading(remote_file_url)
  290. if ret == False:
  291. gid = config.RPC.addUri( remote_file_url, options={"out": menu_file})
  292. self._Downloading = remote_file_url
  293. self._Screen._MsgBox.SetText("Loading")
  294. self._Screen._MsgBox.Draw()
  295. self._Screen.SwapAndShow()
  296. else:
  297. print(gid," url is downloading")
  298. self._Screen._MsgBox.SetText("Loading")
  299. self._Screen._MsgBox.Draw()
  300. self._Screen.SwapAndShow()
  301. else:
  302. #read the local_menu_file, push into stack,display menu
  303. self._Downloading = None
  304. with open(local_menu_file) as json_file:
  305. local_menu_json = json.load(json_file)
  306. print(local_menu_json)
  307. self._MyStack.Push(local_menu_json["list"])
  308. self.SyncList()
  309. self._Screen.Draw()
  310. self._Screen.SwapAndShow()
  311. else:
  312. #download the game probably
  313. remote_file_url = cur_li._Value["file"]
  314. menu_file = remote_file_url.split("master")[1]
  315. local_menu_file = "%s/aria2download%s" % (os.path.expanduser('~'),menu_file )
  316. if FileExists( local_menu_file ) == False:
  317. gid,ret = config.RPC.urlDownloading(remote_file_url)
  318. if ret == False:
  319. gid = config.RPC.addUri( remote_file_url, options={"out": menu_file})
  320. self._Downloading = remote_file_url
  321. print("stack length ",self._MyStack.Length())
  322. if self._MyStack.Length() > 1:## not on the top list page
  323. try:
  324. conn = sqlite3.connect("aria2tasks.db")
  325. c = conn.cursor()
  326. c.execute("INSERT INTO tasks(gid,title,file,type,status,fav) VALUES ('"+gid+"','"+cur_li._Value["title"]+"','"+cur_li._Value["file"]+"','"+cur_li._Value["type"]+"','active','0')")
  327. conn.commit()
  328. conn.close()
  329. except Exception as ex:
  330. print("SQLITE3 ",ex)
  331. else:
  332. print(config.RPC.tellStatus(gid,["status","totalLength","completedLength"]))
  333. self._Screen._MsgBox.SetText("Getting the game now")
  334. self._Screen._MsgBox.Draw()
  335. self._Screen.SwapAndShow()
  336. pygame.time.delay(800)
  337. self._Screen._TitleBar.Redraw()
  338. else:
  339. print("file downloaded")# maybe check it if is installed,then execute it
  340. if cur_li._Value["type"]=="launcher" and cur_li._ReadOnly == False:
  341. game_sh = os.path.join( "%s/apps/Menu/21_Indie Games/" % os.path.expanduser('~'),cur_li._Value["title"],cur_li._Value["title"]+".sh")
  342. #game_sh = reconstruct_broken_string( game_sh)
  343. print("run game: ",game_sh, os.path.exists( game_sh))
  344. self._Screen.RunEXE(game_sh)
  345. def OnLoadCb(self):
  346. self._Scrolled = 0
  347. self._PosY = 0
  348. self._DrawOnce = False
  349. #sync
  350. print("OnLoadCb")
  351. if self._MyStack.Length() == 1:
  352. self._FootMsg[2] = "Remove"
  353. self._FootMsg[1] = "Update store"
  354. else:
  355. self._FootMsg[2] = "Up"
  356. self._FootMsg[1] = ""
  357. self.SyncList()
  358. def OnReturnBackCb(self):
  359. if self._MyStack.Length() == 1:
  360. self._FootMsg[2] = "Remove"
  361. self._FootMsg[1] = "Update store"
  362. else:
  363. self._FootMsg[2] = "Up"
  364. self._FootMsg[1] = ""
  365. self.SyncList()
  366. self._Screen.Draw()
  367. self._Screen.SwapAndShow()
  368. """
  369. self.ReturnToUpLevelPage()
  370. self._Screen.Draw()
  371. self._Screen.SwapAndShow()
  372. """
  373. def KeyDown(self,event):
  374. if IsKeyMenuOrB(event.key):
  375. self.ReturnToUpLevelPage()
  376. self._Screen.Draw()
  377. self._Screen.SwapAndShow()
  378. if IsKeyStartOrA(event.key):
  379. self.Click()
  380. if self._MyStack.Length() == 1:
  381. self._FootMsg[2] = "Remove"
  382. self._FootMsg[1] = "Update store"
  383. else:
  384. self._FootMsg[2] = "Up"
  385. self._FootMsg[1] = ""
  386. self._Screen.Draw()
  387. self._Screen.SwapAndShow()
  388. if event.key == CurKeys["X"]:
  389. #print(self._MyStack.Length() )
  390. if self._MyStack.Length() == 1 and self._PsIndex > 0:
  391. self._Screen.PushPage(self._remove_page)
  392. self._remove_page._StartOrA_Event = self.RemoveGame
  393. self._Screen.Draw()
  394. self._Screen.SwapAndShow()
  395. return
  396. if self._MyStack.Length() > 1:
  397. self._MyStack.Pop()
  398. if self._MyStack.Length() == 1:
  399. self._FootMsg[2] = "Remove"
  400. self._FootMsg[1] = "Update store"
  401. else:
  402. self._FootMsg[2] = "Up"
  403. self._FootMsg[1] = ""
  404. self.SyncList()
  405. self._Screen.Draw()
  406. self._Screen.SwapAndShow()
  407. if event.key == CurKeys["Y"]:
  408. if self._MyStack.Length() == 1:
  409. self._Screen._MsgBox.SetText("Done")
  410. self._Screen._MsgBox.Draw()
  411. self._Screen.SwapAndShow()
  412. if event.key == CurKeys["Up"]:
  413. self.ScrollUp()
  414. self._Screen.Draw()
  415. self._Screen.SwapAndShow()
  416. if event.key == CurKeys["Down"]:
  417. self.ScrollDown()
  418. self._Screen.Draw()
  419. self._Screen.SwapAndShow()
  420. def Draw(self):
  421. self.ClearCanvas()
  422. if len(self._MyList) == 0:
  423. return
  424. else:
  425. if len(self._MyList) * self._MyList[0]._Height > self._Height:
  426. self._Ps._Width = self._Width - 11
  427. self._Ps.Draw()
  428. for i in self._MyList:
  429. if i._PosY > self._Height + self._Height/2:
  430. break
  431. if i._PosY < 0:
  432. continue
  433. i.Draw()
  434. self._Scroller.UpdateSize( len(self._MyList)*self._MyList[0]._Height, self._PsIndex*self._MyList[0]._Height)
  435. self._Scroller.Draw()
  436. else:
  437. self._Ps._Width = self._Width
  438. self._Ps.Draw()
  439. for i in self._MyList:
  440. if i._PosY > self._Height + self._Height/2:
  441. break
  442. if i._PosY < 0:
  443. continue
  444. i.Draw()
  445. if self._HWND != None:
  446. self._HWND.fill(MySkinManager.GiveColor("White"))
  447. self._HWND.blit(self._CanvasHWND,(self._PosX,self._PosY,self._Width, self._Height ) )
  448. class APIOBJ(object):
  449. _Page = None
  450. def __init__(self):
  451. pass
  452. def Init(self,main_screen):
  453. self._Page = GameStorePage()
  454. self._Page._Screen = main_screen
  455. self._Page._Name ="Download games"
  456. self._Page.Init()
  457. def API(self,main_screen):
  458. if main_screen !=None:
  459. main_screen.PushPage(self._Page)
  460. main_screen.Draw()
  461. main_screen.SwapAndShow()
  462. OBJ = APIOBJ()
  463. def Init(main_screen):
  464. OBJ.Init(main_screen)
  465. def API(main_screen):
  466. OBJ.API(main_screen)