#!/usr/bin/python
import os
import subprocess
import tkinter as tk
from tkinter import *
from datetime import datetime

def run_command(command):
	ret_code, output = subprocess.getstatusoutput(command)
	return output.splitlines()

def check_usb():
	stream = os.popen('lsusb | grep "Audio" | cut -d" " -f7')
	output = stream.read()
	if (output.find("PreSonus") != -1):
		return True
	else:
		return False

def check_record_status():
	stream = os.popen('screen -list | grep -c "Pi-ARecord" || true')
	output = stream.read()
	if (output.find("1") != -1):
		return True
	else:
		return False

def record():
	if check_record_status():
		os.system('screen -S record -X stuff "^C"')
		btn_record_var.set('Start Recording')
	else:
		if (check_usb()):
			record_time = datetime.now().strftime("%m-%d-%Y - %H:%M")
			file_name = '/mnt/USB/' + record_time + ".wav"
			command = 'screen -S record -d -m arecord -D plughw:2,0 -f dat -vv /mnt/USB/' + record_time + '.wav'
			os.system(command)
			btn_record_var.set('Stop Recording')
		else:
			btn_record_var.set('PreSonus N/A')

### tkinter variables
run_tk = True
light_mode = True
is_mounted = False
mount_directory = '/mnt/USB/'
record_start_time = None
record_current_time = None

### tkinter theme
btn_light_color = '#BA09C6'
btn_dark_color = '#3C0340'
bg_light_color = '#F171DA'
bg_dark_color = '#8a3f7c'
text_light_color = '#000000'
text_dark_color = '#adadad'

### tkinter functions
def on_exit():
	global run_tk
	if check_record_status():
		record()
	run_tk = False
	main_window.destroy()
	exit()

def restart_ar():
	global run_tk
	if check_record_status():
		record()
	run_tk = False
	main_window.destroy()
	exit()

def btn_record_func():
	record()

def btn_mount_func():
	drives = run_command("lsblk | grep 'sda1'")
	if is_mounted:
		run_command("sudo umount /dev/sda1")
	else:
		command = "sudo mount /dev/sda1 %s" % (mount_directory)
		run_command(command)

def calculate_timer():
	global record_start_time
	global record_current_time
	if check_record_status():
		record_current_time = datetime.now()
		record_elapsed = record_current_time - record_start_time
		label_record_timer_var.set("%02d:%02d:%02d" % (record_elapsed.seconds // 3600, record_elapsed.seconds // 60 % 60, record_elapsed.seconds % 60))

def check_mount_status():
	global is_mounted
	try:
		mounted_drives = run_command("lsblk | grep 'USB'")
		if mounted_drives[0].find("USB") != -1:
			is_mounted = True
			btn_mount_var.set("Unmount USB")
	except:
		is_mounted = False
		btn_mount_var.set("Mount USB")

def check_remaining_space():
	if is_mounted:
		try:
			drive = run_command("df -h %s | grep 'USB'" % (mount_directory))[0].split()
			space = ("Used: %s, Remaining: %s, Percentage: %s" % (drive[2], drive[3], drive[4]))
			label_mount_space_var.set(space)
		except:
			pass
	else:
		label_mount_space_var.set("Used: ???, Remaining: ???, Percentage: ???")

def toggle_gui_func():
	global light_mode
	if light_mode:
		main_window.configure(background=bg_dark_color)
		btn_record.configure(bg=btn_dark_color, foreground=text_dark_color)
		btn_mount.configure(bg=btn_dark_color, foreground=text_dark_color)
		#shutdown_btn.configure(bg=btn_dark_color, foreground=text_dark_color)
		btn_exit.configure(bg=btn_dark_color, foreground=text_dark_color)
		btn_restart.configure(bg=btn_dark_color, foreground=text_dark_color)
		btn_lightmode.configure(bg=btn_dark_color, foreground=text_dark_color)
		light_mode = False
	else:
		main_window.configure(background=bg_light_color)
		btn_record.configure(bg=btn_light_color, foreground=text_light_color)
		btn_mount.configure(bg=btn_light_color, foreground=text_light_color)
		#btn_shutdown.configure(bg=btn_light_color, foreground=text_light_color)
		btn_exit.configure(bg=btn_light_color, foreground=text_light_color)
		btn_restart.configure(bg=btn_light_color, foreground=text_light_color)
		btn_lightmode.configure(bg=btn_light_color, foreground=text_light_color)
		light_mode = True


### tkinter main window setup
main_window = Tk()
main_window.attributes("-fullscreen", False)
main_window.protocol("WM_DELETE_WINDOW", on_exit)
main_window.configure(background=bg_light_color)
main_window.title("Momo\'s Awesome Recorder")
main_window.columnconfigure(0, weight=1)
main_window.columnconfigure(1, weight=1)
main_window.columnconfigure(2, weight=1)
### tkinter record button
btn_record_var = tk.StringVar()
btn_record_var.set('Start Recording')
btn_record = Button(main_window, textvariable=btn_record_var, bg=btn_light_color, command=btn_record_func)
btn_record.grid(column=0, row=0, sticky='NSEW', padx=5, pady=5)
### tkinter record timer
label_record_timer_var = tk.StringVar()
label_record_timer_var.set("00:00:00")
label_record_timer = Label(main_window, textvariable=label_record_timer_var, font=('arial', 18, 'bold'))
label_record_timer.grid(column=0, row=2, sticky='NSEW', padx=5, pady=5)
### tkinter mount button
btn_mount_var = tk.StringVar()
btn_mount_var.set('Mount USB')
btn_mount = Button(main_window, textvariable=btn_mount_var, bg=btn_light_color, command=btn_mount_func)
btn_mount.grid(column=1, row=0, sticky='NSEW', padx=5, pady=5)
### tkinter usb remaining space
label_mount_space_var = tk.StringVar()
label_mount_space_var.set("Used: ???, Remaining: ???, Percentage: ???")
label_mount_space = Label(main_window, textvariable=label_mount_space_var, font=('arial', 18, 'bold'))
label_mount_space.grid(column=1, row=1, sticky='NSEW', padx=5, pady=5)
### tkinter exit program
btn_exit = Button(main_window, text='Exit Recorder', bg=btn_light_color, command=on_exit)
btn_exit.grid(column=2, row=1, sticky='NSEW', padx=5, pady=5)
### tkinter restart program
btn_restart = Button(main_window, text='Restart Recorder', bg=btn_light_color, command=restart_ar)
btn_restart.grid(column=2, row=2, sticky='NSEW', padx=5, pady=5)
### tkinter lightmode button
btn_lightmode_var = tk.StringVar()
btn_lightmode_var.set("Go Dark")
btn_lightmode = Button(main_window, textvariable=btn_lightmode_var, bg=btn_light_color, command=toggle_gui_func)
btn_lightmode.grid(column=1, row=2, sticky='NSEW', padx=5, pady=5)

while run_tk:
	main_window.update_idletasks()
	main_window.update()
	calculate_timer()
	check_mount_status()
	check_remaining_space()