webport/core/handler.py

#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
import json
#from BaseHTTPServer import BaseHTTPRequestHandler,HTTPServer
from http.server import BaseHTTPRequestHandler
import logging
import os
from utils.utils import get_list_overlays, load_config, write_config, sort_install_pkg
from utils.package import search
from utils.findfsdb import on_find 
from core.route import Router
from io import BytesIO
#repl = '<?xml version=\'1.0\' encoding=\'utf-8\'?>\n<!DOCTYPE repositories SYSTEM "http://www.gentoo.org/dtd/repositories.dtd">'

def  main():
	try:
		with open('./pkgs.json', 'tr') as fn:
			data = fn.read()
			pkg_list = json.loads(data)
			print(pkg_list)
		return json.dumps(pkg_list)	
	except Exception(e):
		print(e)
		return "404"

def set_settings_app(params):
	config = load_config()
	#param = self.path.replace("/?st_app=", "") 
	list_param = param.split(',')
	print(list_param)
	for i in list_param:
		if i.startswith('port'):
			port = int(i.split('=')[1])	
		elif i.startswith('Lang'):
			Lang = i.split('=')[1]
				
	write_config(port, Lang)
	print(config)
	print(param)

def search(pkg):
	#param = self.path.replace("/find?pkg=", "")  #request['params']['name']
	Pk_list = []
	Search_result = {}
	if len(param.split('/')) == 2:
		param = param.split('/')[1]
		P_list = on_find(param)
		print(p_list)
	if len(P_list) == 0:
		print("Never Found")
		self.r_t = str(json.dumps({"Package_result": p_list}))
	else:
		for p in p_list:
			print(p)
	if len(param.split("/")) == 2:
		Pk_list.append(search(param.split("/")[1]))
	else:
		Pk_list.append(search(param))
		#print(pk)
	Search_result = {"Package_result": Pk_list}
	#self.r_t = str(json.dumps(search_result))
	return str(json.dumps(Search_result))
	#self.r_t = json.dumps(pkg_list)
def  get_settings_app():
	 return str(json.dumps(load_config()))




class Handler(BaseHTTPRequestHandler):
	def get_data(self):
		#length = int(self.headers['content-length'])
		data = self.rfile.read(length)
		return(str(data))

	def route(self):
		if self.path in route:
			return true
		else:
			return false

	def parse_url(self):
		pass
	
	def do_GET(self):
		
		#print(Router.parse_url(self))
		#print("data:\t" + self.rfile.read())
		#request = Router.parse_url(self)
		self.send_response(200)
		self.p_list = []
		self.r_t  = ""
		#length = int(self.headers['content-length'])
		#self.send_header(['Access-Control-Allow-Origin', '*'],
		#self.send_header(['Access-Control-Allow-Methods', 'GET'],
		#self.send_header(['Access-Control-Allow-Headers', 'X-Requested-With,content-type'])
		#self.send_header(['Access-Control-Allow-Credentials', true])
		#self.end_headers()

		self.end_headers()

		if self.path =="/":
			#print("data:\t" + self.get_data())
			"""
			with open('./views/index.html', 'tr') as f: 
				self.r_t=f.read()
			print(self.client_address)
			"""
			try:
				with open('./pkgs.json', 'tr') as fn:
					data = fn.read()
					pkg_list = json.loads(data)
					print(pkg_list)
			except Exception(e):
				print(e)
			self.r_t = json.dumps(pkg_list)
		elif self.path  == '/main':
			#print(self.rfile.read())
			with open("./README.txt", 'r') as f:
				self.r_t = str(f.read())
				#self.r_t = s
		elif self.path == '/ovelays':
			overlays = get_list_overlays()
			#print(ovls)
			if overlays:                  #			== "":
				overlays ="Error"

			self.r_t=json.dumps({"repositories": overlays})
			
		elif self.path == "":
			with open('./favicon.png', 'rb') as f:
				self.r_t = f.read()
				
		elif self.path == '/logo.png':
			pass
		
		elif self.path.startswith("/?st_app="):
			config = load_config()
			param = self.path.replace("/?st_app=", "") 
			list_param = param.split(',')
			print(list_param)
			for i in list_param:
				if i.startswith('port'):
					port = int(i.split('=')[1])	
				elif i.startswith('Lang'):
					Lang = i.split('=')[1]
				
			write_config(port, Lang)
			print(config)
			print(param)

		elif self.path.startswith("/find?pkg="):
			param = self.path.replace("/find?pkg=", "")  #request['params']['name']
			pk_list = []
			search_result = {}
			if len(param.split('/')) == 2:
				param = param.split('/')[1]
			p_list = on_find(param)
			print(p_list)
			if len(p_list) == 0:
				print("Never Found")
				self.r_t = str(json.dumps({"Package_result": p_list}))
			else:
				for p in p_list:
					print(p)
			if len(param.split("/")) == 2:
				pk_list.append(search(param.split("/")[1]))
			else:
				pk_list.append(search(param))
				#print(pk)
			search_result = {"Package_result": pk_list}
			self.r_t = str(json.dumps(search_result))

		elif self.path == "/get_settings_app":

			self.r_t = str(json.dumps(load_config()))

		elif self.path == '/get_portage':

			#self.r_t = str(sort_inatll_pkg())
			self.r_t = str(json.dumps(scan_config_portage()))

		else:
			self.send_response(404)
			
			self.end_headers()
			print(str(self.client_address[0]) +"\t" + str(404))

        # Send the html message
		#self.wfile.write(bytes(self.r_t, "utf-8"))
	def response(self):
		try:
			return self.wfile.write(BytesIO(self.r_t).getvalue())
		except TypeError:
			#print("TypeError")
			return self.wfile.write(bytes(self.r_t, 'UTF-8'))