YGGverse/Yoda
1
0
Fork 0
mirror of https://github.com/YGGverse/Yoda.git synced 2025-01-15 17:20:08 +00:00
Code Issues Projects Releases Wiki Activity

rename field from request to url, define unique keys

Browse Source
This commit is contained in:
yggverse 2024-11-16 15:26:55 +02:00
parent 3e6f95833d
commit fdc6e739b4
1 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/profile/identity/gemini/auth/database.rs
View File

@ -8,7 +8,7 @@ pub struct Table {
pub gemini_id: i64, pub gemini_id: i64,
} }
/// Storage for `gemini_id` + `request` auth pairs /// Storage for `gemini_id` + `url` auth pairs
pub struct Database { pub struct Database {
connection: Rc<RwLock<Connection>>, connection: Rc<RwLock<Connection>>,
profile_id: Rc<i64>, // multi-profile relationship profile_id: Rc<i64>, // multi-profile relationship
@ -27,11 +27,11 @@ impl Database {
// Getters // Getters
/// Get records from database match current `profile_id` optionally filtered by `request` /// Get records from database match current `profile_id` optionally filtered by `url`
pub fn records(&self, request: Option<&str>) -> Result<Vec<Table>, Error> { pub fn records(&self, url: Option<&str>) -> Result<Vec<Table>, Error> {
let readable = self.connection.read().unwrap(); // @TODO let readable = self.connection.read().unwrap(); // @TODO
let tx = readable.unchecked_transaction()?; let tx = readable.unchecked_transaction()?;
select(&tx, *self.profile_id, request) select(&tx, *self.profile_id, url)
} }
} }
@ -44,25 +44,29 @@ pub fn init(tx: &Transaction) -> Result<usize, Error> {
`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, `id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
`profile_id` INTEGER NOT NULL, `profile_id` INTEGER NOT NULL,
`gemini_id` INTEGER NOT NULL, `gemini_id` INTEGER NOT NULL,
`request` TEXT NOT NULL `is_active` INTEGER NOT NULL,
`url` TEXT NOT NULL,
UNIQUE (
`profile_id`,
`gemini_id`,
`is_active`,
`url`
)
)", )",
[], [],
) )
} }
pub fn select( pub fn select(tx: &Transaction, profile_id: i64, url: Option<&str>) -> Result<Vec<Table>, Error> {
tx: &Transaction,
profile_id: i64,
request: Option<&str>,
) -> Result<Vec<Table>, Error> {
let mut stmt = tx.prepare( let mut stmt = tx.prepare(
"SELECT `id`, "SELECT `id`,
`profile_id`, `profile_id`,
`gemini_id` FROM `profile_identity_gemini_auth` `gemini_id` FROM `profile_identity_gemini_auth`
WHERE `profile_id` = ? AND `request` LIKE ?", WHERE `profile_id` = ? AND `url` LIKE ?",
)?; )?;
let result = stmt.query_map((profile_id, request.unwrap_or("%")), |row| { let result = stmt.query_map((profile_id, url.unwrap_or("%")), |row| {
Ok(Table { Ok(Table {
//id: row.get(0)?, //id: row.get(0)?,
//profile_id: row.get(1)?, //profile_id: row.get(1)?,