#!/usr/bin/env python
# -*- coding: utf-8 -*-

# See COPYING file for copyrights details.

from __future__ import absolute_import
import os
import time
import json

COL_ID,COL_URI,COL_DESC,COL_LAST = range(4)

def _pskpath(project_path):
    return os.path.join(project_path, 'psk')

def _mgtpath(project_path):
    return os.path.join(_pskpath(project_path), 'management.json')

def _default():
    return ['', # default description
            None, # last known URI
            None]  # last connection date

def _LoadData(project_path):
    if os.path.isdir(_pskpath(project_path)):
        _path = _mgtpath(project_path)
        # load known keys metadata
        # {ID:(Desc, LastKnownURI, LastConnect)}
        return json.loads(open(_path).read()) \
               if os.path.exists(_path) else {}
    return {}

def GetData(project_path):
    # [(ID, Desc, LastKnownURI, LastConnect)
    data = []
    loaded_data = _LoadData(project_path)
    # go through all secret files available an build data
    # out of data recoverd from json and list of secret.
    # this implicitly filters IDs out of metadata who's
    # secret is missing
    psk_files = os.listdir(_pskpath(project_path))
    for filename in psk_files:
       if filename.endswith('.secret'):
           ID = filename[:-7]  # strip filename extension
           meta = loaded_data.get(ID,_default())                  
           data.append([ID]+meta)
    return data


def DeleteID(project_path, ID):
    secret_path = os.path.join(_pskpath(project_path), ID+'.secret')
    os.remove(secret_path)

def _StoreData(project_path, data):
    pskpath = _pskpath(project_path)
    if not os.path.isdir(pskpath):
        os.mkdir(pskpath)
    with open(_mgtpath(project_path), 'w') as f:
        f.write(json.dumps(data))

def SaveData(project_path, data):
    to_store = {row[0]:row[1:] for row in data}
    _StoreData(project_path, to_store)

def UpdateID(project_path, ID, secret, URI):
    pskpath = _pskpath(project_path)
    if not os.path.exists(pskpath):
        os.mkdir(pskpath)

    secpath = os.path.join(pskpath, ID+'.secret')
    with open(secpath, 'w') as f:
        f.write(ID+":"+secret)

    data = _LoadData(project_path)
    dataForID = [ID] + (data.get(ID, _default()) if data else _default())
    dataForID[COL_URI] = URI
    # FIXME : could store time instead os a string and use DVC model's cmp 
    # then date display could be smarter, etc - sortable sting hack for now
    dataForID[COL_LAST] = time.strftime('%y/%M/%d-%H:%M:%S')
    data[ID] = dataForID[1:]
    _StoreData(project_path, data)