#!/usr/bin/env python
from flask import Flask, send_from_directory, render_template, session, abort
from flask_kerberos import init_kerberos, requires_authentication
from classes import Network, Channel, Log
from urllib.parse import quote_plus
import sys, os

app = Flask(__name__)
app.config.from_pyfile('application.cfg', silent=True)

app.jinja_env.add_extension('jinja2_highlight.HighlightExtension')
app.jinja_env.extend(jinja2_highlight_cssclass = 'codehilite')

if not "KRB5_KTNAME" in os.environ:
	try:
		os.environ['KRB5_KTNAME'] = app.config['KRB5_KTNAME']
	except KeyError:
		print("Error: No KEYTAB specified in config and \
				KRB5_KTNAME envvar not set",
				file=sys.stderr)
		sys.exit(1)

init_kerberos(app)


def get_files(directory):
	files = os.listdir(directory)
	files.sort()
	return files

@app.route('/')
def index():
	authenticated()
	networks = []
	for network_name in get_files(app.config['ZNC_LOG_DIR']):
		network_url = '{}/{}'.format(app.config['URL'], network_name)
		channels = []

		for channel_name in get_files(os.path.join(app.config['ZNC_LOG_DIR'], network_name)):
			channel_url  = '{}/{}'.format(network_url, quote_plus(channel_name))
			channel = Channel(channel_name, channel_url, '')
			channels += [channel]

		networks += [Network(network_name, network_url, channels)]
	return(render_template('list_networks.html', networks=networks, url=app.config['URL']))

@app.route('/<network_name>')
def get_network(network_name):
	authenticated()
	network_url = '{}/{}'.format(app.config['URL'], network_name)

	channels = []
	for channel_name in get_files(os.path.join(app.config['ZNC_LOG_DIR'], network_name)):
		channel_url  = '{}/{}'.format(network_url, quote_plus(channel_name))
		channel = Channel(channel_name, channel_url, '')

		channels += [channel]

	network = Network(network_name, network_url, channels)
	return(render_template('show_network.html', network=network, url=app.config['URL']))

@app.route('/<network_name>/<channel_name>')
def channel_logs(network_name, channel_name):
	authenticated()
	network_url = '{}/{}'.format(app.config['URL'], network_name)
	channel_url = '{}/{}'.format(network_url, quote_plus(channel_name))

	logs = []
	for log_file in get_files(os.path.join(app.config['ZNC_LOG_DIR'], network_name, channel_name)):
		log_url = '{}/{}'.format(channel_url, log_file)
		log = Log(log_file, log_url)

		logs += [log]

	channel = Channel(channel_name, channel_url, logs)
	network = Network(network_name, network_url, [channel])
	return(render_template('show_channel.html', network=network, channel=channel, url=app.config['URL']))

@app.route('/<network_name>/<channel_name>/<log_file>')
def get_log(network_name, channel_name, log_file):
	authenticated()
	with open(os.path.join(app.config['ZNC_LOG_DIR'], network_name, channel_name, log_file), 'rb') as file:
		log = file.read().decode('utf-8', 'ignore')
	return(render_template('log.html', log=log))

@app.route('/static/<path:filename>')
def send_static(filename):
	return send_from_directory('static', filename)

@app.route('/login')
@requires_authentication
def login(principal):
	if principal == 'kyrias@KYRIASIS.COM':
		session['logged_in'] = True

	else:
		session.pop('logged_in', None)
		return abort(401)

	if session['logged_in']:
		return '''
		Logged in! <a href="/">index</a>
		'''

def authenticated():
	if not session.get('logged_in'):
		abort(401)
	else:
		return True

if __name__ == '__main__':
	app.run(port=app.config['PORT'])