Added support any popular terminals Unix-like OS
This commit is contained in:
parent
e8281f68e3
commit
db0a31d195
325
src/main.rs
Normal file
325
src/main.rs
Normal file
@ -0,0 +1,325 @@
|
|||||||
|
// src/main.rs
|
||||||
|
//! Главный модуль сервера Futriix
|
||||||
|
//!
|
||||||
|
//! Точка входа в приложение, инициализирует сервер и запускает его.
|
||||||
|
//! Использует wait-free архитектуру с lock-free структурами данных.
|
||||||
|
|
||||||
|
mod common;
|
||||||
|
mod server;
|
||||||
|
mod lua_shell;
|
||||||
|
|
||||||
|
use std::env;
|
||||||
|
use std::fs::OpenOptions;
|
||||||
|
use std::io::Write;
|
||||||
|
|
||||||
|
use crate::common::FutriixError;
|
||||||
|
|
||||||
|
/// Функция для логирования в файл
|
||||||
|
fn log_to_file(message: &str) {
|
||||||
|
match OpenOptions::new()
|
||||||
|
.create(true)
|
||||||
|
.append(true)
|
||||||
|
.open("futriix.log")
|
||||||
|
{
|
||||||
|
Ok(mut file) => {
|
||||||
|
// ИСПРАВЛЕНИЕ: Используем системное время с миллисекундами
|
||||||
|
let timestamp = chrono::Local::now().format("%Y-%m-%d %H:%M:%S%.3f").to_string();
|
||||||
|
let log_message = format!("[{}] {}\n", timestamp, message);
|
||||||
|
let _ = file.write_all(log_message.as_bytes());
|
||||||
|
}
|
||||||
|
Err(e) => eprintln!("Failed to write to log file: {}", e),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Простая структура для аргументов командной строки
|
||||||
|
struct Args {
|
||||||
|
config: String,
|
||||||
|
debug: bool,
|
||||||
|
http_port: Option<u16>,
|
||||||
|
https_port: Option<u16>,
|
||||||
|
host: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Простой парсер аргументов командной строки
|
||||||
|
fn parse_args() -> Args {
|
||||||
|
let mut args = Args {
|
||||||
|
config: "config.toml".to_string(),
|
||||||
|
debug: false,
|
||||||
|
http_port: None,
|
||||||
|
https_port: None,
|
||||||
|
host: None,
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut iter = env::args().skip(1);
|
||||||
|
while let Some(arg) = iter.next() {
|
||||||
|
match arg.as_str() {
|
||||||
|
"--config" | "-c" => {
|
||||||
|
if let Some(value) = iter.next() {
|
||||||
|
args.config = value;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"--debug" | "-d" => {
|
||||||
|
args.debug = true;
|
||||||
|
}
|
||||||
|
"--http-port" => {
|
||||||
|
if let Some(value) = iter.next() {
|
||||||
|
if let Ok(port) = value.parse() {
|
||||||
|
args.http_port = Some(port);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"--https-port" => {
|
||||||
|
if let Some(value) = iter.next() {
|
||||||
|
if let Ok(port) = value.parse() {
|
||||||
|
args.https_port = Some(port);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"--host" => {
|
||||||
|
if let Some(value) = iter.next() {
|
||||||
|
args.host = Some(value);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
if arg.starts_with("--config=") {
|
||||||
|
args.config = arg.trim_start_matches("--config=").to_string();
|
||||||
|
} else if arg.starts_with("-c=") {
|
||||||
|
args.config = arg.trim_start_matches("-c=").to_string();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
args
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Функция для вывода текста с ANSI цветом
|
||||||
|
fn print_colored(text: &str, ansi_color: &str) {
|
||||||
|
println!("{}{}\x1b[0m", ansi_color, text);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Функция для вывода текста с цветом #00bfff для проекта futriix
|
||||||
|
fn print_futriix(text: &str) {
|
||||||
|
let futriix_color = "\x1b[38;2;0;191;255m"; // ANSI код для #00bfff
|
||||||
|
println!("{}{}\x1b[0m", futriix_color, text);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Конвертация HEX цвета в ANSI escape code
|
||||||
|
fn hex_to_ansi(hex_color: &str) -> String {
|
||||||
|
let hex = hex_color.trim_start_matches('#');
|
||||||
|
|
||||||
|
if hex.len() == 6 {
|
||||||
|
if let (Ok(r), Ok(g), Ok(b)) = (
|
||||||
|
u8::from_str_radix(&hex[0..2], 16),
|
||||||
|
u8::from_str_radix(&hex[2..4], 16),
|
||||||
|
u8::from_str_radix(&hex[4..6], 16),
|
||||||
|
) {
|
||||||
|
return format!("\x1b[38;2;{};{};{}m", r, g, b);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
"\x1b[38;2;255;255;255m".to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Функция для получения цвета #00bfff как строки ANSI кода для проекта futriix
|
||||||
|
fn get_futriix_color() -> String {
|
||||||
|
"\x1b[38;2;0;191;255m".to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Структура для хранения информации о дистрибутиве Linux
|
||||||
|
#[derive(Debug)]
|
||||||
|
struct LinuxDistribution {
|
||||||
|
name: String,
|
||||||
|
version: String,
|
||||||
|
pretty_name: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl LinuxDistribution {
|
||||||
|
/// Создает строку с полным именем дистрибутива и версией
|
||||||
|
fn full_name(&self) -> String {
|
||||||
|
if !self.version.is_empty() {
|
||||||
|
format!("{} {}", self.name, self.version)
|
||||||
|
} else {
|
||||||
|
self.name.clone()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Получение информации о дистрибутиве Linux
|
||||||
|
/// Читает информацию из файлов /etc/os-release или /etc/lsb-release
|
||||||
|
fn get_linux_distribution() -> LinuxDistribution {
|
||||||
|
// Инициализируем структуру значениями по умолчанию
|
||||||
|
let mut distro = LinuxDistribution {
|
||||||
|
name: "Linux".to_string(),
|
||||||
|
version: String::new(),
|
||||||
|
pretty_name: "Linux".to_string(),
|
||||||
|
};
|
||||||
|
|
||||||
|
// Попробуем прочитать /etc/os-release (современные дистрибутивы)
|
||||||
|
if let Ok(content) = std::fs::read_to_string("/etc/os-release") {
|
||||||
|
for line in content.lines() {
|
||||||
|
if line.starts_with("PRETTY_NAME=") {
|
||||||
|
// Убираем PRETTY_NAME= и кавычки
|
||||||
|
let name = line.trim_start_matches("PRETTY_NAME=");
|
||||||
|
distro.pretty_name = name.trim_matches('"').to_string();
|
||||||
|
} else if line.starts_with("NAME=") {
|
||||||
|
let name = line.trim_start_matches("NAME=");
|
||||||
|
distro.name = name.trim_matches('"').to_string();
|
||||||
|
} else if line.starts_with("VERSION=") || line.starts_with("VERSION_ID=") {
|
||||||
|
// Предпочитаем VERSION_ID для номеров версий, но используем VERSION если есть
|
||||||
|
let version_line = if line.starts_with("VERSION_ID=") {
|
||||||
|
line.trim_start_matches("VERSION_ID=")
|
||||||
|
} else {
|
||||||
|
line.trim_start_matches("VERSION=")
|
||||||
|
};
|
||||||
|
distro.version = version_line.trim_matches('"').to_string();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Если нашли pretty_name, но не нашли отдельное имя - используем pretty_name
|
||||||
|
if distro.name == "Linux" && distro.pretty_name != "Linux" {
|
||||||
|
// Попробуем извлечь имя и версию из pretty_name
|
||||||
|
let pretty = &distro.pretty_name;
|
||||||
|
if let Some(pos) = pretty.find(char::is_numeric) {
|
||||||
|
distro.name = pretty[..pos].trim().to_string();
|
||||||
|
distro.version = pretty[pos..].trim().to_string();
|
||||||
|
} else {
|
||||||
|
distro.name = pretty.clone();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Если не нашли версию в /etc/os-release, попробуем /etc/lsb-release
|
||||||
|
if distro.version.is_empty() {
|
||||||
|
if let Ok(content) = std::fs::read_to_string("/etc/lsb-release") {
|
||||||
|
for line in content.lines() {
|
||||||
|
if line.starts_with("DISTRIB_DESCRIPTION=") {
|
||||||
|
let name = line.trim_start_matches("DISTRIB_DESCRIPTION=");
|
||||||
|
distro.pretty_name = name.trim_matches('"').to_string();
|
||||||
|
|
||||||
|
// Попробуем извлечь версию из описания
|
||||||
|
let desc = &distro.pretty_name;
|
||||||
|
if let Some(pos) = desc.find(char::is_numeric) {
|
||||||
|
distro.name = desc[..pos].trim().to_string();
|
||||||
|
distro.version = desc[pos..]
|
||||||
|
.chars()
|
||||||
|
.take_while(|c| c.is_numeric() || *c == '.')
|
||||||
|
.collect::<String>();
|
||||||
|
}
|
||||||
|
} else if line.starts_with("DISTRIB_ID=") && distro.name == "Linux" {
|
||||||
|
let name = line.trim_start_matches("DISTRIB_ID=");
|
||||||
|
distro.name = name.trim_matches('"').to_string();
|
||||||
|
} else if line.starts_with("DISTRIB_RELEASE=") {
|
||||||
|
let version = line.trim_start_matches("DISTRIB_RELEASE=");
|
||||||
|
distro.version = version.trim_matches('"').to_string();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Если все еще не нашли версию, попробуем прочитать /etc/issue
|
||||||
|
if distro.version.is_empty() {
|
||||||
|
if let Ok(content) = std::fs::read_to_string("/etc/issue") {
|
||||||
|
// Берем первую строку
|
||||||
|
if let Some(first_line) = content.lines().next() {
|
||||||
|
let line = first_line.trim();
|
||||||
|
|
||||||
|
// Ищем номер версии в строке
|
||||||
|
if let Some(pos) = line.find(char::is_numeric) {
|
||||||
|
distro.name = line[..pos].trim().to_string();
|
||||||
|
// Извлекаем версию (первые цифры до пробела)
|
||||||
|
let version_part = &line[pos..];
|
||||||
|
distro.version = version_part
|
||||||
|
.chars()
|
||||||
|
.take_while(|c| c.is_numeric() || *c == '.')
|
||||||
|
.collect::<String>();
|
||||||
|
} else {
|
||||||
|
distro.name = line.to_string();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Если pretty_name еще не установлен, используем имя из issue
|
||||||
|
if distro.pretty_name == "Linux" {
|
||||||
|
distro.pretty_name = distro.name.clone();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Для Arch Linux, которая не всегда показывает версию в os-release
|
||||||
|
if distro.name.to_lowercase().contains("arch") && distro.version.is_empty() {
|
||||||
|
// Архивные версии Arch Linux не имеют номеров версий, но можно проверить наличие файла pacman
|
||||||
|
if std::path::Path::new("/etc/pacman.conf").exists() {
|
||||||
|
distro.version = "Rolling Release".to_string();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Убираем возможные дублирования имени в версии
|
||||||
|
if !distro.version.is_empty() && distro.version.starts_with(&distro.name) {
|
||||||
|
distro.version = distro.version.trim_start_matches(&distro.name).trim().to_string();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Очищаем версию от нечисловых символов в конце
|
||||||
|
if !distro.version.is_empty() {
|
||||||
|
let cleaned_version: String = distro.version
|
||||||
|
.chars()
|
||||||
|
.take_while(|c| c.is_numeric() || *c == '.')
|
||||||
|
.collect();
|
||||||
|
if !cleaned_version.is_empty() {
|
||||||
|
distro.version = cleaned_version;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
distro
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::main]
|
||||||
|
async fn main() -> Result<(), FutriixError> {
|
||||||
|
// Инициализация логирования в файл
|
||||||
|
log_to_file("Starting Futriix server");
|
||||||
|
|
||||||
|
// Вывод приветственного сообщения с цветом #00bfff для проекта futriix
|
||||||
|
let futriix_color = get_futriix_color();
|
||||||
|
println!(); // Добавляем пустую строку перед фразой
|
||||||
|
print_futriix("Futriix Database Server");
|
||||||
|
print_futriix("futriix 3i²(by 26.11.2025)");
|
||||||
|
println!(); // Добавляем пустую строку после фразы
|
||||||
|
|
||||||
|
// Парсим аргументы командной строки
|
||||||
|
let args = parse_args();
|
||||||
|
let config_path = args.config;
|
||||||
|
|
||||||
|
let message = format!("Loading configuration from: {}", config_path);
|
||||||
|
print_futriix(&message);
|
||||||
|
|
||||||
|
// Добавляем пустую строку после сообщения о загрузке конфигурации
|
||||||
|
println!();
|
||||||
|
|
||||||
|
// Выводим информацию о дистрибутиве Linux с версией
|
||||||
|
let distro = get_linux_distribution();
|
||||||
|
print_futriix(&format!("Running on: {}", distro.full_name()));
|
||||||
|
|
||||||
|
log_to_file(&message);
|
||||||
|
log_to_file(&format!("Linux distribution: {} {}", distro.name, distro.version));
|
||||||
|
|
||||||
|
// Создание и запуск сервера
|
||||||
|
match server::FutriixServer::new(&config_path).await {
|
||||||
|
Ok(server) => {
|
||||||
|
log_to_file("Server created successfully");
|
||||||
|
if let Err(e) = server.run().await {
|
||||||
|
let error_message = format!("Server error: {}", e);
|
||||||
|
eprintln!("{}", error_message);
|
||||||
|
log_to_file(&error_message);
|
||||||
|
std::process::exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
let error_message = format!("Failed to create server: {}", e);
|
||||||
|
eprintln!("{}", error_message);
|
||||||
|
log_to_file(&error_message);
|
||||||
|
std::process::exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
log_to_file("Futriix server stopped");
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user