makepad-platform 1.0.0

use {
    std::{
        ptr,
        sync::mpsc::{Sender},
        sync::Arc,
    },
    crate::{
        thread::SignalToUI,
        makepad_live_id::*,
        makepad_objc_sys::{objc_block, objc_block_invoke},
        os::{
            //cocoa_app::{MacosApp, get_macos_class_global},
            apple::apple_sys::*,
            apple::apple_classes::get_apple_class_global,
            apple_util::{
                nsstring_to_string,
                str_to_nsstring,
            },
        },
        web_socket::WebSocketMessage,
        event::{
            NetworkResponseItem,
            NetworkResponse,
            HttpError,
            HttpRequest,
            HttpResponse
        },
    }
};

pub fn define_web_socket_delegate() -> *const Class {
    
    extern "C" fn did_open_with_protocol(_codethis: &Object, _: Sel, _web_socket_task: ObjcId, _open_with_protocol: ObjcId) {
        crate::log!("DID OPEN WITH PROTOCOL");
        //let sender_box: u64 = *this.get_ivar("sender_box");
        // TODO send Open and Close websocket messages
        // lets turn t
    }
    
    extern "C" fn did_close_with_code(_this: &Object, _: Sel, _web_socket_task: ObjcId, _code: usize, _reason: ObjcId) {
        crate::log!("DID CLOSE WITH CODE");
    }
    
    let superclass = class!(NSObject);
    let mut decl = ClassDecl::new("NSURLSessionWebSocketDelegate", superclass).unwrap();
    
    // Add callback methods
    unsafe {
        decl.add_method(sel!(webSocketTask: didOpenWithProtocol:), did_open_with_protocol as extern "C" fn(&Object, Sel, ObjcId, ObjcId));
        decl.add_method(sel!(webSocketTask: didCloseWithCode: reason:), did_close_with_code as extern "C" fn(&Object, Sel, ObjcId, usize, ObjcId));
    }
    decl.add_ivar::<u64>("sender_box");
        
    return decl.register();
}


struct UrlSessionDataDelegateContext{
    sender: Sender<NetworkResponseItem>,
    request_id: LiveId,
    metadata_id: LiveId
}

pub fn define_url_session_data_delegate() -> *const Class {
    extern "C" fn did_receive_response(
        _this: &Object,
        _: Sel,
        _session: ObjcId,
        _data_task: ObjcId,
        _response: ObjcId,
        completion: ObjcId,
    ) {
        unsafe {
            // Allow the session to continue receiving data
            let disposition = NSURLSessionResponseAllow;
            objc_block_invoke!(completion, invoke(
                (disposition): u64
            ));
        }
    }

    extern "C" fn did_receive_data(
        this: &Object,
        _: Sel,
        _session: ObjcId,
        _data_task: ObjcId,
        data: ObjcId,
    ) {
        unsafe {
            let context_box: u64 = *this.get_ivar("context_box");
            let context_box: Box<UrlSessionDataDelegateContext> = Box::from_raw(context_box as *mut _);
            
            // Get the data bytes
            let bytes: *const u8 = msg_send![data, bytes];
            let length: usize = msg_send![data, length];
            let data_bytes: &[u8] = std::slice::from_raw_parts(bytes, length);

            // Send the data chunk
            let message = NetworkResponseItem {
                request_id: context_box.request_id,
                response: NetworkResponse::HttpStreamResponse(HttpResponse{
                    headers: Default::default(),
                    metadata_id: context_box.metadata_id,
                    status_code: 0,
                    body:Some(data_bytes.to_vec())
                }),
            };
            
            let _ = context_box.sender.send(message);
            // keep the sender_box
            let _ = Box::into_raw(context_box);
        }
    }

    extern "C" fn did_complete_with_error(
        this: &Object,
        _: Sel,
        _session: ObjcId,
        _task: ObjcId,
        error: ObjcId,
    ) {
        unsafe {
            let context_box: u64 = *this.get_ivar("context_box");
            let context_box: Box<UrlSessionDataDelegateContext> = Box::from_raw(context_box as *mut _);
            if error != nil {
                // Handle error
                let error_str: String = nsstring_to_string(msg_send![error, localizedDescription]);
                let message = NetworkResponseItem {
                    request_id: context_box.request_id,
                    response: NetworkResponse::HttpRequestError(
                        HttpError{
                            metadata_id: context_box.metadata_id, 
                            message: error_str
                        }
                    ),
                };
                context_box.sender.send(message).unwrap();
            } else {
                // Indicate that streaming is complete
                let message = NetworkResponseItem {
                    request_id: context_box.request_id,
                    response: NetworkResponse::HttpStreamComplete(HttpResponse{
                        headers: Default::default(),
                        metadata_id: context_box.metadata_id,
                        status_code: 0,
                        body:None
                    }),
                };
                context_box.sender.send(message).unwrap();
            }
        }
    }

    let superclass = class!(NSObject);
    let mut decl = ClassDecl::new("NSURLSessionDataDelegate", superclass).unwrap();

    unsafe {
        decl.add_method(
            sel!(URLSession:dataTask:didReceiveResponse:completionHandler:),
            did_receive_response as extern "C" fn(&Object, Sel, ObjcId, ObjcId, ObjcId, ObjcId),
        );
        decl.add_method(
            sel!(URLSession:dataTask:didReceiveData:),
            did_receive_data as extern "C" fn(&Object, Sel, ObjcId, ObjcId, ObjcId),
        );
        decl.add_method(
            sel!(URLSession:task:didCompleteWithError:),
            did_complete_with_error as extern "C" fn(&Object, Sel, ObjcId, ObjcId, ObjcId),
        );
    }

    // Add ivars to store the sender and request ID
    decl.add_ivar::<u64>("context_box");

    decl.register()
}

// this allows for locally signed SSL certificates to pass
// TODO make this configurable
pub fn define_url_session_delegate() -> *const Class {
    extern "C" fn did_receive_challenge(_this: &Object, _: Sel, _session: ObjcId, challenge: ObjcId, completion: ObjcId) {
        unsafe{
            let pspace: ObjcId = msg_send![challenge, protectionSpace];
            let trust: ObjcId = msg_send![pspace, serverTrust];
            if trust == nil{
                objc_block_invoke!(completion, invoke(
                    (0): usize,
                    (nil): ObjcId
                ));
            }
            else{
                let credential: ObjcId = msg_send![class!(NSURLCredential), credentialForTrust:trust];
                objc_block_invoke!(completion, invoke(
                    (0): usize,
                    (credential): ObjcId
                ));
            }
        }
    }
        
    let superclass = class!(NSObject);
    let mut decl = ClassDecl::new("NSURLSessionDelegate", superclass).unwrap();
    // Add callback methods
    unsafe {
        decl.add_method(sel!(URLSession: didReceiveChallenge: completionHandler:), did_receive_challenge as extern "C" fn(&Object, Sel, ObjcId,ObjcId, ObjcId));
    }
    return decl.register();
}


unsafe fn make_ns_request(request: &HttpRequest) -> ObjcId {
    // Prepare the NSMutableURLRequest instance
    let url: ObjcId =
    msg_send![class!(NSURL), URLWithString: str_to_nsstring(&request.url)];
    
    let mut ns_request: ObjcId = msg_send![class!(NSMutableURLRequest), alloc];
    
    ns_request = msg_send![ns_request, initWithURL: url];
    let () = msg_send![ns_request, setHTTPMethod: str_to_nsstring(&request.method.to_string())];
    
    for (key, values) in request.headers.iter() {
        for value in values {
            let () = msg_send![ns_request, addValue: str_to_nsstring(value) forHTTPHeaderField: str_to_nsstring(key)];
        }
    }
    
    if let Some(body) = request.body.as_ref() {
        let nsdata: ObjcId = msg_send![class!(NSData), dataWithBytes: body.as_ptr() length: body.len()];
        let () = msg_send![ns_request, setHTTPBody: nsdata];
    }
    ns_request
}

pub struct OsWebSocket{
    data_task: Arc<ObjcId>,
    rx_sender: Sender<WebSocketMessage>
}

impl OsWebSocket{

    pub fn send_message(&mut self, message:WebSocketMessage)->Result<(),()>{
        unsafe{
            let rx_sender = self.rx_sender.clone();
            let handler = objc_block!(move | error: ObjcId | {
                if error != ptr::null_mut() {
                    let error_str: String = nsstring_to_string(msg_send![error, localizedDescription]);
                    rx_sender.send(WebSocketMessage::Error(error_str)).unwrap();
                }
            });
                       
            let msg: ObjcId = match &message{
                WebSocketMessage::String(data)=>{
                    let nsstring = str_to_nsstring(data);
                    let msg:ObjcId = msg_send![class!(NSURLSessionWebSocketMessage), alloc];
                    let () = msg_send![msg, initWithString: nsstring];
                    msg
                }
                WebSocketMessage::Binary(data)=>{
                    let nsdata: ObjcId = msg_send![class!(NSData), dataWithBytes: data.as_ptr() length: data.len()];
                    let msg: ObjcId = msg_send![class!(NSURLSessionWebSocketMessage), alloc];
                    let () = msg_send![msg, initWithData: nsdata];
                    msg
                }
                _=>panic!()
            };
                        
            let () = msg_send![*Arc::as_ptr(&self.data_task), sendMessage: msg completionHandler: &handler];
            Ok(())
        } 
    }
    
    pub fn close(&self){
        unsafe{
            let () = msg_send![*self.data_task, cancel];
        }
    }
            
    pub fn open(_socket_id:u64, request: HttpRequest, rx_sender:Sender<WebSocketMessage>)->OsWebSocket{
        unsafe {
            //self.lazy_init_ns_url_session();
            let ns_request = make_ns_request(&request);
            
            let session: ObjcId = if request.ignore_ssl_cert{
                let config: ObjcId = msg_send![class!(NSURLSessionConfiguration), defaultSessionConfiguration];
                let deleg: ObjcId = msg_send![get_apple_class_global().url_session_delegate, new];
                msg_send![class!(NSURLSession), sessionWithConfiguration: config delegate: deleg delegateQueue:nil]
            }     
            else{
                msg_send![class!(NSURLSession), sharedSession]
            };
            
            //let session  = self.ns_url_session.unwrap();
            let data_task: ObjcId = msg_send![session, webSocketTaskWithRequest: ns_request];
            let web_socket_delegate_instance: ObjcId = msg_send![get_apple_class_global().web_socket_delegate, new];
            
            let () = msg_send![data_task, setMaximumMessageSize:5*1024*1024];
             
            unsafe fn set_message_receive_handler(data_task2: Arc<ObjcId>,  rx_sender: Sender<WebSocketMessage>) {
                let data_task = data_task2.clone();
                let handler = objc_block!(move | message: ObjcId, error: ObjcId | {
                    if error != ptr::null_mut() {
                        let error_str: String = nsstring_to_string(msg_send![error, localizedDescription]);
                        rx_sender.send(WebSocketMessage::Error(error_str)).unwrap();
                        SignalToUI::set_ui_signal();
                        return;
                    }
                    let ty: usize = msg_send![message, type];
                    if ty == 0 { // binary
                        let data: ObjcId = msg_send![message, data];
                        let bytes: *const u8 = msg_send![data, bytes];
                        let length: usize = msg_send![data, length];
                        let data_bytes: &[u8] = std::slice::from_raw_parts(bytes, length);
                        let message = WebSocketMessage::Binary(data_bytes.to_vec());
                        rx_sender.send(message).unwrap();
                        SignalToUI::set_ui_signal();
                    }
                    else { // string
                        let string: ObjcId = msg_send![message, string];
                        let message = WebSocketMessage::String(nsstring_to_string(string));
                        rx_sender.send(message).unwrap();
                        SignalToUI::set_ui_signal();
                    }
                    set_message_receive_handler(data_task.clone(), rx_sender.clone())
                });
                let () = msg_send![*Arc::as_ptr(&data_task2), receiveMessageWithCompletionHandler: &handler];
            }
            let () = msg_send![data_task, setDelegate: web_socket_delegate_instance];
            let data_task = Arc::new(data_task);
            set_message_receive_handler(data_task.clone(), rx_sender.clone());
            let () = msg_send![*Arc::as_ptr(&data_task), resume];
            OsWebSocket{
                rx_sender,
                data_task
            }
        }
    }
        
}

struct HttpReq{
    request_id: LiveId,
    data_task: RcObjcId
}

#[derive(Default)]
pub struct AppleHttpRequests{
    requests: Vec<HttpReq>
}

impl AppleHttpRequests{
    pub fn cancel_http_request(&mut self, request_id: LiveId){
        self.requests.retain(|v|{
            if v.request_id == request_id{
                unsafe{
                    let () = msg_send![v.data_task.as_id(), cancel];
                }
                false
            }
            else{
                true
            }
        })
        
    }
    
    pub fn handle_response_item(&mut self, item:&NetworkResponseItem){
        match &item.response{
            NetworkResponse::HttpRequestError(_) |
            NetworkResponse::HttpResponse(_) |
            NetworkResponse::HttpStreamComplete(_) => {
                self.requests.retain(|v| v.request_id != item.request_id);
            }
            _=>{
            }
        }
    }
    
    pub fn make_http_request(&mut self, request_id: LiveId, request: HttpRequest, networking_sender: Sender<NetworkResponseItem>) {
        unsafe {
            let ignore_ssl_cert = request.ignore_ssl_cert;
            let is_streaming = request.is_streaming;
            let metadata_id = request.metadata_id;
            let ns_request = make_ns_request(&request);
            
            let session: ObjcId = if ignore_ssl_cert{
                let config: ObjcId = msg_send![class!(NSURLSessionConfiguration), defaultSessionConfiguration];
                let deleg: ObjcId = msg_send![get_apple_class_global().url_session_delegate, new];
                msg_send![class!(NSURLSession), sessionWithConfiguration: config delegate: deleg delegateQueue:nil]
            }
            else{
                msg_send![class!(NSURLSession), sharedSession]
            };
                    
            if is_streaming{ // its using the streaming delegate
                let context_box = Box::into_raw(Box::new(UrlSessionDataDelegateContext{
                    request_id,
                    metadata_id: request.metadata_id,
                    sender: networking_sender
                })) as u64;
                let url_session_data_delegate_instance: ObjcId = msg_send![get_apple_class_global().url_session_data_delegate, new];
                        
                (*url_session_data_delegate_instance).set_ivar("context_box", context_box);
                
                let data_task: ObjcId = msg_send![session, dataTaskWithRequest: ns_request];
                let () = msg_send![data_task, setDelegate: url_session_data_delegate_instance];
                let () = msg_send![data_task, resume];
                self.requests.push(HttpReq{
                    request_id,
                    data_task: RcObjcId::from_unowned(NonNull::new(data_task).unwrap())
                })
            }
            else{ // its using the completion handler
                let response_handler = objc_block!(move | data: ObjcId, response: ObjcId, error: ObjcId | {
                    if error != ptr::null_mut() {
                        let error_str: String = nsstring_to_string(msg_send![error, localizedDescription]);
                        let message = NetworkResponseItem {
                            request_id,
                            response: NetworkResponse::HttpRequestError(HttpError{
                                metadata_id: metadata_id, 
                                message: error_str
                            })
                        };
                        networking_sender.send(message).unwrap();
                        return;
                    }
                                            
                    let bytes: *const u8 = msg_send![data, bytes];
                    let length: usize = msg_send![data, length];
                    let data_bytes: &[u8] = std::slice::from_raw_parts(bytes, length);
                    let response_code: u16 = msg_send![response, statusCode];
                    let headers: ObjcId = msg_send![response, allHeaderFields];
                    let mut response = HttpResponse::new(
                        request.metadata_id,
                        response_code,
                        "".to_string(),
                        Some(data_bytes.to_vec()),
                    );
                                            
                    let key_enumerator: ObjcId = msg_send![headers, keyEnumerator];
                    let mut key: ObjcId = msg_send![key_enumerator, nextObject];
                    while key != ptr::null_mut() {
                        let value: ObjcId = msg_send![headers, objectForKey: key];
                        let key_str = nsstring_to_string(key);
                        let value_str = nsstring_to_string(value);
                        response.set_header(key_str, value_str);
                                                        
                        key = msg_send![key_enumerator, nextObject];
                    }
                                
                    let message = NetworkResponseItem {
                        request_id,
                        response: NetworkResponse::HttpResponse(response)
                    };
                    networking_sender.send(message).unwrap();
                });
                
                let data_task: ObjcId = msg_send![session, dataTaskWithRequest: ns_request completionHandler: &response_handler];
                let () = msg_send![data_task, resume];
                self.requests.push(HttpReq{
                    request_id,
                    data_task: RcObjcId::from_unowned(NonNull::new(data_task).unwrap())
                })
            }
        }
    }
}