| 1 | |
|
| 2 | |
|
| 3 | |
|
| 4 | |
|
| 5 | |
|
| 6 | |
|
| 7 | |
|
| 8 | |
|
| 9 | |
|
| 10 | | use crate::cbor::{self, Value}; |
| 11 | | use crate::hash::{hash_object, ChangeHash, KeyId, ObjectRef, ObjectType}; |
| 12 | | use crate::ids::{ChangeId, NodeId}; |
| 13 | | use crate::lines::Lines; |
| 14 | | use std::collections::BTreeMap; |
| 15 | |
|
| 16 | | pub const FORMAT_VERSION: u64 = 1; |
| 17 | |
|
| 18 | |
|
| 19 | |
|
| 20 | |
|
| 21 | |
|
| 22 | |
|
| 23 | | pub const CHANGE_VERSION: u64 = 3; |
| 24 | |
|
| 25 | |
|
| 26 | | #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)] |
| 27 | | pub enum TextForm { |
| 28 | | |
| 29 | | Cbor, |
| 30 | | |
| 31 | | Raw, |
| 32 | | |
| 33 | | |
| 34 | | CborFallback, |
| 35 | | } |
| 36 | |
|
| 37 | |
|
| 38 | |
|
| 39 | |
|
| 40 | | pub const TEXT_BLOB_MIN: usize = 4096; |
| 41 | |
|
| 42 | | #[derive(Debug, Clone, PartialEq, Eq, thiserror::Error)] |
| 43 | | pub enum ObjectError { |
| 44 | | #[error("cbor: {0}")] |
| 45 | | Cbor(#[from] cbor::CborError), |
| 46 | | #[error("schema: {0}")] |
| 47 | | Schema(String), |
| 48 | | } |
| 49 | |
|
| 50 | 18× | pub(crate) fn schema<T>(msg: impl Into<String>) -> Result<T, ObjectError> { |
| 51 | 18× | Err(ObjectError::Schema(msg.into())) |
| 52 | 18× | } |
| 53 | |
|
| 54 | | pub(crate) type Map = BTreeMap<u64, Value>; |
| 55 | |
|
| 56 | 872934× | pub(crate) fn field<'a>(m: &'a Map, k: u64, what: &str) -> Result<&'a Value, ObjectError> { |
| 57 | 872934× | m.get(&k) |
| 58 | 872934× | .ok_or_else(|| ObjectError::Schema(format!("missing field {k} ({what})"))) |
| 59 | 872934× | } |
| 60 | 359311× | pub(crate) fn uint(v: &Value, what: &str) -> Result<u64, ObjectError> { |
| 61 | 359311× | v.as_uint() |
| 62 | 359311× | .ok_or_else(|| ObjectError::Schema(format!("{what}: expected unsigned integer"))) |
| 63 | 359311× | } |
| 64 | 63415× | pub(crate) fn u32_(v: &Value, what: &str) -> Result<u32, ObjectError> { |
| 65 | 63415× | u32::try_from(uint(v, what)?).map_err(|_| ObjectError::Schema(format!("{what}: exceeds u32"))) |
| 66 | 63415× | } |
| 67 | 118662× | pub(crate) fn text<'a>(v: &'a Value, what: &str) -> Result<&'a str, ObjectError> { |
| 68 | 118662× | v.as_text() |
| 69 | 118662× | .ok_or_else(|| ObjectError::Schema(format!("{what}: expected text"))) |
| 70 | 118662× | } |
| 71 | 264721× | pub(crate) fn array<'a>(v: &'a Value, what: &str) -> Result<&'a [Value], ObjectError> { |
| 72 | 264721× | v.as_array() |
| 73 | 264721× | .ok_or_else(|| ObjectError::Schema(format!("{what}: expected array"))) |
| 74 | 264721× | } |
| 75 | 128320× | pub(crate) fn map<'a>(v: &'a Value, what: &str) -> Result<&'a Map, ObjectError> { |
| 76 | 128320× | v.as_map() |
| 77 | 128320× | .ok_or_else(|| ObjectError::Schema(format!("{what}: expected map"))) |
| 78 | 128320× | } |
| 79 | 377313× | pub(crate) fn bytes<'a>(v: &'a Value, what: &str) -> Result<&'a [u8], ObjectError> { |
| 80 | 377313× | v.as_bytes() |
| 81 | 377313× | .ok_or_else(|| ObjectError::Schema(format!("{what}: expected bytes"))) |
| 82 | 377313× | } |
| 83 | 127360× | pub(crate) fn oref(v: &Value, what: &str) -> Result<ObjectRef, ObjectError> { |
| 84 | 127360× | ObjectRef::from_slice(bytes(v, what)?) |
| 85 | 127359× | .ok_or_else(|| ObjectError::Schema(format!("{what}: expected 32 bytes"))) |
| 86 | 127360× | } |
| 87 | 145818× | pub(crate) fn node_id(v: &Value, what: &str) -> Result<NodeId, ObjectError> { |
| 88 | 145818× | NodeId::from_slice(bytes(v, what)?) |
| 89 | 145818× | .ok_or_else(|| ObjectError::Schema(format!("{what}: expected 16 bytes"))) |
| 90 | 145818× | } |
| 91 | 25705× | pub(crate) fn check_version(m: &Map) -> Result<(), ObjectError> { |
| 92 | 25705× | let v = uint(field(m, 0, "version")?, "version")?; |
| 93 | 25705× | if v != FORMAT_VERSION { |
| 94 | 0× | let exe = std::env::current_exe() |
| 95 | 0× | .map(|p| format!(" at {}", p.display())) |
| 96 | 0× | .unwrap_or_default(); |
| 97 | 0× | let which = format!("oxy {}{exe}", crate::VERSION); |
| 98 | 0× | return schema(if v > FORMAT_VERSION { |
| 99 | 0× | format!( |
| 100 | | "this data is format {v}, written by a newer oxy, and {which} reads format {FORMAT_VERSION}: install the current oxy, and check that no older copy comes first on PATH" |
| 101 | | ) |
| 102 | | } else { |
| 103 | 0× | format!( |
| 104 | | "this data is format {v}, written by an older oxy, and {which} reads format {FORMAT_VERSION} only" |
| 105 | | ) |
| 106 | | }); |
| 107 | 25705× | } |
| 108 | 25705× | Ok(()) |
| 109 | 25705× | } |
| 110 | |
|
| 111 | |
|
| 112 | |
|
| 113 | |
|
| 114 | | #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash, Default)] |
| 115 | | pub struct Hlc { |
| 116 | | pub wall_ms: u64, |
| 117 | | pub logical: u32, |
| 118 | | } |
| 119 | |
|
| 120 | | impl Hlc { |
| 121 | 5728× | pub fn now() -> Hlc { |
| 122 | 5728× | let wall_ms = std::time::SystemTime::now() |
| 123 | 5728× | .duration_since(std::time::UNIX_EPOCH) |
| 124 | 5728× | .map(|d| d.as_millis() as u64) |
| 125 | 5728× | .unwrap_or(0); |
| 126 | 5728× | Hlc { |
| 127 | 5728× | wall_ms, |
| 128 | 5728× | logical: 0, |
| 129 | 5728× | } |
| 130 | 5728× | } |
| 131 | 11419× | pub(crate) fn to_value(self) -> Value { |
| 132 | 11419× | Value::map([ |
| 133 | 11419× | (0, Value::Uint(self.wall_ms)), |
| 134 | 11419× | (1, Value::Uint(self.logical as u64)), |
| 135 | 11419× | ]) |
| 136 | 11419× | } |
| 137 | 33059× | pub(crate) fn from_value(v: &Value) -> Result<Hlc, ObjectError> { |
| 138 | 33059× | let m = map(v, "hlc")?; |
| 139 | | Ok(Hlc { |
| 140 | 33059× | wall_ms: uint(field(m, 0, "hlc.wall_ms")?, "hlc.wall_ms")?, |
| 141 | 33059× | logical: u32_(field(m, 1, "hlc.logical")?, "hlc.logical")?, |
| 142 | | }) |
| 143 | 33059× | } |
| 144 | | } |
| 145 | |
|
| 146 | |
|
| 147 | |
|
| 148 | | #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)] |
| 149 | | pub enum NodeKind { |
| 150 | | File, |
| 151 | | Dir, |
| 152 | | Symlink, |
| 153 | | |
| 154 | | Subrepo, |
| 155 | | } |
| 156 | |
|
| 157 | | impl NodeKind { |
| 158 | 72894× | pub(crate) fn tag(self) -> u64 { |
| 159 | 72894× | match self { |
| 160 | 71977× | NodeKind::File => 0, |
| 161 | 894× | NodeKind::Dir => 1, |
| 162 | 3× | NodeKind::Symlink => 2, |
| 163 | 20× | NodeKind::Subrepo => 3, |
| 164 | | } |
| 165 | 72894× | } |
| 166 | 96152× | pub(crate) fn from_tag(t: u64) -> Result<NodeKind, ObjectError> { |
| 167 | 96152× | Ok(match t { |
| 168 | 94159× | 0 => NodeKind::File, |
| 169 | 1892× | 1 => NodeKind::Dir, |
| 170 | 9× | 2 => NodeKind::Symlink, |
| 171 | 92× | 3 => NodeKind::Subrepo, |
| 172 | 0× | _ => return schema(format!("unknown node kind {t}")), |
| 173 | | }) |
| 174 | 96152× | } |
| 175 | | } |
| 176 | |
|
| 177 | |
|
| 178 | | #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Default, PartialOrd, Ord)] |
| 179 | | pub struct Mode { |
| 180 | | pub exec: bool, |
| 181 | | pub opaque: bool, |
| 182 | | } |
| 183 | |
|
| 184 | | impl Mode { |
| 185 | | pub const TEXT: Mode = Mode { |
| 186 | | exec: false, |
| 187 | | opaque: false, |
| 188 | | }; |
| 189 | | pub const OPAQUE: Mode = Mode { |
| 190 | | exec: false, |
| 191 | | opaque: true, |
| 192 | | }; |
| 193 | 72553× | pub(crate) fn bits(self) -> u64 { |
| 194 | 72553× | (self.exec as u64) | ((self.opaque as u64) << 1) |
| 195 | 72553× | } |
| 196 | 96112× | pub(crate) fn from_bits(b: u64) -> Result<Mode, ObjectError> { |
| 197 | 96112× | if b > 3 { |
| 198 | 1× | return schema(format!("unknown mode bits {b}")); |
| 199 | 96111× | } |
| 200 | 96111× | Ok(Mode { |
| 201 | 96111× | exec: b & 1 != 0, |
| 202 | 96111× | opaque: b & 2 != 0, |
| 203 | 96111× | }) |
| 204 | 96112× | } |
| 205 | | } |
| 206 | |
|
| 207 | |
|
| 208 | |
|
| 209 | | #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord)] |
| 210 | | pub struct OpRef { |
| 211 | | pub change: ChangeHash, |
| 212 | | pub index: u32, |
| 213 | | } |
| 214 | |
|
| 215 | | impl OpRef { |
| 216 | 553× | fn to_value(self) -> Value { |
| 217 | 553× | Value::Array(vec![ |
| 218 | 553× | Value::Bytes(self.change.0.to_vec()), |
| 219 | 553× | Value::Uint(self.index as u64), |
| 220 | 553× | ]) |
| 221 | 553× | } |
| 222 | 830× | fn from_value(v: &Value, what: &str) -> Result<OpRef, ObjectError> { |
| 223 | 830× | let a = array(v, what)?; |
| 224 | 830× | if a.len() != 2 { |
| 225 | 0× | return schema(format!("{what}: expected [hash, index]")); |
| 226 | 830× | } |
| 227 | | Ok(OpRef { |
| 228 | 830× | change: oref(&a[0], what)?, |
| 229 | 830× | index: u32_(&a[1], what)?, |
| 230 | | }) |
| 231 | 830× | } |
| 232 | | |
| 233 | 841× | pub fn resolve(self, me: ChangeHash) -> OpRef { |
| 234 | 841× | if self.change.is_zero() { |
| 235 | 1× | OpRef { change: me, ..self } |
| 236 | | } else { |
| 237 | 840× | self |
| 238 | | } |
| 239 | 841× | } |
| 240 | | } |
| 241 | |
|
| 242 | | #[derive(Debug, Clone, PartialEq, Eq)] |
| 243 | | pub enum TreeOp { |
| 244 | | AddNode { |
| 245 | | node: NodeId, |
| 246 | | kind: NodeKind, |
| 247 | | mode: Mode, |
| 248 | | parent: NodeId, |
| 249 | | name: String, |
| 250 | | }, |
| 251 | | RemoveNode { |
| 252 | | node: NodeId, |
| 253 | | placement: OpRef, |
| 254 | | }, |
| 255 | | Move { |
| 256 | | node: NodeId, |
| 257 | | old: OpRef, |
| 258 | | parent: NodeId, |
| 259 | | name: String, |
| 260 | | }, |
| 261 | | SetMode { |
| 262 | | node: NodeId, |
| 263 | | old: OpRef, |
| 264 | | mode: Mode, |
| 265 | | }, |
| 266 | | } |
| 267 | |
|
| 268 | |
|
| 269 | 46394× | pub fn validate_name(name: &str) -> Result<(), ObjectError> { |
| 270 | 46394× | if name.is_empty() || name == "." || name == ".." || name.contains('/') || name.contains('\0') { |
| 271 | 6× | return schema(format!("invalid placement name {name:?}")); |
| 272 | 46388× | } |
| 273 | 46388× | Ok(()) |
| 274 | 46394× | } |
| 275 | |
|
| 276 | | impl TreeOp { |
| 277 | 115× | pub fn node(&self) -> NodeId { |
| 278 | 115× | match self { |
| 279 | 94× | TreeOp::AddNode { node, .. } |
| 280 | 10× | | TreeOp::RemoveNode { node, .. } |
| 281 | 10× | | TreeOp::Move { node, .. } |
| 282 | 115× | | TreeOp::SetMode { node, .. } => *node, |
| 283 | | } |
| 284 | 115× | } |
| 285 | |
|
| 286 | 6190× | fn to_value(&self) -> Value { |
| 287 | 11852× | let nid = |n: &NodeId| Value::Bytes(n.0.to_vec()); |
| 288 | 6190× | match self { |
| 289 | | TreeOp::AddNode { |
| 290 | 5637× | node, |
| 291 | 5637× | kind, |
| 292 | 5637× | mode, |
| 293 | 5637× | parent, |
| 294 | 5637× | name, |
| 295 | 5637× | } => Value::map([ |
| 296 | 5637× | (0, Value::Uint(0)), |
| 297 | 5637× | (1, nid(node)), |
| 298 | 5637× | (2, Value::Uint(kind.tag())), |
| 299 | 5637× | (3, Value::Uint(mode.bits())), |
| 300 | 5637× | (4, nid(parent)), |
| 301 | 5637× | (5, Value::Text(name.clone())), |
| 302 | 5637× | ]), |
| 303 | 465× | TreeOp::RemoveNode { node, placement } => Value::map([ |
| 304 | 465× | (0, Value::Uint(1)), |
| 305 | 465× | (1, nid(node)), |
| 306 | 465× | (2, placement.to_value()), |
| 307 | 465× | ]), |
| 308 | | TreeOp::Move { |
| 309 | 25× | node, |
| 310 | 25× | old, |
| 311 | 25× | parent, |
| 312 | 25× | name, |
| 313 | 25× | } => Value::map([ |
| 314 | 25× | (0, Value::Uint(2)), |
| 315 | 25× | (1, nid(node)), |
| 316 | 25× | (2, old.to_value()), |
| 317 | 25× | (4, nid(parent)), |
| 318 | 25× | (5, Value::Text(name.clone())), |
| 319 | 25× | ]), |
| 320 | 63× | TreeOp::SetMode { node, old, mode } => Value::map([ |
| 321 | 63× | (0, Value::Uint(3)), |
| 322 | 63× | (1, nid(node)), |
| 323 | 63× | (2, old.to_value()), |
| 324 | 63× | (3, Value::Uint(mode.bits())), |
| 325 | 63× | ]), |
| 326 | | } |
| 327 | 6190× | } |
| 328 | |
|
| 329 | 47071× | fn from_value(v: &Value) -> Result<TreeOp, ObjectError> { |
| 330 | 47071× | let m = map(v, "tree_op")?; |
| 331 | 47071× | let tag = uint(field(m, 0, "tree_op tag")?, "tree_op tag")?; |
| 332 | 47071× | let node = node_id(field(m, 1, "tree_op.node")?, "tree_op.node")?; |
| 333 | 47071× | let op = match tag { |
| 334 | | 0 => { |
| 335 | 46241× | let name = text(field(m, 5, "name")?, "name")?.to_owned(); |
| 336 | 46241× | validate_name(&name)?; |
| 337 | | TreeOp::AddNode { |
| 338 | 46241× | node, |
| 339 | 46241× | kind: NodeKind::from_tag(uint(field(m, 2, "kind")?, "kind")?)?, |
| 340 | 46241× | mode: Mode::from_bits(uint(field(m, 3, "mode")?, "mode")?)?, |
| 341 | 46241× | parent: node_id(field(m, 4, "parent")?, "parent")?, |
| 342 | 46241× | name, |
| 343 | | } |
| 344 | | } |
| 345 | | 1 => TreeOp::RemoveNode { |
| 346 | 662× | node, |
| 347 | 662× | placement: OpRef::from_value(field(m, 2, "placement")?, "placement")?, |
| 348 | | }, |
| 349 | | 2 => { |
| 350 | 134× | let name = text(field(m, 5, "name")?, "name")?.to_owned(); |
| 351 | 134× | validate_name(&name)?; |
| 352 | | TreeOp::Move { |
| 353 | 134× | node, |
| 354 | 134× | old: OpRef::from_value(field(m, 2, "old")?, "old")?, |
| 355 | 134× | parent: node_id(field(m, 4, "parent")?, "parent")?, |
| 356 | 134× | name, |
| 357 | | } |
| 358 | | } |
| 359 | | 3 => TreeOp::SetMode { |
| 360 | 34× | node, |
| 361 | 34× | old: OpRef::from_value(field(m, 2, "old")?, "old")?, |
| 362 | 34× | mode: Mode::from_bits(uint(field(m, 3, "mode")?, "mode")?)?, |
| 363 | | }, |
| 364 | 0× | _ => return schema(format!("unknown tree op tag {tag}")), |
| 365 | | }; |
| 366 | 47071× | if node.is_root() { |
| 367 | 0× | return schema("tree op on the root node"); |
| 368 | 47071× | } |
| 369 | 47071× | Ok(op) |
| 370 | 47071× | } |
| 371 | | } |
| 372 | |
|
| 373 | |
|
| 374 | |
|
| 375 | | #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)] |
| 376 | | pub enum AtomRef { |
| 377 | | Start, |
| 378 | | End, |
| 379 | | Atom { change: ChangeHash, index: u32 }, |
| 380 | | } |
| 381 | |
|
| 382 | | impl AtomRef { |
| 383 | 15212× | fn to_value(self) -> Value { |
| 384 | 15212× | match self { |
| 385 | 5414× | AtomRef::Start => Value::Uint(0), |
| 386 | 5560× | AtomRef::End => Value::Uint(1), |
| 387 | 4238× | AtomRef::Atom { change, index } => Value::Array(vec![ |
| 388 | 4238× | Value::Bytes(change.0.to_vec()), |
| 389 | 4238× | Value::Uint(index as u64), |
| 390 | 4238× | ]), |
| 391 | | } |
| 392 | 15212× | } |
| 393 | 97682× | fn from_value(v: &Value, what: &str) -> Result<AtomRef, ObjectError> { |
| 394 | 7857× | match v { |
| 395 | 44252× | Value::Uint(0) => Ok(AtomRef::Start), |
| 396 | 45573× | Value::Uint(1) => Ok(AtomRef::End), |
| 397 | 7857× | Value::Array(a) if a.len() == 2 => Ok(AtomRef::Atom { |
| 398 | 7857× | change: oref(&a[0], what)?, |
| 399 | 7857× | index: u32_(&a[1], what)?, |
| 400 | | }), |
| 401 | 0× | _ => schema(format!("{what}: expected atom reference")), |
| 402 | | } |
| 403 | 97682× | } |
| 404 | 61376× | pub fn resolve(self, me: ChangeHash) -> AtomRef { |
| 405 | 9036× | match self { |
| 406 | 9036× | AtomRef::Atom { change, index } if change.is_zero() => { |
| 407 | 34× | AtomRef::Atom { change: me, index } |
| 408 | | } |
| 409 | 61342× | other => other, |
| 410 | | } |
| 411 | 61376× | } |
| 412 | | } |
| 413 | |
|
| 414 | |
|
| 415 | | #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)] |
| 416 | | pub struct AtomRange { |
| 417 | | pub change: ChangeHash, |
| 418 | | pub from: u32, |
| 419 | | pub to: u32, |
| 420 | | } |
| 421 | |
|
| 422 | | impl AtomRange { |
| 423 | 2396× | fn to_value(self) -> Value { |
| 424 | 2396× | Value::Array(vec![ |
| 425 | 2396× | Value::Bytes(self.change.0.to_vec()), |
| 426 | 2396× | Value::Uint(self.from as u64), |
| 427 | 2396× | Value::Uint(self.to as u64), |
| 428 | 2396× | ]) |
| 429 | 2396× | } |
| 430 | 3599× | fn from_value(v: &Value) -> Result<AtomRange, ObjectError> { |
| 431 | 3599× | let a = array(v, "atom range")?; |
| 432 | 3599× | if a.len() != 3 { |
| 433 | 0× | return schema("atom range: expected [hash, from, to]"); |
| 434 | 3599× | } |
| 435 | 3599× | let r = AtomRange { |
| 436 | 3599× | change: oref(&a[0], "range.change")?, |
| 437 | 3599× | from: u32_(&a[1], "range.from")?, |
| 438 | 3599× | to: u32_(&a[2], "range.to")?, |
| 439 | | }; |
| 440 | 3599× | if r.from >= r.to { |
| 441 | 0× | return schema("atom range: empty or inverted"); |
| 442 | 3599× | } |
| 443 | 3599× | Ok(r) |
| 444 | 3599× | } |
| 445 | 4303× | pub fn resolve(self, me: ChangeHash) -> AtomRange { |
| 446 | 4303× | if self.change.is_zero() { |
| 447 | 18× | AtomRange { change: me, ..self } |
| 448 | | } else { |
| 449 | 4285× | self |
| 450 | | } |
| 451 | 4303× | } |
| 452 | | } |
| 453 | |
|
| 454 | | #[derive(Debug, Clone, PartialEq, Eq)] |
| 455 | | pub enum Edit { |
| 456 | | Insert { |
| 457 | | node: NodeId, |
| 458 | | after: AtomRef, |
| 459 | | before: AtomRef, |
| 460 | | lines: crate::lines::Lines, |
| 461 | | }, |
| 462 | | Delete { |
| 463 | | node: NodeId, |
| 464 | | atoms: Vec<AtomRange>, |
| 465 | | }, |
| 466 | | Replace { |
| 467 | | node: NodeId, |
| 468 | | old: Option<ObjectRef>, |
| 469 | | new: Option<ObjectRef>, |
| 470 | | }, |
| 471 | | } |
| 472 | |
|
| 473 | | impl Edit { |
| 474 | 45951× | pub fn node(&self) -> NodeId { |
| 475 | 45951× | match self { |
| 476 | 42906× | Edit::Insert { node, .. } | Edit::Delete { node, .. } | Edit::Replace { node, .. } => { |
| 477 | 45951× | *node |
| 478 | | } |
| 479 | | } |
| 480 | 45951× | } |
| 481 | |
|
| 482 | | |
| 483 | 9829× | fn stored_text(&self, version: u64) -> bool { |
| 484 | 7609× | matches!(self, Edit::Insert { lines, .. } |
| 485 | 7609× | if version >= 2 && cbor::byte_strings_len(lines) >= TEXT_BLOB_MIN) |
| 486 | 9829× | } |
| 487 | |
|
| 488 | | |
| 489 | | |
| 490 | | |
| 491 | | |
| 492 | 9822× | fn to_value(&self, stored: Option<(ObjectRef, TextForm)>, compact: bool) -> Value { |
| 493 | 9822× | let nid = |n: &NodeId| Value::Bytes(n.0.to_vec()); |
| 494 | 9822× | let opt = |o: &Option<ObjectRef>| match o { |
| 495 | 173× | Some(r) => Value::Bytes(r.0.to_vec()), |
| 496 | 115× | None => Value::Null, |
| 497 | 288× | }; |
| 498 | 9822× | match (self, stored) { |
| 499 | | ( |
| 500 | | Edit::Insert { |
| 501 | 70× | node, |
| 502 | 70× | after, |
| 503 | 70× | before, |
| 504 | 70× | lines, |
| 505 | | }, |
| 506 | 70× | Some((r, form)), |
| 507 | | ) => { |
| 508 | 70× | let mut fields = vec![ |
| 509 | 70× | (0, Value::Uint(0)), |
| 510 | 70× | (1, nid(node)), |
| 511 | 70× | (2, after.to_value()), |
| 512 | 70× | (3, before.to_value()), |
| 513 | 70× | (5, Value::Bytes(r.0.to_vec())), |
| 514 | 70× | (6, Value::Uint(lines.len() as u64)), |
| 515 | | ]; |
| 516 | 70× | if form == TextForm::CborFallback { |
| 517 | 1× | fields.push((7, Value::Uint(1))); |
| 518 | 69× | } |
| 519 | 70× | Value::map(fields) |
| 520 | | } |
| 521 | | ( |
| 522 | | Edit::Insert { |
| 523 | 7536× | node, |
| 524 | 7536× | after, |
| 525 | 7536× | before, |
| 526 | 7536× | lines, |
| 527 | | }, |
| 528 | | None, |
| 529 | 7536× | ) => Value::map([ |
| 530 | 7536× | (0, Value::Uint(0)), |
| 531 | 7536× | (1, nid(node)), |
| 532 | 7536× | (2, after.to_value()), |
| 533 | 7536× | (3, before.to_value()), |
| 534 | | ( |
| 535 | | 4, |
| 536 | 7536× | if compact { |
| 537 | 7530× | Value::Encoded(cbor::encode_byte_strings(lines)) |
| 538 | | } else { |
| 539 | 1008× | Value::Array(lines.iter().map(|l| Value::Bytes(l.to_vec())).collect()) |
| 540 | | }, |
| 541 | | ), |
| 542 | | ]), |
| 543 | 2072× | (Edit::Delete { node, atoms }, _) => Value::map([ |
| 544 | 2072× | (0, Value::Uint(1)), |
| 545 | 2072× | (1, nid(node)), |
| 546 | | ( |
| 547 | | 2, |
| 548 | 2396× | Value::Array(atoms.iter().map(|r| r.to_value()).collect()), |
| 549 | | ), |
| 550 | | ]), |
| 551 | 144× | (Edit::Replace { node, old, new }, _) => Value::map([ |
| 552 | 144× | (0, Value::Uint(2)), |
| 553 | 144× | (1, nid(node)), |
| 554 | 144× | (2, opt(old)), |
| 555 | 144× | (3, opt(new)), |
| 556 | 144× | ]), |
| 557 | | } |
| 558 | 9822× | } |
| 559 | |
|
| 560 | 7738× | fn from_value(v: &Value, version: u64, resolve: &mut Resolver) -> Result<Edit, ObjectError> { |
| 561 | 7738× | let m = map(v, "edit")?; |
| 562 | 7738× | let tag = uint(field(m, 0, "edit tag")?, "edit tag")?; |
| 563 | 7738× | let node = node_id(field(m, 1, "edit.node")?, "edit.node")?; |
| 564 | 7738× | let opt = |v: &Value, what: &str| -> Result<Option<ObjectRef>, ObjectError> { |
| 565 | 1568× | if v.is_null() { |
| 566 | 689× | Ok(None) |
| 567 | | } else { |
| 568 | 879× | Ok(Some(oref(v, what)?)) |
| 569 | | } |
| 570 | 1568× | }; |
| 571 | 7738× | Ok(match tag { |
| 572 | | 0 => { |
| 573 | 4210× | let lines = match (m.get(&4), m.get(&5)) { |
| 574 | 3921× | (Some(inline), None) => { |
| 575 | 3921× | let lines = array_lines(inline)?; |
| 576 | 3921× | if version >= 2 && cbor::byte_strings_len(&lines) >= TEXT_BLOB_MIN { |
| 577 | 1× | return schema("an insert this large is stored as a text blob in a version 2 change"); |
| 578 | 3920× | } |
| 579 | 3920× | if lines.is_empty() { |
| 580 | 0× | return schema("insert with no lines"); |
| 581 | 3920× | } |
| 582 | 3920× | lines |
| 583 | | } |
| 584 | 289× | (None, Some(text)) if version >= 2 => { |
| 585 | 289× | let text = oref(text, "insert text")?; |
| 586 | 289× | let count = u32_(field(m, 6, "insert line count")?, "insert line count")?; |
| 587 | 289× | if count == 0 { |
| 588 | 0× | return schema("insert with no lines"); |
| 589 | 289× | } |
| 590 | 289× | let form = match (version, m.get(&7)) { |
| 591 | 1× | (2, None) => TextForm::Cbor, |
| 592 | 287× | (_, None) => TextForm::Raw, |
| 593 | 1× | (3.., Some(Value::Uint(1))) => TextForm::CborFallback, |
| 594 | 0× | _ => return schema("unknown text blob form"), |
| 595 | | }; |
| 596 | | |
| 597 | 289× | resolve(&text, count, form)? |
| 598 | | } |
| 599 | | _ => { |
| 600 | 0× | return schema( |
| 601 | | "an insert has either its lines or, in a version 2 change, a text blob", |
| 602 | | ) |
| 603 | | } |
| 604 | | }; |
| 605 | | Edit::Insert { |
| 606 | 4207× | node, |
| 607 | 4207× | after: AtomRef::from_value(field(m, 2, "after")?, "after")?, |
| 608 | 4207× | before: AtomRef::from_value(field(m, 3, "before")?, "before")?, |
| 609 | 4207× | lines, |
| 610 | | } |
| 611 | | } |
| 612 | | 1 => { |
| 613 | 2744× | let atoms = array(field(m, 2, "atoms")?, "atoms")? |
| 614 | 2744× | .iter() |
| 615 | 2744× | .map(AtomRange::from_value) |
| 616 | 2744× | .collect::<Result<Vec<_>, _>>()?; |
| 617 | 2744× | if atoms.is_empty() { |
| 618 | 0× | return schema("delete with no atoms"); |
| 619 | 2744× | } |
| 620 | 2744× | Edit::Delete { node, atoms } |
| 621 | | } |
| 622 | | 2 => Edit::Replace { |
| 623 | 784× | node, |
| 624 | 784× | old: opt(field(m, 2, "old")?, "old")?, |
| 625 | 784× | new: opt(field(m, 3, "new")?, "new")?, |
| 626 | | }, |
| 627 | 0× | _ => return schema(format!("unknown edit tag {tag}")), |
| 628 | | }) |
| 629 | 7738× | } |
| 630 | | } |
| 631 | |
|
| 632 | |
|
| 633 | |
|
| 634 | | #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)] |
| 635 | | pub enum ChangeKind { |
| 636 | | Normal, |
| 637 | | Resolution, |
| 638 | | WorkingCopy, |
| 639 | | } |
| 640 | |
|
| 641 | | impl ChangeKind { |
| 642 | 11546× | pub(crate) fn tag(self) -> u64 { |
| 643 | 11546× | match self { |
| 644 | 9088× | ChangeKind::Normal => 0, |
| 645 | 41× | ChangeKind::Resolution => 1, |
| 646 | 2417× | ChangeKind::WorkingCopy => 2, |
| 647 | | } |
| 648 | 11546× | } |
| 649 | 17449× | pub(crate) fn from_tag(t: u64) -> Result<ChangeKind, ObjectError> { |
| 650 | 17449× | Ok(match t { |
| 651 | 14574× | 0 => ChangeKind::Normal, |
| 652 | 118× | 1 => ChangeKind::Resolution, |
| 653 | 2757× | 2 => ChangeKind::WorkingCopy, |
| 654 | 0× | _ => return schema(format!("unknown change kind {t}")), |
| 655 | | }) |
| 656 | 17449× | } |
| 657 | | } |
| 658 | |
|
| 659 | |
|
| 660 | | #[derive(Debug, Clone, PartialEq, Eq)] |
| 661 | | pub struct Change { |
| 662 | | |
| 663 | | |
| 664 | | pub version: u64, |
| 665 | | pub change_id: ChangeId, |
| 666 | | pub deps: Vec<ChangeHash>, |
| 667 | | pub author: KeyId, |
| 668 | | pub hlc: Hlc, |
| 669 | | pub message: String, |
| 670 | | pub tree_ops: Vec<TreeOp>, |
| 671 | | pub edits: Vec<Edit>, |
| 672 | | pub kind: ChangeKind, |
| 673 | | } |
| 674 | |
|
| 675 | | impl Change { |
| 676 | 5548× | pub fn normalize(&mut self) { |
| 677 | 5548× | self.deps.sort(); |
| 678 | 5548× | self.deps.dedup(); |
| 679 | 5548× | } |
| 680 | |
|
| 681 | 6× | pub fn to_value(&self) -> Value { |
| 682 | 6× | self.value_with(&mut |_| true, false) |
| 683 | 6× | } |
| 684 | |
|
| 685 | | |
| 686 | | |
| 687 | | |
| 688 | | |
| 689 | 7727× | fn stored_texts(&self, sink: &mut TextSink) -> Vec<Option<(ObjectRef, TextForm)>> { |
| 690 | | use rayon::prelude::*; |
| 691 | 7727× | let mut refs: Vec<Option<(ObjectRef, TextForm)>> = vec![None; self.edits.len()]; |
| 692 | 7727× | let stored: Vec<usize> = (0..self.edits.len()) |
| 693 | 9822× | .filter(|i| self.edits[*i].stored_text(self.version)) |
| 694 | 7727× | .collect(); |
| 695 | | |
| 696 | | |
| 697 | | |
| 698 | 7727× | let mut batches: Vec<&[usize]> = Vec::new(); |
| 699 | 7727× | let (mut start, mut bytes) = (0, 0usize); |
| 700 | 7727× | for (k, i) in stored.iter().enumerate() { |
| 701 | 70× | if let Edit::Insert { lines, .. } = &self.edits[*i] { |
| 702 | 70× | bytes += lines.bytes().len(); |
| 703 | 70× | } |
| 704 | 70× | if k + 1 - start == 256 || bytes >= 32 << 20 { |
| 705 | 0× | batches.push(&stored[start..=k]); |
| 706 | 0× | (start, bytes) = (k + 1, 0); |
| 707 | 70× | } |
| 708 | | } |
| 709 | 7727× | if start < stored.len() { |
| 710 | 69× | batches.push(&stored[start..]); |
| 711 | 7658× | } |
| 712 | 7727× | for batch in batches { |
| 713 | 69× | let made: Vec<StoredText> = batch |
| 714 | 69× | .par_iter() |
| 715 | 70× | .map(|i| { |
| 716 | 70× | let Edit::Insert { lines, .. } = &self.edits[*i] else { |
| 717 | 0× | unreachable!("only inserts are stored as text"); |
| 718 | | }; |
| 719 | 70× | let form = text_form(self.version, lines); |
| 720 | 70× | let content = match form { |
| 721 | 68× | TextForm::Raw => lines.bytes().to_vec(), |
| 722 | 2× | TextForm::Cbor | TextForm::CborFallback => cbor::encode_byte_strings(lines), |
| 723 | | }; |
| 724 | 70× | StoredText { |
| 725 | 70× | blob: Blob::of(&content), |
| 726 | 70× | content, |
| 727 | 70× | form, |
| 728 | 70× | } |
| 729 | 70× | }) |
| 730 | 69× | .collect(); |
| 731 | 70× | for (i, t) in batch.iter().zip(&made) { |
| 732 | 70× | refs[*i] = Some((t.blob.hash(), t.form)); |
| 733 | 70× | } |
| 734 | 69× | if !sink(made) { |
| 735 | 0× | break; |
| 736 | 69× | } |
| 737 | | } |
| 738 | 7727× | refs |
| 739 | 7727× | } |
| 740 | |
|
| 741 | | |
| 742 | | |
| 743 | 7727× | fn value_with(&self, sink: &mut TextSink, compact: bool) -> Value { |
| 744 | 7727× | Value::map([ |
| 745 | 7727× | (0, Value::Uint(self.version)), |
| 746 | 7727× | (1, Value::Bytes(self.change_id.0.to_vec())), |
| 747 | | ( |
| 748 | | 2, |
| 749 | | Value::Array( |
| 750 | 7727× | self.deps |
| 751 | 7727× | .iter() |
| 752 | 9419× | .map(|d| Value::Bytes(d.0.to_vec())) |
| 753 | 7727× | .collect(), |
| 754 | | ), |
| 755 | | ), |
| 756 | 7727× | (3, Value::Bytes(self.author.0.to_vec())), |
| 757 | 7727× | (4, self.hlc.to_value()), |
| 758 | 7727× | (5, Value::Text(self.message.clone())), |
| 759 | | ( |
| 760 | | 6, |
| 761 | 7727× | Value::Array(self.tree_ops.iter().map(|t| t.to_value()).collect()), |
| 762 | | ), |
| 763 | | ( |
| 764 | | 7, |
| 765 | | Value::Array( |
| 766 | 7727× | self.edits |
| 767 | 7727× | .iter() |
| 768 | 7727× | .zip(self.stored_texts(sink)) |
| 769 | 9822× | .map(|(e, stored)| e.to_value(stored, compact)) |
| 770 | 7727× | .collect(), |
| 771 | | ), |
| 772 | | ), |
| 773 | 7727× | (8, Value::Uint(self.kind.tag())), |
| 774 | 7727× | (9, Value::Map(BTreeMap::new())), |
| 775 | | ]) |
| 776 | 7727× | } |
| 777 | |
|
| 778 | 4060× | pub fn encode(&self) -> Vec<u8> { |
| 779 | 4060× | debug_assert!( |
| 780 | 4060× | self.deps.windows(2).all(|w| w[0] < w[1]), |
| 781 | | "deps must be normalized" |
| 782 | | ); |
| 783 | 4060× | cbor::encode(&self.value_with(&mut |_| true, true)) |
| 784 | 4060× | } |
| 785 | |
|
| 786 | | |
| 787 | | |
| 788 | 115× | pub fn encode_parts(&self) -> (Vec<u8>, Vec<StoredText>) { |
| 789 | 115× | debug_assert!( |
| 790 | 115× | self.deps.windows(2).all(|w| w[0] < w[1]), |
| 791 | | "deps must be normalized" |
| 792 | | ); |
| 793 | 115× | let mut texts = Vec::new(); |
| 794 | 115× | let bytes = cbor::encode(&self.value_with( |
| 795 | 5× | &mut |batch| { |
| 796 | 5× | texts.extend(batch); |
| 797 | 5× | true |
| 798 | 5× | }, |
| 799 | | true, |
| 800 | | )); |
| 801 | 115× | (bytes, texts) |
| 802 | 115× | } |
| 803 | |
|
| 804 | | |
| 805 | | |
| 806 | | |
| 807 | 3546× | pub fn encode_streaming<E>( |
| 808 | 3546× | &self, |
| 809 | 3546× | mut store: impl FnMut(Vec<StoredText>) -> Result<(), E>, |
| 810 | 3546× | ) -> Result<Vec<u8>, E> { |
| 811 | 3546× | debug_assert!( |
| 812 | 3546× | self.deps.windows(2).all(|w| w[0] < w[1]), |
| 813 | | "deps must be normalized" |
| 814 | | ); |
| 815 | 3546× | let mut failed = None; |
| 816 | 3546× | let value = self.value_with( |
| 817 | 59× | &mut |batch| match store(batch) { |
| 818 | 59× | Ok(()) => true, |
| 819 | 0× | Err(e) => { |
| 820 | 0× | failed = Some(e); |
| 821 | 0× | false |
| 822 | | } |
| 823 | 59× | }, |
| 824 | | true, |
| 825 | | ); |
| 826 | 3546× | match failed { |
| 827 | 0× | Some(e) => Err(e), |
| 828 | 3546× | None => Ok(cbor::encode(&value)), |
| 829 | | } |
| 830 | 3546× | } |
| 831 | |
|
| 832 | | |
| 833 | 3× | pub fn has_stored_text(&self) -> bool { |
| 834 | 7× | self.edits.iter().any(|e| e.stored_text(self.version)) |
| 835 | 3× | } |
| 836 | |
|
| 837 | 4053× | pub fn hash(&self) -> ChangeHash { |
| 838 | 4053× | hash_object(ObjectType::Change, &self.encode()) |
| 839 | 4053× | } |
| 840 | |
|
| 841 | | |
| 842 | | |
| 843 | 4× | pub fn decode(bytes: &[u8]) -> Result<Change, ObjectError> { |
| 844 | 4× | Self::decode_with(bytes, &mut |r, _, _| { |
| 845 | 1× | schema(format!( |
| 846 | | "text blob {} needs a store to read it from", |
| 847 | 1× | r.short(12) |
| 848 | | )) |
| 849 | 1× | }) |
| 850 | 4× | } |
| 851 | |
|
| 852 | | |
| 853 | | |
| 854 | 5239× | pub fn decode_with(bytes: &[u8], resolve: &mut Resolver) -> Result<Change, ObjectError> { |
| 855 | 5239× | let v = cbor::decode(bytes)?; |
| 856 | 5239× | Self::from_value_with(&v, resolve) |
| 857 | 5239× | } |
| 858 | |
|
| 859 | 10047× | pub fn from_value(v: &Value) -> Result<Change, ObjectError> { |
| 860 | 10047× | Self::from_value_with(v, &mut |r, _, _| { |
| 861 | 0× | schema(format!( |
| 862 | | "text blob {} needs a store to read it from", |
| 863 | 0× | r.short(12) |
| 864 | | )) |
| 865 | 0× | }) |
| 866 | 10047× | } |
| 867 | |
|
| 868 | 15286× | pub(crate) fn from_value_with( |
| 869 | 15286× | v: &Value, |
| 870 | 15286× | resolve: &mut Resolver, |
| 871 | 15286× | ) -> Result<Change, ObjectError> { |
| 872 | 15286× | let m = map(v, "change")?; |
| 873 | 15286× | let version = uint(field(m, 0, "version")?, "version")?; |
| 874 | 15286× | if !(1..=CHANGE_VERSION).contains(&version) { |
| 875 | 1× | return schema(format!("unsupported change format version {version}")); |
| 876 | 15285× | } |
| 877 | 15285× | let change_id = ChangeId::from_slice(bytes(field(m, 1, "change_id")?, "change_id")?) |
| 878 | 15285× | .ok_or_else(|| ObjectError::Schema("change_id: expected 16 bytes".into()))?; |
| 879 | 15285× | let deps = array(field(m, 2, "deps")?, "deps")? |
| 880 | 15285× | .iter() |
| 881 | 15285× | .map(|d| oref(d, "dep")) |
| 882 | 15285× | .collect::<Result<Vec<_>, _>>()?; |
| 883 | 15285× | if !deps.windows(2).all(|w| w[0] < w[1]) { |
| 884 | 1× | return schema("deps must be sorted and unique"); |
| 885 | 15284× | } |
| 886 | 15284× | if deps.iter().any(|d| d.is_zero()) { |
| 887 | 0× | return schema("deps may not contain the self reference"); |
| 888 | 15284× | } |
| 889 | 15284× | let extra = map(field(m, 9, "extra")?, "extra")?; |
| 890 | 15284× | if !extra.is_empty() { |
| 891 | 1× | return schema("extra must be empty in format version 1"); |
| 892 | 15283× | } |
| 893 | | Ok(Change { |
| 894 | 15283× | version, |
| 895 | 15283× | change_id, |
| 896 | 15283× | deps, |
| 897 | 15283× | author: oref(field(m, 3, "author")?, "author")?, |
| 898 | 15283× | hlc: Hlc::from_value(field(m, 4, "hlc")?)?, |
| 899 | 15283× | message: text(field(m, 5, "message")?, "message")?.to_owned(), |
| 900 | 15283× | tree_ops: array(field(m, 6, "tree_ops")?, "tree_ops")? |
| 901 | 15283× | .iter() |
| 902 | 15283× | .map(TreeOp::from_value) |
| 903 | 15283× | .collect::<Result<Vec<_>, _>>()?, |
| 904 | 15283× | edits: array(field(m, 7, "edits")?, "edits")? |
| 905 | 15283× | .iter() |
| 906 | 15283× | .map(|e| Edit::from_value(e, version, resolve)) |
| 907 | 15283× | .collect::<Result<Vec<_>, _>>()?, |
| 908 | 15280× | kind: ChangeKind::from_tag(uint(field(m, 8, "kind")?, "kind")?)?, |
| 909 | | }) |
| 910 | 15286× | } |
| 911 | | } |
| 912 | |
|
| 913 | |
|
| 914 | |
|
| 915 | |
|
| 916 | |
|
| 917 | | pub type Resolver<'a> = dyn FnMut(&ObjectRef, u32, TextForm) -> Result<Lines, ObjectError> + 'a; |
| 918 | |
|
| 919 | |
|
| 920 | |
|
| 921 | | type TextSink<'a> = dyn FnMut(Vec<StoredText>) -> bool + 'a; |
| 922 | |
|
| 923 | |
|
| 924 | |
|
| 925 | | #[derive(Clone, Copy, Debug, PartialEq, Eq)] |
| 926 | | pub enum TextRef { |
| 927 | | |
| 928 | | Blob(ObjectRef, u32, TextForm), |
| 929 | | |
| 930 | | |
| 931 | | Inline { |
| 932 | | change: ChangeHash, |
| 933 | | start: u32, |
| 934 | | end: u32, |
| 935 | | }, |
| 936 | | } |
| 937 | |
|
| 938 | |
|
| 939 | |
|
| 940 | 30915× | pub fn inline_lines(array: &[u8]) -> Result<Lines, ObjectError> { |
| 941 | 30915× | let v = cbor::decode(array)?; |
| 942 | 30915× | let lines = array_lines(&v)?; |
| 943 | 30915× | if lines.is_empty() { |
| 944 | 0× | return schema("insert with no lines"); |
| 945 | 30915× | } |
| 946 | 30915× | Ok(lines) |
| 947 | 30915× | } |
| 948 | |
|
| 949 | |
|
| 950 | 34842× | fn array_lines(v: &Value) -> Result<Lines, ObjectError> { |
| 951 | 34842× | Ok(array(v, "lines")? |
| 952 | 34842× | .iter() |
| 953 | 57201× | .map(|l| bytes(l, "line")) |
| 954 | 34842× | .collect::<Result<Vec<&[u8]>, _>>()? |
| 955 | 34842× | .into_iter() |
| 956 | 34842× | .collect()) |
| 957 | 34842× | } |
| 958 | |
|
| 959 | | impl Change { |
| 960 | | |
| 961 | | |
| 962 | | |
| 963 | | |
| 964 | | |
| 965 | 7164× | pub fn decode_skeleton( |
| 966 | 7164× | bytes: &[u8], |
| 967 | 7164× | me: ChangeHash, |
| 968 | 7164× | ) -> Result<(Change, Vec<TextRef>), ObjectError> { |
| 969 | 7164× | let mut c = cbor::Cursor::new(bytes); |
| 970 | 7164× | let (major, n) = c.head()?; |
| 971 | 7164× | if major != 5 { |
| 972 | 0× | return schema("change: expected map"); |
| 973 | 7164× | } |
| 974 | 7164× | let mut fields = Map::new(); |
| 975 | 7164× | let mut edits = Vec::new(); |
| 976 | 7164× | let mut texts = Vec::new(); |
| 977 | 7164× | let mut version = None; |
| 978 | 7164× | let mut last: Option<u64> = None; |
| 979 | 7164× | for _ in 0..n { |
| 980 | 71638× | let k = map_key(&mut c, &mut last)?; |
| 981 | 71638× | if k == 7 { |
| 982 | 7164× | let Some(v) = version else { |
| 983 | 0× | return schema("change: version must come first"); |
| 984 | | }; |
| 985 | 7164× | let (am, an) = c.head()?; |
| 986 | 7164× | if am != 4 { |
| 987 | 0× | return schema("edits: expected array"); |
| 988 | 7164× | } |
| 989 | 7164× | for _ in 0..an { |
| 990 | 47203× | edits.push(edit_skeleton(&mut c, v, me, &mut texts)?); |
| 991 | | } |
| 992 | 7163× | fields.insert(7, Value::Array(Vec::new())); |
| 993 | | } else { |
| 994 | 64474× | let val = c.value()?; |
| 995 | 64474× | if k == 0 { |
| 996 | 7164× | version = Some(uint(&val, "version")?); |
| 997 | 57310× | } |
| 998 | 64474× | fields.insert(k, val); |
| 999 | | } |
| 1000 | | } |
| 1001 | 7163× | if !c.at_end() { |
| 1002 | 0× | return Err(ObjectError::Cbor(cbor::CborError::TrailingBytes)); |
| 1003 | 7163× | } |
| 1004 | 7163× | let mut change = Change::from_value(&Value::Map(fields))?; |
| 1005 | 7163× | change.edits = edits; |
| 1006 | 7163× | Ok((change, texts)) |
| 1007 | 7164× | } |
| 1008 | | } |
| 1009 | |
|
| 1010 | | impl Change { |
| 1011 | | |
| 1012 | | |
| 1013 | | |
| 1014 | 2880× | pub fn decode_meta(bytes: &[u8]) -> Result<crate::state::ChangeMeta, ObjectError> { |
| 1015 | 2880× | let mut c = cbor::Cursor::new(bytes); |
| 1016 | 2880× | let (major, n) = c.head()?; |
| 1017 | 2880× | if major != 5 { |
| 1018 | 0× | return schema("change: expected map"); |
| 1019 | 2880× | } |
| 1020 | 2880× | let mut fields = Map::new(); |
| 1021 | 2880× | let mut last: Option<u64> = None; |
| 1022 | 2880× | for _ in 0..n { |
| 1023 | 28800× | let k = map_key(&mut c, &mut last)?; |
| 1024 | 28800× | if k == 6 || k == 7 { |
| 1025 | 5760× | c.skip()?; |
| 1026 | 5760× | fields.insert(k, Value::Array(Vec::new())); |
| 1027 | | } else { |
| 1028 | 23040× | fields.insert(k, c.value()?); |
| 1029 | | } |
| 1030 | | } |
| 1031 | 2880× | if !c.at_end() { |
| 1032 | 0× | return Err(ObjectError::Cbor(cbor::CborError::TrailingBytes)); |
| 1033 | 2880× | } |
| 1034 | 2880× | Ok(crate::state::ChangeMeta::of(&Change::from_value( |
| 1035 | 2880× | &Value::Map(fields), |
| 1036 | 0× | )?)) |
| 1037 | 2880× | } |
| 1038 | | } |
| 1039 | |
|
| 1040 | |
|
| 1041 | 332254× | fn map_key(c: &mut cbor::Cursor, last: &mut Option<u64>) -> Result<u64, ObjectError> { |
| 1042 | 332254× | let (km, k) = c.head()?; |
| 1043 | 332254× | if km != 0 || last.is_some_and(|l| k <= l) { |
| 1044 | 0× | return Err(ObjectError::Cbor(cbor::CborError::BadMapKey)); |
| 1045 | 332254× | } |
| 1046 | 332254× | *last = Some(k); |
| 1047 | 332254× | Ok(k) |
| 1048 | 332254× | } |
| 1049 | |
|
| 1050 | |
|
| 1051 | 47203× | fn edit_skeleton( |
| 1052 | 47203× | c: &mut cbor::Cursor, |
| 1053 | 47203× | version: u64, |
| 1054 | 47203× | me: ChangeHash, |
| 1055 | 47203× | texts: &mut Vec<TextRef>, |
| 1056 | 47203× | ) -> Result<Edit, ObjectError> { |
| 1057 | 47203× | let (major, n) = c.head()?; |
| 1058 | 47203× | if major != 5 { |
| 1059 | 0× | return schema("edit: expected map"); |
| 1060 | 47203× | } |
| 1061 | 47203× | let mut fields = Map::new(); |
| 1062 | 47203× | let mut inline: Option<(usize, usize, u64)> = None; |
| 1063 | 47203× | let mut last: Option<u64> = None; |
| 1064 | 47203× | for _ in 0..n { |
| 1065 | 231816× | let k = map_key(c, &mut last)?; |
| 1066 | 231816× | if k == 4 { |
| 1067 | 44635× | let start = c.pos(); |
| 1068 | 44635× | let (am, count) = c.head()?; |
| 1069 | 44635× | if am != 4 { |
| 1070 | 0× | return schema("lines: expected array"); |
| 1071 | 44635× | } |
| 1072 | 44635× | for _ in 0..count { |
| 1073 | 59919× | c.skip()?; |
| 1074 | | } |
| 1075 | 44635× | inline = Some((start, c.pos(), count)); |
| 1076 | 44635× | continue; |
| 1077 | 187181× | } |
| 1078 | 187181× | fields.insert(k, c.value()?); |
| 1079 | | } |
| 1080 | 47203× | let Some((start, end, count)) = inline else { |
| 1081 | 2568× | return Edit::from_value(&Value::Map(fields), version, &mut |text, count, form| { |
| 1082 | 120× | texts.push(TextRef::Blob(*text, count, form)); |
| 1083 | 120× | Ok(Lines::default()) |
| 1084 | 120× | }); |
| 1085 | | }; |
| 1086 | 44635× | if uint(field(&fields, 0, "edit tag")?, "edit tag")? != 0 || fields.contains_key(&5) { |
| 1087 | 0× | return schema("only an insert has inline lines, and not with a text blob"); |
| 1088 | 44635× | } |
| 1089 | 44635× | if count == 0 { |
| 1090 | 0× | return schema("insert with no lines"); |
| 1091 | 44635× | } |
| 1092 | 44635× | if version >= 2 && end - start >= TEXT_BLOB_MIN { |
| 1093 | 1× | return schema("an insert this large is stored as a text blob in a version 2 change"); |
| 1094 | 44634× | } |
| 1095 | 44634× | let (Ok(start), Ok(end)) = (u32::try_from(start), u32::try_from(end)) else { |
| 1096 | 0× | return schema("change too large"); |
| 1097 | | }; |
| 1098 | 44634× | texts.push(TextRef::Inline { |
| 1099 | 44634× | change: me, |
| 1100 | 44634× | start, |
| 1101 | 44634× | end, |
| 1102 | 44634× | }); |
| 1103 | | Ok(Edit::Insert { |
| 1104 | 44634× | node: node_id(field(&fields, 1, "edit.node")?, "edit.node")?, |
| 1105 | 44634× | after: AtomRef::from_value(field(&fields, 2, "after")?, "after")?, |
| 1106 | 44634× | before: AtomRef::from_value(field(&fields, 3, "before")?, "before")?, |
| 1107 | 44634× | lines: Lines::default(), |
| 1108 | | }) |
| 1109 | 47203× | } |
| 1110 | |
|
| 1111 | |
|
| 1112 | |
|
| 1113 | | #[derive(Debug, Clone, PartialEq, Eq)] |
| 1114 | | pub struct StoredText { |
| 1115 | | pub blob: Blob, |
| 1116 | | pub content: Vec<u8>, |
| 1117 | | pub form: TextForm, |
| 1118 | | } |
| 1119 | |
|
| 1120 | |
|
| 1121 | 70× | fn text_form(version: u64, lines: &Lines) -> TextForm { |
| 1122 | 69× | match version { |
| 1123 | 70× | ..=2 => TextForm::Cbor, |
| 1124 | 69× | _ if lines.is_split_form() => TextForm::Raw, |
| 1125 | 1× | _ => TextForm::CborFallback, |
| 1126 | | } |
| 1127 | 70× | } |
| 1128 | |
|
| 1129 | |
|
| 1130 | |
|
| 1131 | |
|
| 1132 | 188× | pub fn text_lines(content: &[u8], count: u32, form: TextForm) -> Result<Lines, ObjectError> { |
| 1133 | 188× | let lines = match form { |
| 1134 | 182× | TextForm::Raw => Lines::split(content), |
| 1135 | 6× | TextForm::Cbor | TextForm::CborFallback => array_lines(&cbor::decode(content)?)?, |
| 1136 | | }; |
| 1137 | | |
| 1138 | | |
| 1139 | 188× | if cbor::byte_strings_len(&lines) < TEXT_BLOB_MIN { |
| 1140 | 2× | return schema(format!( |
| 1141 | | "a text blob of {} bytes is short enough to be inline", |
| 1142 | 2× | content.len() |
| 1143 | | )); |
| 1144 | 186× | } |
| 1145 | 186× | if form == TextForm::CborFallback && lines.is_split_form() { |
| 1146 | 1× | return schema("these lines are stored as their bytes in a version 3 change"); |
| 1147 | 185× | } |
| 1148 | 185× | if lines.len() != count as usize { |
| 1149 | 2× | return schema(format!( |
| 1150 | | "a text blob holds {} lines, the insert says {count}", |
| 1151 | 2× | lines.len() |
| 1152 | | )); |
| 1153 | 183× | } |
| 1154 | 183× | Ok(lines) |
| 1155 | 188× | } |
| 1156 | |
|
| 1157 | |
|
| 1158 | |
|
| 1159 | |
|
| 1160 | | #[derive(Debug, Clone, PartialEq, Eq)] |
| 1161 | | pub struct Blob { |
| 1162 | | pub length: u64, |
| 1163 | | pub chunks: Vec<ObjectRef>, |
| 1164 | | } |
| 1165 | |
|
| 1166 | | impl Blob { |
| 1167 | | |
| 1168 | | |
| 1169 | 452× | pub fn of(content: &[u8]) -> Blob { |
| 1170 | | Blob { |
| 1171 | 452× | length: content.len() as u64, |
| 1172 | 452× | chunks: crate::chunk::chunks(content) |
| 1173 | 452× | .into_iter() |
| 1174 | 555× | .map(|c| hash_object(ObjectType::Chunk, c)) |
| 1175 | 452× | .collect(), |
| 1176 | | } |
| 1177 | 452× | } |
| 1178 | |
|
| 1179 | 574× | pub fn encode(&self) -> Vec<u8> { |
| 1180 | 574× | cbor::encode(&Value::map([ |
| 1181 | 574× | (0, Value::Uint(FORMAT_VERSION)), |
| 1182 | 574× | (1, Value::Uint(self.length)), |
| 1183 | | ( |
| 1184 | | 2, |
| 1185 | | Value::Array( |
| 1186 | 574× | self.chunks |
| 1187 | 574× | .iter() |
| 1188 | 776× | .map(|c| Value::Bytes(c.0.to_vec())) |
| 1189 | 574× | .collect(), |
| 1190 | | ), |
| 1191 | | ), |
| 1192 | | ])) |
| 1193 | 574× | } |
| 1194 | 373× | pub fn hash(&self) -> ObjectRef { |
| 1195 | 373× | hash_object(ObjectType::Blob, &self.encode()) |
| 1196 | 373× | } |
| 1197 | 464× | pub fn decode(bytes: &[u8]) -> Result<Blob, ObjectError> { |
| 1198 | 464× | let v = cbor::decode(bytes)?; |
| 1199 | 464× | let m = map(&v, "blob")?; |
| 1200 | 464× | check_version(m)?; |
| 1201 | | Ok(Blob { |
| 1202 | 464× | length: uint(field(m, 1, "length")?, "length")?, |
| 1203 | 464× | chunks: array(field(m, 2, "chunks")?, "chunks")? |
| 1204 | 464× | .iter() |
| 1205 | 693× | .map(|c| oref(c, "chunk")) |
| 1206 | 464× | .collect::<Result<Vec<_>, _>>()?, |
| 1207 | | }) |
| 1208 | 464× | } |
| 1209 | | } |
| 1210 | |
|
| 1211 | |
|
| 1212 | |
|
| 1213 | |
|
| 1214 | |
|
| 1215 | | #[derive(Debug, Clone, PartialEq, Eq)] |
| 1216 | | pub struct Checkpoint { |
| 1217 | | pub heads: Vec<ChangeHash>, |
| 1218 | | pub count: u64, |
| 1219 | | pub payload: ObjectRef, |
| 1220 | | } |
| 1221 | |
|
| 1222 | | impl Checkpoint { |
| 1223 | 12× | pub fn encode(&self) -> Vec<u8> { |
| 1224 | 12× | cbor::encode(&Value::map([ |
| 1225 | 12× | (0, Value::Uint(FORMAT_VERSION)), |
| 1226 | | ( |
| 1227 | | 1, |
| 1228 | | Value::Array( |
| 1229 | 12× | self.heads |
| 1230 | 12× | .iter() |
| 1231 | 13× | .map(|h| Value::Bytes(h.0.to_vec())) |
| 1232 | 12× | .collect(), |
| 1233 | | ), |
| 1234 | | ), |
| 1235 | 12× | (2, Value::Uint(self.count)), |
| 1236 | 12× | (3, Value::Bytes(self.payload.0.to_vec())), |
| 1237 | | ])) |
| 1238 | 12× | } |
| 1239 | 0× | pub fn hash(&self) -> ObjectRef { |
| 1240 | 0× | hash_object(ObjectType::Checkpoint, &self.encode()) |
| 1241 | 0× | } |
| 1242 | 88× | pub fn decode(bytes: &[u8]) -> Result<Checkpoint, ObjectError> { |
| 1243 | 88× | let v = cbor::decode(bytes)?; |
| 1244 | 88× | let m = map(&v, "checkpoint")?; |
| 1245 | 88× | check_version(m)?; |
| 1246 | | Ok(Checkpoint { |
| 1247 | 88× | heads: array(field(m, 1, "heads")?, "heads")? |
| 1248 | 88× | .iter() |
| 1249 | 90× | .map(|h| oref(h, "head")) |
| 1250 | 88× | .collect::<Result<Vec<_>, _>>()?, |
| 1251 | 88× | count: uint(field(m, 2, "count")?, "count")?, |
| 1252 | 88× | payload: oref(field(m, 3, "payload")?, "payload")?, |
| 1253 | | }) |
| 1254 | 88× | } |
| 1255 | | } |
| 1256 | |
|
| 1257 | |
|
| 1258 | |
|
| 1259 | |
|
| 1260 | | #[derive(Debug, Clone, PartialEq, Eq, Default)] |
| 1261 | | pub struct StateObject { |
| 1262 | | pub heads: Vec<ChangeHash>, |
| 1263 | | } |
| 1264 | |
|
| 1265 | | impl StateObject { |
| 1266 | 3079× | pub fn new(mut heads: Vec<ChangeHash>) -> StateObject { |
| 1267 | 3079× | heads.sort(); |
| 1268 | 3079× | heads.dedup(); |
| 1269 | 3079× | StateObject { heads } |
| 1270 | 3079× | } |
| 1271 | 3078× | pub fn encode(&self) -> Vec<u8> { |
| 1272 | 3078× | debug_assert!(self.heads.windows(2).all(|w| w[0] < w[1])); |
| 1273 | 3078× | cbor::encode(&Value::map([ |
| 1274 | 3078× | (0, Value::Uint(FORMAT_VERSION)), |
| 1275 | | ( |
| 1276 | | 1, |
| 1277 | | Value::Array( |
| 1278 | 3078× | self.heads |
| 1279 | 3078× | .iter() |
| 1280 | 3837× | .map(|h| Value::Bytes(h.0.to_vec())) |
| 1281 | 3078× | .collect(), |
| 1282 | | ), |
| 1283 | | ), |
| 1284 | | ])) |
| 1285 | 3078× | } |
| 1286 | 2× | pub fn hash(&self) -> ObjectRef { |
| 1287 | 2× | hash_object(ObjectType::State, &self.encode()) |
| 1288 | 2× | } |
| 1289 | 5181× | pub fn decode(bytes: &[u8]) -> Result<StateObject, ObjectError> { |
| 1290 | 5181× | let v = cbor::decode(bytes)?; |
| 1291 | 5181× | let m = map(&v, "state")?; |
| 1292 | 5181× | check_version(m)?; |
| 1293 | 5181× | let heads = array(field(m, 1, "heads")?, "heads")? |
| 1294 | 5181× | .iter() |
| 1295 | 8484× | .map(|h| oref(h, "head")) |
| 1296 | 5181× | .collect::<Result<Vec<_>, _>>()?; |
| 1297 | 5181× | if !heads.windows(2).all(|w| w[0] < w[1]) { |
| 1298 | 0× | return schema("state heads must be sorted and unique"); |
| 1299 | 5181× | } |
| 1300 | 5181× | Ok(StateObject { heads }) |
| 1301 | 5181× | } |
| 1302 | | } |
| 1303 | |
|
| 1304 | | #[cfg(test)] |
| 1305 | | pub(crate) mod tests { |
| 1306 | | use super::*; |
| 1307 | |
|
| 1308 | 16× | pub(crate) fn sample_change() -> Change { |
| 1309 | 16× | let n = NodeId([7; 16]); |
| 1310 | 16× | let dep = ObjectRef([1; 32]); |
| 1311 | 16× | let mut c = Change { |
| 1312 | 16× | version: crate::objects::CHANGE_VERSION, |
| 1313 | 16× | change_id: ChangeId([9; 16]), |
| 1314 | 16× | deps: vec![ObjectRef([2; 32]), dep], |
| 1315 | 16× | author: ObjectRef([3; 32]), |
| 1316 | 16× | hlc: Hlc { |
| 1317 | 16× | wall_ms: 1_700_000_000_000, |
| 1318 | 16× | logical: 4, |
| 1319 | 16× | }, |
| 1320 | 16× | message: "Add greeting".into(), |
| 1321 | 16× | tree_ops: vec![ |
| 1322 | 16× | TreeOp::AddNode { |
| 1323 | 16× | node: n, |
| 1324 | 16× | kind: NodeKind::File, |
| 1325 | 16× | mode: Mode::TEXT, |
| 1326 | 16× | parent: NodeId::ROOT, |
| 1327 | 16× | name: "hello.txt".into(), |
| 1328 | 16× | }, |
| 1329 | 16× | TreeOp::SetMode { |
| 1330 | 16× | node: n, |
| 1331 | 16× | old: OpRef { |
| 1332 | 16× | change: ObjectRef::ZERO, |
| 1333 | 16× | index: 0, |
| 1334 | 16× | }, |
| 1335 | 16× | mode: Mode { |
| 1336 | 16× | exec: true, |
| 1337 | 16× | opaque: false, |
| 1338 | 16× | }, |
| 1339 | 16× | }, |
| 1340 | 16× | ], |
| 1341 | 16× | edits: vec![ |
| 1342 | 16× | Edit::Insert { |
| 1343 | 16× | node: n, |
| 1344 | 16× | after: AtomRef::Start, |
| 1345 | 16× | before: AtomRef::End, |
| 1346 | 16× | lines: vec![b"hello\n".to_vec(), b"world\n".to_vec()].into(), |
| 1347 | 16× | }, |
| 1348 | 16× | Edit::Delete { |
| 1349 | 16× | node: n, |
| 1350 | 16× | atoms: vec![AtomRange { |
| 1351 | 16× | change: dep, |
| 1352 | 16× | from: 0, |
| 1353 | 16× | to: 2, |
| 1354 | 16× | }], |
| 1355 | 16× | }, |
| 1356 | 16× | Edit::Replace { |
| 1357 | 16× | node: n, |
| 1358 | 16× | old: None, |
| 1359 | 16× | new: Some(ObjectRef([5; 32])), |
| 1360 | 16× | }, |
| 1361 | 16× | ], |
| 1362 | 16× | kind: ChangeKind::Normal, |
| 1363 | 16× | }; |
| 1364 | 16× | c.normalize(); |
| 1365 | 16× | c |
| 1366 | 16× | } |
| 1367 | |
|
| 1368 | | #[test] |
| 1369 | 1× | fn change_round_trips_and_hash_is_stable() { |
| 1370 | 1× | let c = sample_change(); |
| 1371 | 1× | let bytes = c.encode(); |
| 1372 | 1× | let d = Change::decode(&bytes).unwrap(); |
| 1373 | 1× | assert_eq!(c, d); |
| 1374 | 1× | assert_eq!(c.hash(), d.hash()); |
| 1375 | 1× | assert_eq!(d.encode(), bytes); |
| 1376 | | |
| 1377 | 1× | let mut e = c.clone(); |
| 1378 | 1× | e.message.push('!'); |
| 1379 | 1× | assert_ne!(e.hash(), c.hash()); |
| 1380 | 1× | } |
| 1381 | |
|
| 1382 | | #[test] |
| 1383 | 1× | fn compact_encoding_is_the_plain_one() { |
| 1384 | | |
| 1385 | | |
| 1386 | 1× | let c = sample_change(); |
| 1387 | 1× | assert!(!c.has_stored_text(), "the sample's insert is inline"); |
| 1388 | 1× | let plain = cbor::encode(&c.to_value()); |
| 1389 | 1× | assert_eq!(c.encode(), plain); |
| 1390 | 1× | assert_eq!(c.encode_parts().0, plain); |
| 1391 | 1× | let streamed: Result<Vec<u8>, ()> = c.encode_streaming(|_| Ok(())); |
| 1392 | 1× | assert_eq!(streamed.unwrap(), plain); |
| 1393 | 1× | assert_eq!(Change::decode(&plain).unwrap(), c); |
| 1394 | 1× | } |
| 1395 | |
|
| 1396 | | #[test] |
| 1397 | 1× | fn decode_rejects_bad_changes() { |
| 1398 | 1× | let c = sample_change(); |
| 1399 | 1× | let mut v = c.to_value(); |
| 1400 | 1× | if let Value::Map(m) = &mut v { |
| 1401 | 1× | m.insert(0, Value::Uint(4)); |
| 1402 | 1× | } |
| 1403 | 1× | assert!( |
| 1404 | 1× | matches!(Change::from_value(&v), Err(ObjectError::Schema(s)) if s.contains("version")) |
| 1405 | | ); |
| 1406 | |
|
| 1407 | 1× | let mut v = c.to_value(); |
| 1408 | 1× | if let Value::Map(m) = &mut v { |
| 1409 | 1× | m.insert( |
| 1410 | 1× | 2, |
| 1411 | 1× | Value::Array(vec![Value::Bytes(vec![2; 32]), Value::Bytes(vec![1; 32])]), |
| 1412 | 1× | ); |
| 1413 | 1× | } |
| 1414 | 1× | assert!( |
| 1415 | 1× | matches!(Change::from_value(&v), Err(ObjectError::Schema(s)) if s.contains("sorted")) |
| 1416 | | ); |
| 1417 | |
|
| 1418 | 1× | let mut v = c.to_value(); |
| 1419 | 1× | if let Value::Map(m) = &mut v { |
| 1420 | 1× | m.insert(9, Value::map([(0, Value::Null)])); |
| 1421 | 1× | } |
| 1422 | 1× | assert!( |
| 1423 | 1× | matches!(Change::from_value(&v), Err(ObjectError::Schema(s)) if s.contains("extra")) |
| 1424 | | ); |
| 1425 | 1× | } |
| 1426 | |
|
| 1427 | 6× | fn big_lines(n: usize) -> Lines { |
| 1428 | 6× | (0..n) |
| 1429 | 3000× | .map(|i| format!("line {i} of a large insert\n").into_bytes()) |
| 1430 | 6× | .collect() |
| 1431 | 6× | } |
| 1432 | |
|
| 1433 | 6× | fn with_insert(mut c: Change, lines: Lines) -> Change { |
| 1434 | 6× | c.edits[0] = Edit::Insert { |
| 1435 | 6× | node: NodeId([7; 16]), |
| 1436 | 6× | after: AtomRef::Start, |
| 1437 | 6× | before: AtomRef::End, |
| 1438 | 6× | lines, |
| 1439 | 6× | }; |
| 1440 | 6× | c |
| 1441 | 6× | } |
| 1442 | |
|
| 1443 | | #[test] |
| 1444 | 1× | fn large_inserts_are_stored_as_text_blobs() { |
| 1445 | 1× | assert!( |
| 1446 | 1× | !sample_change().has_stored_text(), |
| 1447 | | "a small insert stays inline" |
| 1448 | | ); |
| 1449 | 1× | let lines = big_lines(500); |
| 1450 | 1× | let c = with_insert(sample_change(), lines.clone()); |
| 1451 | 1× | assert!(c.has_stored_text()); |
| 1452 | 1× | let (bytes, texts) = c.encode_parts(); |
| 1453 | 1× | assert_eq!(bytes, c.encode()); |
| 1454 | 1× | assert!(bytes.len() < 1024, "{}", bytes.len()); |
| 1455 | 1× | assert_eq!(texts.len(), 1); |
| 1456 | | |
| 1457 | 1× | assert_eq!(texts[0].form, TextForm::Raw); |
| 1458 | 1× | assert_eq!(texts[0].content, lines.bytes()); |
| 1459 | 1× | assert_eq!(texts[0].blob, Blob::of(&texts[0].content)); |
| 1460 | 1× | assert!(Change::decode(&bytes).is_err(), "reading it needs the text"); |
| 1461 | 1× | let d = Change::decode_with(&bytes, &mut |r, n, form| { |
| 1462 | 1× | assert_eq!(*r, texts[0].blob.hash()); |
| 1463 | 1× | assert_eq!(form, TextForm::Raw); |
| 1464 | 1× | text_lines(&texts[0].content, n, form) |
| 1465 | 1× | }) |
| 1466 | 1× | .unwrap(); |
| 1467 | 1× | assert_eq!(d, c); |
| 1468 | 1× | assert_eq!(d.hash(), c.hash()); |
| 1469 | | |
| 1470 | 1× | let mut v2 = c.clone(); |
| 1471 | 1× | v2.version = 2; |
| 1472 | 1× | let (b2, t2) = v2.encode_parts(); |
| 1473 | 1× | assert_eq!(t2[0].form, TextForm::Cbor); |
| 1474 | 1× | assert_eq!(t2[0].content, cbor::encode_byte_strings(&lines)); |
| 1475 | 1× | let d2 = Change::decode_with(&b2, &mut |_, n, form| text_lines(&t2[0].content, n, form)) |
| 1476 | 1× | .unwrap(); |
| 1477 | 1× | assert_eq!(d2, v2); |
| 1478 | | |
| 1479 | | |
| 1480 | 1× | let odd: Lines = (0..500) |
| 1481 | 500× | .map(|i| format!("no newline {i}").into_bytes()) |
| 1482 | 1× | .collect(); |
| 1483 | 1× | let c3 = with_insert(sample_change(), odd.clone()); |
| 1484 | 1× | let (b3, t3) = c3.encode_parts(); |
| 1485 | 1× | assert_eq!(t3[0].form, TextForm::CborFallback); |
| 1486 | 1× | let d3 = Change::decode_with(&b3, &mut |_, n, form| text_lines(&t3[0].content, n, form)) |
| 1487 | 1× | .unwrap(); |
| 1488 | 1× | assert_eq!(d3, c3); |
| 1489 | 1× | let marked = cbor::encode_byte_strings(&lines); |
| 1490 | 1× | assert!(text_lines(&marked, 500, TextForm::CborFallback).is_err()); |
| 1491 | 1× | } |
| 1492 | |
|
| 1493 | | #[test] |
| 1494 | 1× | fn version_1_changes_keep_their_inline_inserts_and_hashes() { |
| 1495 | 1× | let mut c = with_insert(sample_change(), big_lines(500)); |
| 1496 | 1× | c.version = 1; |
| 1497 | 1× | let bytes = c.encode(); |
| 1498 | 1× | assert!(bytes.len() > TEXT_BLOB_MIN); |
| 1499 | 1× | let (again, texts) = c.encode_parts(); |
| 1500 | 1× | assert!(texts.is_empty()); |
| 1501 | 1× | assert_eq!(again, bytes); |
| 1502 | 1× | let d = Change::decode(&bytes).unwrap(); |
| 1503 | 1× | assert_eq!(d.version, 1); |
| 1504 | 1× | assert_eq!(d.encode(), bytes); |
| 1505 | | |
| 1506 | 1× | let mut v = c.to_value(); |
| 1507 | 1× | if let Value::Map(m) = &mut v { |
| 1508 | 1× | m.insert(0, Value::Uint(2)); |
| 1509 | 1× | } |
| 1510 | 1× | assert!( |
| 1511 | 1× | matches!(Change::from_value(&v), Err(ObjectError::Schema(s)) if s.contains("text blob")) |
| 1512 | | ); |
| 1513 | 1× | } |
| 1514 | |
|
| 1515 | | #[test] |
| 1516 | 1× | fn a_skeleton_leaves_every_insert_out_and_says_where_its_lines_are() { |
| 1517 | 1× | let mut c = with_insert(sample_change(), big_lines(500)); |
| 1518 | 1× | c.edits.push(Edit::Insert { |
| 1519 | 1× | node: NodeId([8; 16]), |
| 1520 | 1× | after: AtomRef::Start, |
| 1521 | 1× | before: AtomRef::End, |
| 1522 | 1× | lines: vec![b"small\n".to_vec(), b"insert\n".to_vec()].into(), |
| 1523 | 1× | }); |
| 1524 | 1× | let (bytes, stored) = c.encode_parts(); |
| 1525 | 1× | let me = ObjectRef([4; 32]); |
| 1526 | 1× | let (skel, texts) = Change::decode_skeleton(&bytes, me).unwrap(); |
| 1527 | 1× | assert_eq!(texts.len(), 2); |
| 1528 | 1× | let mut filled = skel.clone(); |
| 1529 | 1× | let mut next = texts.iter(); |
| 1530 | 4× | for e in filled.edits.iter_mut() { |
| 1531 | 4× | if let Edit::Insert { lines, .. } = e { |
| 1532 | 2× | assert!(lines.is_empty(), "no line kept"); |
| 1533 | 2× | *lines = match next.next().unwrap() { |
| 1534 | 1× | TextRef::Blob(r, n, form) => { |
| 1535 | 1× | assert_eq!(*r, stored[0].blob.hash()); |
| 1536 | 1× | text_lines(&stored[0].content, *n, *form).unwrap() |
| 1537 | | } |
| 1538 | 1× | TextRef::Inline { change, start, end } => { |
| 1539 | 1× | assert_eq!(*change, me); |
| 1540 | 1× | inline_lines(&bytes[*start as usize..*end as usize]).unwrap() |
| 1541 | | } |
| 1542 | | }; |
| 1543 | 2× | } |
| 1544 | | } |
| 1545 | 1× | assert_eq!(filled, c, "filled in, it is the change"); |
| 1546 | | |
| 1547 | 1× | let mut v1 = with_insert(sample_change(), big_lines(500)); |
| 1548 | 1× | v1.version = 1; |
| 1549 | 1× | assert!(Change::decode_skeleton(&v1.encode(), me).is_ok()); |
| 1550 | 1× | let mut v = v1.to_value(); |
| 1551 | 1× | if let Value::Map(m) = &mut v { |
| 1552 | 1× | m.insert(0, Value::Uint(2)); |
| 1553 | 1× | } |
| 1554 | 1× | assert!(Change::decode_skeleton(&cbor::encode(&v), me).is_err()); |
| 1555 | 1× | } |
| 1556 | |
|
| 1557 | | #[test] |
| 1558 | 1× | fn metadata_reads_without_the_operations() { |
| 1559 | 1× | let c = with_insert(sample_change(), big_lines(500)); |
| 1560 | 1× | let (bytes, _) = c.encode_parts(); |
| 1561 | 1× | let m = Change::decode_meta(&bytes).unwrap(); |
| 1562 | 1× | assert_eq!(m, crate::state::ChangeMeta::of(&c)); |
| 1563 | 1× | } |
| 1564 | |
|
| 1565 | | #[test] |
| 1566 | 1× | fn text_blobs_are_checked() { |
| 1567 | 1× | let lines = big_lines(500); |
| 1568 | 1× | let content = cbor::encode_byte_strings(&lines); |
| 1569 | 1× | assert_eq!(content.len(), cbor::byte_strings_len(&lines)); |
| 1570 | 1× | assert_eq!( |
| 1571 | | content, |
| 1572 | 1× | cbor::encode(&Value::Array( |
| 1573 | 500× | lines.iter().map(|l| Value::Bytes(l.to_vec())).collect() |
| 1574 | | )) |
| 1575 | | ); |
| 1576 | 1× | assert_eq!(text_lines(&content, 500, TextForm::Cbor).unwrap(), lines); |
| 1577 | 1× | assert!(text_lines(&content, 499, TextForm::Cbor).is_err()); |
| 1578 | 1× | assert_eq!( |
| 1579 | 1× | text_lines(lines.bytes(), 500, TextForm::Raw).unwrap(), |
| 1580 | | lines |
| 1581 | | ); |
| 1582 | 1× | assert!(text_lines(lines.bytes(), 501, TextForm::Raw).is_err()); |
| 1583 | 1× | let small = cbor::encode_byte_strings(&vec![b"x\n".to_vec()].into()); |
| 1584 | 1× | assert!( |
| 1585 | 1× | text_lines(&small, 1, TextForm::Cbor).is_err(), |
| 1586 | | "short enough to be inline" |
| 1587 | | ); |
| 1588 | 1× | assert!( |
| 1589 | 1× | text_lines(b"x\n", 1, TextForm::Raw).is_err(), |
| 1590 | | "short enough to be inline" |
| 1591 | | ); |
| 1592 | 1× | } |
| 1593 | |
|
| 1594 | | #[test] |
| 1595 | 1× | fn names_are_validated() { |
| 1596 | 5× | for bad in ["", ".", "..", "a/b", "a\0b"] { |
| 1597 | 5× | assert!(validate_name(bad).is_err(), "{bad:?}"); |
| 1598 | | } |
| 1599 | 1× | assert!(validate_name("Foo.c").is_ok()); |
| 1600 | 1× | assert!(validate_name("...").is_ok()); |
| 1601 | 1× | } |
| 1602 | |
|
| 1603 | | #[test] |
| 1604 | 1× | fn mode_and_kind_tags() { |
| 1605 | 3× | for m in [ |
| 1606 | 1× | Mode::TEXT, |
| 1607 | 1× | Mode::OPAQUE, |
| 1608 | 1× | Mode { |
| 1609 | 1× | exec: true, |
| 1610 | 1× | opaque: true, |
| 1611 | 1× | }, |
| 1612 | 1× | ] { |
| 1613 | 3× | assert_eq!(Mode::from_bits(m.bits()).unwrap(), m); |
| 1614 | | } |
| 1615 | 1× | assert!(Mode::from_bits(4).is_err()); |
| 1616 | 4× | for k in [ |
| 1617 | 1× | NodeKind::File, |
| 1618 | 1× | NodeKind::Dir, |
| 1619 | 1× | NodeKind::Symlink, |
| 1620 | 1× | NodeKind::Subrepo, |
| 1621 | 1× | ] { |
| 1622 | 4× | assert_eq!(NodeKind::from_tag(k.tag()).unwrap(), k); |
| 1623 | | } |
| 1624 | 1× | } |
| 1625 | |
|
| 1626 | | #[test] |
| 1627 | 1× | fn self_references_resolve() { |
| 1628 | 1× | let me = ObjectRef([8; 32]); |
| 1629 | 1× | assert_eq!( |
| 1630 | 1× | AtomRef::Atom { |
| 1631 | 1× | change: ObjectRef::ZERO, |
| 1632 | 1× | index: 3 |
| 1633 | 1× | } |
| 1634 | 1× | .resolve(me), |
| 1635 | 1× | AtomRef::Atom { |
| 1636 | 1× | change: me, |
| 1637 | 1× | index: 3 |
| 1638 | 1× | } |
| 1639 | | ); |
| 1640 | 1× | assert_eq!(AtomRef::Start.resolve(me), AtomRef::Start); |
| 1641 | 1× | let other = ObjectRef([1; 32]); |
| 1642 | 1× | assert_eq!( |
| 1643 | 1× | OpRef { |
| 1644 | 1× | change: other, |
| 1645 | 1× | index: 1 |
| 1646 | 1× | } |
| 1647 | 1× | .resolve(me) |
| 1648 | | .change, |
| 1649 | | other |
| 1650 | | ); |
| 1651 | 1× | } |
| 1652 | |
|
| 1653 | | #[test] |
| 1654 | 1× | fn state_object_round_trip() { |
| 1655 | 1× | let st = StateObject::new(vec![ |
| 1656 | 1× | ObjectRef([2; 32]), |
| 1657 | 1× | ObjectRef([1; 32]), |
| 1658 | 1× | ObjectRef([2; 32]), |
| 1659 | | ]); |
| 1660 | 1× | assert_eq!(st.heads, vec![ObjectRef([1; 32]), ObjectRef([2; 32])]); |
| 1661 | 1× | assert_eq!(StateObject::decode(&st.encode()).unwrap(), st); |
| 1662 | 1× | assert_eq!(StateObject::new(vec![]).heads, vec![]); |
| 1663 | 1× | assert_ne!(StateObject::new(vec![]).hash(), st.hash()); |
| 1664 | 1× | } |
| 1665 | |
|
| 1666 | | #[test] |
| 1667 | 1× | fn blob_round_trip() { |
| 1668 | 1× | let b = Blob { |
| 1669 | 1× | length: 10, |
| 1670 | 1× | chunks: vec![ObjectRef([1; 32]), ObjectRef([2; 32])], |
| 1671 | 1× | }; |
| 1672 | 1× | assert_eq!(Blob::decode(&b.encode()).unwrap(), b); |
| 1673 | 1× | } |
| 1674 | | } |