YGGverse/Yoda
1
0
Fork 0
mirror of https://github.com/YGGverse/Yoda.git synced 2026-04-01 00:55:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

separate children proxy components

Browse source
This commit is contained in:
yggverse 2025-07-26 04:26:20 +03:00
parent 02f6419b92
commit b15e4a5fc1
11 changed files with 310 additions and 193 deletions
Download patch file Download diff file Expand all files Collapse all files

252
src/profile/proxy/database.rs
View file

@ -1,252 +0,0 @@
mod ignore;
mod rule;
use anyhow::Result;
use gtk::glib::DateTime;
use ignore::Ignore;
use r2d2::Pool;
use r2d2_sqlite::SqliteConnectionManager;
use rule::Rule;
use sqlite::Transaction;
pub struct Database {
pool: Pool<SqliteConnectionManager>,
profile_id: i64,
}
impl Database {
// Constructors
pub fn init(pool: &Pool<SqliteConnectionManager>, profile_id: i64) -> Self {
Self {
pool: pool.clone(),
profile_id,
}
}
// Getters
pub fn rules(&self) -> Result<Vec<Rule>> {
rules(&self.pool.get()?.unchecked_transaction()?, self.profile_id)
}
pub fn ignores(&self) -> Result<Vec<Ignore>> {
ignores(&self.pool.get()?.unchecked_transaction()?, self.profile_id)
}
// Setters
pub fn clean_rules(&self, keep_id: Vec<i64>) -> Result<()> {
let mut c = self.pool.get()?;
let tx = c.transaction()?;
clean_rules(&tx, keep_id)?;
tx.commit()?;
Ok(())
}
pub fn persist_rule(
&self,
id: Option<i64>,
time: DateTime,
is_enabled: bool,
priority: i32,
request: String,
url: String,
) -> Result<i64> {
let mut c = self.pool.get()?;
let tx = c.transaction()?;
let id = match id {
Some(id) => {
update_rule(&tx, id, time, is_enabled, priority, request, url)?;
id
}
None => insert_rule(
&tx,
self.profile_id,
time,
is_enabled,
priority,
request,
url,
)?,
};
tx.commit()?;
Ok(id)
}
}
// Low-level DB API
pub fn init(tx: &Transaction) -> Result<usize> {
let mut s = 0;
s += tx.execute(
"CREATE TABLE IF NOT EXISTS `profile_proxy_ignore`
(
`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
`profile_id` INTEGER NOT NULL,
`time` INTEGER NOT NULL,
`is_enabled` INTEGER NOT NULL,
`host` VARCHAR(255) NOT NULL,
FOREIGN KEY (`profile_id`) REFERENCES `profile` (`id`)
)",
[],
)?;
s += tx.execute(
"CREATE TABLE IF NOT EXISTS `profile_proxy_rule`
(
`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
`profile_id` INTEGER NOT NULL,
`time` INTEGER NOT NULL,
`is_enabled` INTEGER NOT NULL,
`priority` INTEGER NOT NULL,
`request` VARCHAR(1024) NOT NULL,
`url` VARCHAR(255) NOT NULL,
FOREIGN KEY (`profile_id`) REFERENCES `profile` (`id`)
)",
[],
)?;
Ok(s)
}
pub fn clean_rules(tx: &Transaction, keep_id: Vec<i64>) -> Result<usize> {
if keep_id.is_empty() {
return Ok(0);
}
Ok(tx.execute(
&format!(
"DELETE FROM `profile_proxy_rule` WHERE `id` NOT IN ({})",
keep_id
.into_iter()
.map(|id| id.to_string())
.collect::<Vec<String>>()
.join(",")
),
[],
)?)
}
pub fn insert_rule(
tx: &Transaction,
profile_id: i64,
time: DateTime,
is_enabled: bool,
priority: i32,
request: String,
url: String,
) -> Result<i64> {
tx.execute(
"INSERT INTO `profile_proxy_rule` (
`profile_id`,
`time`,
`is_enabled`,
`priority`,
`request`,
`url`
) VALUES (?, ?, ?, ?, ?, ?)",
(
profile_id,
time.to_unix(),
is_enabled,
priority,
request,
url,
),
)?;
Ok(tx.last_insert_rowid())
}
pub fn update_rule(
tx: &Transaction,
id: i64,
time: DateTime,
is_enabled: bool,
priority: i32,
request: String,
url: String,
) -> Result<usize> {
Ok(tx.execute(
"UPDATE `profile_proxy_rule`
SET `time` = ?,
`is_enabled` = ?,
`priority` = ?,
`request` = ?,
`url` = ?
WHERE `id` = ?",
(time.to_unix(), is_enabled, priority, request, url, id),
)?)
}
pub fn ignores(tx: &Transaction, profile_id: i64) -> Result<Vec<Ignore>> {
let mut stmt = tx.prepare(
"SELECT `id`,
`profile_id`,
`time`,
`host`,
`is_enabled`
FROM `profile_proxy_ignore`
WHERE `profile_id` = ?",
)?;
let result = stmt.query_map([profile_id], |row| {
Ok(Ignore {
//id: row.get(0)?,
//profile_id: row.get(1)?,
//time: DateTime::from_unix_local(row.get(2)?).unwrap(),
host: row.get(3)?,
is_enabled: row.get(4)?,
})
})?;
let mut records = Vec::new();
for record in result {
let table = record?;
records.push(table);
}
Ok(records)
}
pub fn rules(tx: &Transaction, profile_id: i64) -> Result<Vec<Rule>> {
let mut stmt = tx.prepare(
"SELECT `id`,
`profile_id`,
`time`,
`is_enabled`,
`priority`,
`request`,
`url`
FROM `profile_proxy_rule`
WHERE `profile_id` = ?
ORDER BY `priority` ASC",
)?;
let result = stmt.query_map([profile_id], |row| {
Ok(Rule {
id: row.get(0)?,
//profile_id: row.get(1)?,
time: DateTime::from_unix_local(row.get(2)?).unwrap(),
is_enabled: row.get(3)?,
priority: row.get(4)?,
request: row.get(5)?,
url: row.get(6)?,
})
})?;
let mut records = Vec::new();
for record in result {
let table = record?;
records.push(table);
}
Ok(records)
}