aipack 0.8.22

Command Agent runner to accelerate production coding with genai.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

use crate::model::base::{self, DbBmc};
use crate::model::{EntityType, Id, ModelManager, Result};
use crate::support::text;
use modql::SqliteFromRow;
use modql::field::{Fields, HasSqliteFields as _};

// region:    --- Public Types

// NOTE: Not needed
// #[derive(Debug, Clone, Fields, SqliteFromRow)]
// pub struct Ucontent {
// 	pub id: Id,
// 	pub uid: Uuid,

// 	pub ctime: EpochUs,
// 	pub mtime: EpochUs,

// 	pub hash: String,
// 	pub is_json: bool,
// 	pub content: Option<String>,
// }

#[derive(Debug, Clone, Fields, SqliteFromRow)]
pub struct UcontentForCreate {
	pub hash: String,
	pub is_json: bool,
	pub content: Option<String>,
}

// endregion: --- Public Types

// region:    --- Bmc

pub struct UcontentBmc;

impl DbBmc for UcontentBmc {
	const TABLE: &'static str = "ucontent";
	const ENTITY_TYPE: EntityType = EntityType::Ucontent;
}

impl UcontentBmc {
	pub fn get_or_create_for_text(mm: &ModelManager, content: &str, is_json: bool) -> Result<Id> {
		let is_json_part = if is_json { "is_json=true" } else { "is_json=false" };
		let length_part = format!("length={}", content.len());
		let hash = text::blake3_b64u(&[is_json_part, &length_part, content]);

		if let Some(id) = Self::get_id_by_hash(mm, &hash)? {
			return Ok(id);
		}

		let ucontent_c = UcontentForCreate {
			hash,
			is_json,
			content: Some(content.to_string()),
		};

		Self::create(mm, ucontent_c)
	}
}

/// Private
impl UcontentBmc {
	fn create(mm: &ModelManager, ucontent_c: UcontentForCreate) -> Result<Id> {
		let fields = ucontent_c.sqlite_not_none_fields();
		base::create::<Self>(mm, fields)
	}

	pub fn get_id_by_hash(mm: &ModelManager, hash: &str) -> Result<Option<Id>> {
		let sql = "SELECT id FROM ucontent WHERE hash = ? LIMIT 1";
		let id = mm.db().exec_returning_as_optional::<i64>(sql, (hash,))?;

		Ok(id.map(|id| id.into()))
	}
}

// endregion: --- Bmc