#!/usr/bin/python3
import subprocess
import Pyro4
from pathlib import Path
import logging as log
import signal
import time
from zfssmd_worker import ZFSSM_Daemon_Refresh

ENCODING = 'utf-8'
UNIX_SOCK=Path('/run/zfssmd.sock')
UNIX_PID=Path('/run/zfssmd.pid')
CONN_ID='058b7dde9ec53de9235cfc57a07ce17a9eabfce3'

class ZfsSnapshotManagerDaemon(object):
    # TODO docstrings class and methods
    snapshots = list()
    last_refreshed=0
    def __init__(self):
        self.last_refreshed = self.call_zfs_list_snapshots()

    @Pyro4.expose
    def call_zfs_list_snapshots(self):
        command = ('zfs list -Hp -t snapshot -o name,creation,used,referenced -s name')
        ## TODO try except this shit
        ## TODO log exec time of this command
        p = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
        snapshot_list = [snapshot.decode(ENCODING) for snapshot in p.stdout]
        retval = p.wait() ## TODO Implement retval actions
        last_refreshed = time.time()
        self.snapshots = snapshot_list
        self.last_refreshed = last_refreshed
        return last_refreshed

    @Pyro4.expose
    def get_current_list(self):
        return {self.last_refreshed:self.snapshots}

# TODO socket start and delete handling + SIGINT handling

def check_start_conditions():
    ## TODO test if already running PID
    ## FIXME For now we're just forcing start with deleting sock file first
    if (UNIX_SOCK.is_socket() or UNIX_SOCK.is_file()):
        try:
            UNIX_SOCK.unlink()
        except Exception as e:
            log.error("Couldn't delete sock file" + str(UNIX_SOCK))
            log.error(e)
            raise SystemExit(1)
    else:
        log.debug("Socket not set. Continuing.")

def sigterm_handler(_signo, _stack_frame):
    log.warning("Received Termination signal. Cleaning up.")
    UNIX_SOCK.unlink()
    raise SystemExit(0)

def start_daemon():
    signal.signal(signal.SIGTERM, sigterm_handler)
    signal.signal(signal.SIGINT, sigterm_handler)
    check_start_conditions()
    daemon = Pyro4.Daemon(port=None,unixsocket=str(UNIX_SOCK))
    uri = daemon.register(ZfsSnapshotManagerDaemon(), CONN_ID)
    log.info(uri)
    ZFSSM_Daemon_Refresh()
    daemon.requestLoop()

start_daemon()