#// Copyright (c) 2000-2017 Ericsson Telecom AB                                                         //
#// All rights reserved. This program and the accompanying materials are made available under the terms //
#// of the Eclipse Public License v2.0 which accompanies this distribution, and is available at         //
#// https://www.eclipse.org/org/documents/epl-2.0/EPL-2.0.html                                                           //
#/////////////////////////////////////////////////////////////////////////////////////////////////////////
import json, os

class FileUserHandler:
    '''
    This class stores usergroup information in a json file and provides an interface for editing the usergroups.
    '''

    def __init__(self, directory):
        self._directory = directory
        with open(os.path.join(self._directory, 'userGroups.json'), 'r') as f:
            self._userGroups = json.load(f)

    def _saveUserGroups(self):
        with open(os.path.join(self._directory, 'userGroups.json'), 'w') as f:
            json.dump(self._userGroups, f, indent = 4)

    def getUserGroups(self, username):
        return set(self._userGroups['users'].get(username, []))

    def addGroup(self, group, save = True):
        if group not in self._userGroups['groups']:
            self._userGroups['groups'].append(group)
            if save:
                self._saveUserGroups()

    def removeGroup(self, group, save = True):
        if group in self._userGroups['groups']:
            self._userGroups['groups'].remove(group)
            for user in self._userGroups['users'].keys():
                self.removeUserFromGroup(user, group, False)
            if save:
                self._saveUserGroups()

    def addUserToGroup(self, username, group, save = True):
        if group in self._userGroups['groups']:
            if username not in self._userGroups['users']:
                self._userGroups['users'][username] = []
            if group not in self._userGroups['users'][username]:
                self._userGroups['users'][username].append(group)
                if save:
                    self._saveUserGroups()

    def removeUserFromGroup(self, username, group, save = True):
        if username in self._userGroups['users'] and group in self._userGroups['users'][username]:
            self._userGroups['users'][username].remove(group)
            if len(self._userGroups['users'][username]) == 0:
                self._userGroups['users'].pop(username, [])
            if save:
                self._saveUserGroups()

    def listGroups(self):
        return self._userGroups['groups']

    def listUsersInGroup(self, group):
        users = set()
        for user in self._userGroups['users']:
            if group in self._userGroups['users'][user]:
                users.add(user)
        return users