From 25bb2ca451b12e3741534a8d3bbcbda3b674a7b8 Mon Sep 17 00:00:00 2001 From: MeexReay Date: Tue, 17 Jun 2025 05:22:28 +0300 Subject: [PATCH] move load_ functions to ctx module --- src/ctx.rs | 33 +++++++++++++++++++++++++++++++-- src/main.rs | 36 ++---------------------------------- 2 files changed, 33 insertions(+), 36 deletions(-) diff --git a/src/ctx.rs b/src/ctx.rs index 0fed2d3..6f460a9 100644 --- a/src/ctx.rs +++ b/src/ctx.rs @@ -1,7 +1,7 @@ use std::{ collections::HashMap, error::Error, - fs::OpenOptions, + fs::{self, OpenOptions}, io::{Cursor, Read, Write}, net::IpAddr, sync::{ @@ -17,7 +17,36 @@ use log::info; use md5::{Digest, Md5}; use rand::{Rng, distr::Alphanumeric}; -use crate::{Args, load_accounts, load_messages}; +use crate::Args; + +fn load_accounts(accounts_file: Option) -> Vec { + if let Some(accounts_file) = accounts_file.clone() { + if fs::exists(&accounts_file).expect("error checking accounts file") { + fs::read(&accounts_file) + .expect("error reading accounts file") + .split(|o| *o == b'\n') + .filter(|o| !o.is_empty()) + .map(|o| Account::from_bytes(o.to_vec())) + .collect() + } else { + Vec::new() + } + } else { + Vec::new() + } +} + +fn load_messages(messages_file: Option) -> Vec { + if let Some(messages_file) = messages_file.clone() { + if fs::exists(&messages_file).expect("error checking messages file") { + fs::read(&messages_file).expect("error reading messages file") + } else { + Vec::new() + } + } else { + Vec::new() + } +} pub struct Context { pub args: Arc, diff --git a/src/main.rs b/src/main.rs index 798464a..a7d721f 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,46 +1,14 @@ -use std::{fs, sync::Arc}; +use std::sync::Arc; use clap::Parser; use log::info; -use crate::{ - ctx::{Account, Context}, - proto::run_listener, -}; +use crate::{ctx::Context, proto::run_listener}; pub mod ctx; pub mod logic; pub mod proto; -fn load_accounts(accounts_file: Option) -> Vec { - if let Some(accounts_file) = accounts_file.clone() { - if fs::exists(&accounts_file).expect("error checking accounts file") { - fs::read(&accounts_file) - .expect("error reading accounts file") - .split(|o| *o == b'\n') - .filter(|o| !o.is_empty()) - .map(|o| Account::from_bytes(o.to_vec())) - .collect() - } else { - Vec::new() - } - } else { - Vec::new() - } -} - -fn load_messages(messages_file: Option) -> Vec { - if let Some(messages_file) = messages_file.clone() { - if fs::exists(&messages_file).expect("error checking messages file") { - fs::read(&messages_file).expect("error reading messages file") - } else { - Vec::new() - } - } else { - Vec::new() - } -} - #[derive(Parser, Debug)] #[command(version)] pub struct Args {