sRAC/src/logic.rs

use std::{
    error::Error,
    net::SocketAddr,
    sync::{Arc, atomic::Ordering},
};

use bRAC::proto::{connect, read_messages, register_user, send_message, send_message_auth};
use chrono::Local;
use log::info;

use crate::ctx::{Account, Context, add_message};

pub fn on_total_size(ctx: Arc<Context>, _: SocketAddr) -> Result<u64, Box<dyn Error>> {
    if let Some(url) = ctx.args.proxy_to.as_ref() {
        return read_messages(
            &mut connect(url, ctx.args.use_proxy.clone())?,
            1024, // TODO: softcode this
            0,
            false,
        )?
        .map(|o| o.1 as u64)
        .ok_or("err on reading in proxy mode".into()); // TODO: fix reading two times
    }

    let messages_len = ctx.messages.read().unwrap().len() as u64;
    let offset = ctx.messages_offset.load(Ordering::SeqCst);

    if let Some(splash) = &ctx.args.splash {
        Ok(messages_len + splash.len() as u64 + offset)
    } else {
        Ok(messages_len + offset)
    }
}

pub fn on_total_data(
    ctx: Arc<Context>,
    _: SocketAddr,
    _sent_size: Option<u64>,
) -> Result<Vec<u8>, Box<dyn Error>> {
    if let Some(url) = ctx.args.proxy_to.as_ref() {
        return read_messages(
            &mut connect(url, ctx.args.use_proxy.clone())?,
            1024, // TODO: softcode this
            0,
            false,
        )?
        .map(|o| (o.0.join("\n") + "\n").as_bytes().to_vec())
        .ok_or("err on reading in proxy mode".into()); // TODO: fix reading two times
    }

    let mut messages = ctx.messages.read().unwrap().clone();
    let offset = ctx.messages_offset.load(Ordering::SeqCst);

    let mut messages = if offset > 0 {
        let mut buf = vec![0; offset as usize];
        buf.append(&mut messages);
        buf
    } else {
        messages
    };

    if let Some(splash) = &ctx.args.splash {
        messages.append(&mut splash.clone().as_bytes().to_vec());
    }

    Ok(messages)
}

pub fn on_chunked_data(
    ctx: Arc<Context>,
    _: SocketAddr,
    _sent_size: Option<u64>,
    client_has: u64,
) -> Result<Vec<u8>, Box<dyn Error>> {
    if let Some(url) = ctx.args.proxy_to.as_ref() {
        return read_messages(
            &mut connect(url, ctx.args.use_proxy.clone())?,
            1024, // TODO: softcode this
            client_has as usize,
            true,
        )?
        .map(|o| (o.0.join("\n") + "\n").as_bytes().to_vec())
        .ok_or("err on reading in proxy mode".into());
    }

    let messages = ctx.messages.read().unwrap().clone();
    let offset = ctx.messages_offset.load(Ordering::SeqCst);
    let client_has = if let Some(splash) = &ctx.args.splash {
        client_has - splash.len() as u64
    } else {
        client_has
    };

    if client_has <= offset {
        Ok(messages)
    } else {
        let client_has = (client_has - offset) as usize;
        Ok(messages[client_has..].to_vec())
    }
}

pub fn on_send_message(
    ctx: Arc<Context>,
    addr: SocketAddr,
    message: Vec<u8>,
) -> Result<(), Box<dyn Error>> {
    if let Some(url) = ctx.args.proxy_to.as_ref() {
        return send_message(
            &mut connect(url, ctx.args.use_proxy.clone())?,
            &String::from_utf8_lossy(&message),
        ); // TODO: make brac accept message in bytes
    }

    if !ctx.args.auth_only {
        let mut message = message;
        message.truncate(ctx.args.message_limit);

        add_message(&message, ctx, Some(addr.ip()))?;
    }
    Ok(())
}

pub fn on_send_auth_message(
    ctx: Arc<Context>,
    _: SocketAddr,
    name: &str,
    password: &str,
    text: &str,
) -> Result<Option<u8>, Box<dyn Error>> {
    if let Some(url) = ctx.args.proxy_to.as_ref() {
        return match send_message_auth(
            &mut connect(url, ctx.args.use_proxy.clone())?,
            name,
            password,
            text,
        ) {
            Ok(0) => Ok(None),
            Ok(n) => Ok(Some(n)),
            Err(err) => Err(err),
        };
    }

    if let Some(acc) = ctx.get_account(name) {
        if acc.check_password(password) {
            let mut name = name.to_string();
            name.truncate(256); // TODO: softcode this

            let mut password = password.to_string();
            password.truncate(256); // TODO: softcode this

            let mut text = text.to_string();
            text.truncate(ctx.args.message_limit);

            add_message(text.as_bytes(), ctx, None)?;

            Ok(None)
        } else {
            Ok(Some(0x02))
        }
    } else {
        Ok(Some(0x01))
    }
}

pub fn on_register_user(
    ctx: Arc<Context>,
    addr: SocketAddr,
    name: &str,
    password: &str,
) -> Result<Option<u8>, Box<dyn Error>> {
    if let Some(url) = ctx.args.proxy_to.as_ref() {
        return Ok(
            match register_user(
                &mut connect(url, ctx.args.use_proxy.clone())?,
                name,
                password,
            ) {
                Ok(true) => None,
                _ => Some(0x01),
            },
        );
    }

    let addr = addr.ip().to_string();

    let now: i64 = Local::now().timestamp_millis();

    if ctx.get_account(name).is_some()
        || (if let Some(acc) = ctx.get_account_by_addr(&addr) {
            ((now - acc.date()) as usize) < 1000 * ctx.args.register_timeout
        } else {
            false
        })
    {
        return Ok(Some(0x01));
    }

    let account = Account::new(name.to_string(), password.to_string(), addr, now);

    info!("user registered: {name}");

    ctx.push_account(account)?;

    Ok(None)
}

pub fn on_server_info(_: Arc<Context>, _: SocketAddr) -> Result<(u8, String), Box<dyn Error>> {
    Ok((0x03, format!("sRAC {}", env!("CARGO_PKG_VERSION"))))
}