#/usr/bin/env python 
# -*- coding: UTF-8 -*-
import os, sys
import json
from io import StringIO  
from utils.utils import get_list_overlays
from GenRecovers import ScanRecoverFile
# создаём разширеную базу пактов с предлизительно структурой
#pkg_list ={"category": [{name: name_pkg, versions:[list_version]}]}
# и клладём это всё в json
#

AliaseCategory ={"www-client":	"Браузеры",
				 "www-server": "Сервера",
				 "game":"Игры",
				 "dev-lang":"Програмироыввние",
				 "media-video": "Видео",
				 "media-sound": "Аудио"
				 }
def create_db():
	overlays = get_list_overlays()
	recovers = ScanRecoverFile()
	pkg_list ={}
	port_dir =["/var/lib/layman/", "/usr/portage"]
	if not os.path.exists('./pkgs.json'):
		print("Create ramdb")
		pkg_name =""
		all_pkgs = []
		with open('./pkgs.json', 'a') as fn:
			for p in port_dir:
				for d, dirs, files in os.walk(p):
					for f in files:
						if f.endswith('.ebuild'):
							#print(d +"/" + f)
							pkg_name =""
							try:
								#ver=int(f.replace('.ebuild', '').split('-')[-1][0])
								for pn in f.replace('.ebuild', '').split('-')[:-1]:
									pkg_name = pkg_name + pn + "-"
							except TypeError:
								for pn in f.replace('.ebuild', '').split('-')[:-2]:
									pkg_name = pkg_name + pn + "-"
							except Exception as e:
								print(e)

							if not pkg_name[:-1] in all_pkgs:

								#print(str(d.split("/")[-2] +"/" + d.split("/")[-1] +"\n"))
								if str(d.split("/")[-2] +"/" + d.split("/")[-1]) in AliaseCategory:
									all_pkgs.append(AliaseCategory[d])
								else:
									all_pkgs.append(str(d.split("/")[-2] +"/" + d.split("/")[-1]))
									print(all_pkgs[str(d.split("/")[-2] +"/" + d.split("/")[-1])])

							if d.split("/")[-2] not in pkg_list.keys():
								pkg_list[str(d.split("/")[-2])] = [] 
								pkg_list[d.split("/")[-2]].append(d.split('/')[-1])
								#print([d.split("/")[-2]])
							else:
								if d.split('/')[-1] not in pkg_list[d.split('/')[-2]]:
									pkg_list[d.split("/")[-2]].append(d.split('/')[-1])
									#print([d.split('/')[-1]])

			fn.write(json.dumps({"all_pkgs": all_pkgs,"Catalog": pkg_list, "aliases": AliaseCategory,  "overlays": overlays, "recovers": recovers}))
		print(len(pkg_list))

		#print("Found:\t"+ str(len(pkg_list.keys())) +  " category\n" + str(len(all_pkgs)) +"packages\n")

def on_find(p_v):
	if not os.path.exists('./pkgs.json'):
		create_db()
	p = []
	ret_p =""
	ret = {}
	with open('./pkgs.json', 'r') as  fn:
		data = fn.read()
		pkg_list= fn.read()
		
		pkg_list = data.split("\n")
		for i in pkg_list:
		if p_v in i and not i in p:
			print(i)
			p.append(str(i))
			ret_p = ret_p +"\t" + i
	print("Find in template:\t" + str(len(p)))

	#ret = {"Name": ret_p.split("\t")}
	#print(ret_p)	
	print(p)
	return pkg_list #json.dumps(ret)

if __name__ == '__main__':
	if not os.path.exists('./pkgs.txt'):
		create_db()
	if len(sys.argv) >=2:
		on_find(sys.argv[1])
	else:
		print("No element to  find")