<?php
function insert_token($admin_wp_id,$user_id,$token, $is_pass){
    if(r_token($admin_wp_id,$user_id) != null){
        return u_token($admin_wp_id,$user_id,$token, $is_pass);
    }else{
        return c_token($admin_wp_id,$user_id,$token);
    }
}
function r_token($admin_wp_id,$users_id){
    $rv = new stdClass();
    $status = false;
    
    global $wpdb;
    if($users_id != 0 ){
        $sql_statement = 'select * from keylab_users_token where users_id = '.$users_id;
        return $wpdb->get_row($sql_statement);
    }else{
        $sql_statement = 'select * from keylab_users_token where admin_wp_id = '.$admin_wp_id;
        return $wpdb->get_row($sql_statement);
    }
}
function has_token($token,$is_pass){
    $rv = new stdClass();
    
    global $wpdb;

        if($is_pass){
            $sql_statement = "select * from keylab_users_token where pass_token = '$token'";
        }else{
            $sql_statement = "select * from keylab_users_token where token = '$token'";
        }
    
    return $wpdb->get_row($sql_statement);
}
function c_token($admin_wp_id,$users_id,$token){
    global $wpdb;

    $rv = new stdClass();
    $status = false;
    
    $status = $wpdb->insert("keylab_users_token",array(
        'admin_wp_id' => $admin_wp_id,
        'users_id'    => $users_id,
        'token'       => $token,
    ));

    return $status;
}
function u_token($admin_wp_id,$users_id,$token,$is_pass){
    $rv = new stdClass();
    $status = false;
    
    global $wpdb;
    if($admin_wp_id == 0 ){
        if($is_pass){
            $status = $wpdb->update("keylab_users_token",array(
                'pass_token'       => $token,
                'token_creation_timestamp'  => current_time( 'mysql' )
            ), array(
                'users_id'    => $users_id
            ), array(
                '%s','%s'
            ));
        }else{
            $status = $wpdb->update("keylab_users_token",array(
                'token'       => $token,
                'token_creation_timestamp'  => current_time( 'mysql' )
            ), array(
                'users_id'    => $users_id
            ), array(
                '%s','%s'
            ));
        }
    }else{
        if($is_pass){
            $status = $wpdb->update("keylab_users_token",array(
                'pass_token'       => $token,
                'token_creation_timestamp'  => current_time( 'mysql' )
            ), array(
                'admin_wp_id'    => $admin_wp_id
            ), array(
                '%s','%s'
            ));
        }else{
            $status = $wpdb->update("keylab_users_token",array(
                'token'       => $token,
                'token_creation_timestamp'  => current_time( 'mysql' )
            ), array(
                'admin_wp_id'    => $admin_wp_id
            ), array(
                '%s','%s'
            ));
        } 
    }
    return $status;
}
?>