#!/usr/bin/env python

# -*- coding: utf-8 -*-

# See COPYING file for copyrights details.

from __future__ import absolute_import

import os

import wx

import wx.dataview as dv

import json

def _GetInitialData(psk_path):

# [(ID, Desc, LastKnownURI, LastConnect)

data = []

data_path = os.path.join(psk_path, 'management.json')

if os.path.isdir(psk_path):

# load known keys metadata

# {ID:(Desc, LastKnownURI, LastConnect)}

recovered_data = json.loads(open(data_path).read()) \

if os.path.exists(data_path) else {}

# 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(psk_path)

for filename in psk_files:

if filename.endswith('.secret'):

ID = filename[:-7] # strip filename extension

meta = recovered_data.get(ID,

['', # default description

None, # last known URI

None]) # last connection date

data.append([ID]+meta)

return data

def _DeleteID(psk_path, ID):

secret_path = os.path.join(psk_path, ID+'.secret')

os.remove(secret_path)

def _SaveData(psk_path, data):

if not os.path.isdir(psk_path):

os.mkdir(psk_path)

data_path = os.path.join(psk_path, 'management.json')

to_store = {row[0]:row[1:] for row in data}

with open(data_path, 'w') as f:

f.write(json.dumps(to_store))

class IDManagerModel(dv.PyDataViewIndexListModel):

def __init__(self, psk_path, columncount):

self.psk_path = psk_path

self.columncount = columncount

self.data = _GetInitialData(psk_path)

dv.PyDataViewIndexListModel.__init__(self, len(self.data))

def _saveData(self):

_SaveData(self.psk_path, self.data)

def GetColumnType(self, col):

return "string"

def GetValueByRow(self, row, col):

return self.data[row][col]

def SetValueByRow(self, value, row, col):

self.data[row][col] = value

self._saveData()

def GetColumnCount(self):

return len(self.data[0]) if self.data else self.columncount

def GetCount(self):

return len(self.data)

def GetAttrByRow(self, row, col, attr):

if col == 3:

attr.SetColour('blue')

attr.SetBold(True)

return True

return False

def Compare(self, item1, item2, col, ascending):

if not ascending: # swap sort order?

item2, item1 = item1, item2

row1 = self.GetRow(item1)

row2 = self.GetRow(item2)

if col == 0:

return cmp(int(self.data[row1][col]), int(self.data[row2][col]))

else:

return cmp(self.data[row1][col], self.data[row2][col])

def DeleteRows(self, rows):

rows = list(rows)

rows.sort(reverse=True)

for row in rows:

del self.data[row]

_DeleteID(self.psk_path, ID)

self.RowDeleted(row)

self._saveData()

def AddRow(self, value):

self.data.append(value)

self.RowAppended()

self._saveData()

colflags = dv.DATAVIEW_COL_RESIZABLE|dv.DATAVIEW_COL_SORTABLE

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

class IDManager(wx.Panel):

def __init__(self, parent, ctr, SelectURICallBack=None, SelectIDCallBack=None, **kwargs):

wx.Panel.__init__(self, parent, -1, size=(400,200))

self.isManager = SelectURICallBack is None and SelectIDCallBack is None

self.SelectURICallBack = SelectURICallBack

self.SelectIDCallBack = SelectIDCallBack

dvStyle = wx.BORDER_THEME | dv.DV_ROW_LINES

if self.isManager :

# no multiple selection in selector mode

dvStyle |= dv.DV_MULTIPLE

self.dvc = dv.DataViewCtrl(self, style = dvStyle)

args = lambda *a,**k:(a,k)

ColumnsDesc = [

args(_("ID"), COL_ID, width = 100),

args(_("Last URI"), COL_URI, width = 80),

args(_("Description"), COL_DESC, width = 200,

mode = dv.DATAVIEW_CELL_EDITABLE

if self.isManager

else dv.DATAVIEW_CELL_INERT),

args(_("Last connection"), COL_LAST, width = 100),

]

self.model = IDManagerModel(

os.path.join(str(ctr.ProjectPath), 'psk'),

len(ColumnsDesc))

self.dvc.AssociateModel(self.model)

for a,k in ColumnsDesc:

self.dvc.AppendTextColumn(*a,**dict(k, flags = colflags))

# TODO : when select,

# - update ID field of scheme editor

# - enable use URI button

self.Sizer = wx.BoxSizer(wx.VERTICAL)

self.Sizer.Add(self.dvc, 1, wx.EXPAND)

btnbox = wx.BoxSizer(wx.HORIZONTAL)

if self.isManager :

# deletion of secret and metadata

deleteButton = wx.Button(self, label=_("Delete ID"))

self.Bind(wx.EVT_BUTTON, self.OnDeleteButton, deleteButton)

btnbox.Add(deleteButton, 0, wx.LEFT|wx.RIGHT, 5)

# export all

exportButton = wx.Button(self, label=_("Export all"))

self.Bind(wx.EVT_BUTTON, self.OnExportButton, exportButton)

btnbox.Add(exportButton, 0, wx.LEFT|wx.RIGHT, 5)

# import with a merge -> duplicates are asked for

importButton = wx.Button(self, label=_("Import"))

self.Bind(wx.EVT_BUTTON, self.OnImportButton, importButton)

btnbox.Add(importButton, 0, wx.LEFT|wx.RIGHT, 5)

else :

# selector mode

# use last known URI button

# TODO : disable use URI button until something selected

self.useURIButton = wx.Button(self, label=_("Use last URI"))

self.Bind(wx.EVT_BUTTON, self.OnUseURIButton, self.useURIButton)

self.useURIButton.Disable()

btnbox.Add(self.useURIButton, 0, wx.LEFT|wx.RIGHT, 5)

self.Sizer.Add(btnbox, 0, wx.TOP|wx.BOTTOM, 5)

self.Bind(dv.EVT_DATAVIEW_SELECTION_CHANGED, self.OnSelectionChanged, self.dvc)

def OnDeleteButton(self, evt):

items = self.dvc.GetSelections()

rows = [self.model.GetRow(item) for item in items]

# Ask if user really wants to delete

if wx.MessageBox(_('Are you sure to delete selected IDs?'),

_('Delete IDs'),

wx.YES_NO | wx.CENTRE | wx.NO_DEFAULT) != wx.YES:

return

self.model.DeleteRows(rows)

def OnSelectionChanged(self, evt):

if not self.isManager :

items = self.dvc.GetSelections()

somethingSelected = len(items) > 0

self.useURIButton.Enable(somethingSelected)

if somethingSelected:

row = self.model.GetRow(items[0])

ID = self.model.GetValueByRow(row, COL_ID)

self.SelectIDCallBack(ID)

def OnUseURIButton(self, evt):

row = self.model.GetRow(self.dvc.GetSelections()[0])

URI = self.model.GetValueByRow(row, COL_URI)

if URI:

self.SelectURICallBack(URI)

def OnExportButton(self, evt):

# TODO

wx.MessageBox(_('?'),

_('Mhe'),

wx.YES_NO | wx.CENTRE | wx.NO_DEFAULT)

def OnImportButton(self, evt):

# TODO

wx.MessageBox(_('?'),

_('Mhe'),

wx.YES_NO | wx.CENTRE | wx.NO_DEFAULT)