/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */

#include "config.h"

#include "argv.h"
#include "auth.h"
#include "vnc.h"

#include <guacamole/argv.h>
#include <guacamole/client.h>
#include <guacamole/protocol.h>
#include <guacamole/socket.h>
#include <rfb/rfbclient.h>
#include <rfb/rfbproto.h>

#include <pthread.h>
#include <string.h>

char* guac_vnc_get_password(rfbClient* client) {
    guac_client* gc = rfbClientGetClientData(client, GUAC_VNC_CLIENT_KEY);
    guac_vnc_client* vnc_client = ((guac_vnc_client*) gc->data);
    guac_vnc_settings* settings = vnc_client->settings;
    
    /* If the client does not support the "required" instruction, just return
        the configuration data. */
    if (!guac_client_owner_supports_required(gc))
        return ((guac_vnc_client*) gc->data)->settings->password;
    
    /* If password isn't around, prompt for it. */
    if (settings->password == NULL || strcmp(settings->password, "") == 0) {
        
        guac_argv_register(GUAC_VNC_ARGV_PASSWORD, guac_vnc_argv_callback, NULL, 0);
        
        const char* params[] = {GUAC_VNC_ARGV_PASSWORD, NULL};
        
        /* Send the request for password to the owner. */
        guac_client_owner_send_required(gc, params);
                
        /* Wait for the arguments to be returned */
        guac_argv_await(params);

    }
    
    return settings->password;
    
}

rfbCredential* guac_vnc_get_credentials(rfbClient* client, int credentialType) {
    
    guac_client* gc = rfbClientGetClientData(client, GUAC_VNC_CLIENT_KEY);
    guac_vnc_client* vnc_client = ((guac_vnc_client*) gc->data);
    guac_vnc_settings* settings = vnc_client->settings;
    
    /* Handle request for Username/Password credentials */
    if (credentialType == rfbCredentialTypeUser) {
        rfbCredential *creds = malloc(sizeof(rfbCredential));
        
        /* If the client supports the "required" instruction, prompt for and
           update those. */
        if (guac_client_owner_supports_required(gc)) {
            char* params[3] = {NULL};
            int i = 0;

            /* Check if username is null or empty. */
            if (settings->username == NULL) {
                guac_argv_register(GUAC_VNC_ARGV_USERNAME, guac_vnc_argv_callback, NULL, 0);
                params[i] = GUAC_VNC_ARGV_USERNAME;
                i++;
            }

            /* Check if password is null or empty. */
            if (settings->password == NULL) {
                guac_argv_register(GUAC_VNC_ARGV_PASSWORD, guac_vnc_argv_callback, NULL, 0);
                params[i] = GUAC_VNC_ARGV_PASSWORD;
                i++;
            }

            params[i] = NULL;

            /* If we have empty parameters, request them. */
            if (i > 0) {

                /* Send required parameters to owner. */
                guac_client_owner_send_required(gc, (const char**) params);

                /* Wait for the parameters to be returned. */
                guac_argv_await((const char**) params);

                return creds;

            }
        }
        
        /* Copy the values and return the credential set. */
        creds->userCredential.username = strdup(settings->username);
        creds->userCredential.password = strdup(settings->password);
        return creds;
        
    }

    guac_client_abort(gc, GUAC_PROTOCOL_STATUS_SERVER_ERROR,
            "Unsupported credential type requested.");
    guac_client_log(gc, GUAC_LOG_DEBUG,
            "Unable to provide requested type of credential: %d.",
            credentialType);
    return NULL;
    
}