const numClosed = await session.clients.close(sessions);

WriteLine($"Closing session with id '{clientSession.SessionId}'");

await session.ClientControl.CloseAsync(clientSession.SessionId);

final ClientControl clientControl = session.feature(ClientControl.class);

// close a session using the session id
clientControl.close(session.getSessionId());

// close multiple sessions using a session filter
clientControl.close("$Principal is 'client'");

static int on_session_closed(
    void *context)
{
    // session has been closed
    return HANDLER_SUCCESS;
}

void close_session(
    SESSION_T *session,
    SESSION_ID_T *session_id)
{
    DIFFUSION_CLIENT_CLOSE_WITH_SESSION_PARAMS_T params = {
        .session_id = session_id,
        .on_closed = on_session_closed,
    };

    diffusion_client_close_with_session(session, params, NULL);
}

static int on_sessions_closed(
    int number_of_matching_sessions,
    void *context)
{
    // sessions have been closed
    return HANDLER_SUCCESS;
}

void close_sessions_with_filter(
    SESSION_T *session,
    char *session_filter)
{
    DIFFUSION_CLIENT_CLOSE_WITH_FILTER_PARAMS_T params = {.filter = session_filter, .on_clients_closed = on_sessions_closed};

    diffusion_client_close_with_filter(session, params, NULL);
}

//  Copyright (C) 2021, 2022 Push Technology Ltd.
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//  http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.

import Foundation
import Diffusion

class ClientControl {

    func queue_event_handler(session: PTDiffusionSession) {
        class QueueEventHandler: PTDiffusionClientQueueEventListener {
            func diffusionQueueEventListenerRegistration(_ registration: PTDiffusionRegistration,
                                                         didReportPolicyChange policy: PTDiffusionClientQueuePolicy,
                                                         forSession sessionId: PTDiffusionSessionId) {
                print("Session '%@' changed policy to %@", sessionId, policy)
            }

            func diffusionQueueEventListenerRegistrationDidClose(_ registration: PTDiffusionRegistration) {
                print("Queue event listener has closed")
            }

            func diffusionQueueEventListenerRegistration(_ registration: PTDiffusionRegistration,
                                                         didFailWithError error: Error) {
                print("Queue event listener failed with error: %@", error.localizedDescription)
            }
        }

        let handler = QueueEventHandler()
        session.clientControl.setQueueEventHandler(handler) { (registration, error) in
            if (error != nil) {
                print("An error has occurred while setting the queue event handler: %@", error!.localizedDescription)
            }
            else {
                print("Queue event handler has been successfully registered")
            }
        }
    }


    func close_client(session: PTDiffusionSession, sessionId: PTDiffusionSessionId) {
        session.clientControl.closeClient(with: sessionId) { (error) in
            if (error != nil) {
                print("An error has occurred while closing session '%@': %@", sessionId, error!.localizedDescription)
            }
            else {
                print("Session '%@' has been successfully closed", sessionId)
            }
        }
    }
}

session.clients.changeRoles(sessions, ['ROLES_TO_ADD'], ['ROLES_TO_REMOVE']);

WriteLine("Removing the Administrator role for the session.");

await session.ClientControl.ChangeRolesAsync(adminSession.SessionId, new Collection<string>() { "ADMINISTRATOR" }, new Collection<string>());

final ClientControl clientControl = session.feature(ClientControl.class);

final Set<String> rolesToRemove =  Collections.singleton("CLIENT");
final Set<String> rolesToAdd = Collections.singleton("OPERATOR");

//change roles with a session id
clientControl.changeRoles(
    session.getSessionId(),
    rolesToRemove,
    rolesToAdd
);

// change roles with a session filter
clientControl.changeRoles(
    "$Principal is 'client'",
    rolesToRemove,
    rolesToAdd
);

static int on_roles_changed(
    void *context)
{
    // roles have been successfully changed
    return HANDLER_SUCCESS;
}

void change_roles(
    SESSION_T *session,
    SESSION_ID_T *session_id,
    SET_T *roles_to_add,
    SET_T *roles_to_remove)
{
    DIFFUSION_CHANGE_ROLES_WITH_SESSION_ID_PARAMS_T params =
        {.session_id = session_id, .roles_to_add = roles_to_add, .roles_to_remove = roles_to_remove, .on_roles_changed = on_roles_changed};
    diffusion_change_roles_with_session_id(session, params, NULL);
}

static int on_roles_changed_filter(
    int number_of_matching_sessions,
    void *context)
{
    // roles have been successfully changed
    return HANDLER_SUCCESS;
}

void change_roles_with_filter(
    SESSION_T *session,
    char *session_filter,
    SET_T *roles_to_add,
    SET_T *roles_to_remove)
{
    DIFFUSION_CHANGE_ROLES_WITH_FILTER_PARAMS_T params = {
        .filter = session_filter,
        .roles_to_add = roles_to_add,
        .roles_to_remove = roles_to_remove,
        .on_roles_changed = on_roles_changed_filter};
    diffusion_change_roles_with_filter(session, params, NULL);
}