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

use crate::PermanentMapEdits;
use serde_json::Value;

// When the PermanentMapEdits format changes, add a transformation here to automatically convert
// edits written with the old format.
//
// This problem is often solved with something like protocol buffers, but the resulting proto
// usually winds up with permanent legacy fields, unless the changes are purely additive. For
// example, protobufs wouldn't have helped with the fix_intersection_ids problem. Explicit
// transformation is easier!
pub fn upgrade(mut value: Value) -> Result<PermanentMapEdits, String> {
    // I don't remember the previous schema change before this. If someone files a bug and has an
    // older file, can add support for it then.
    fix_offset(&mut value);
    fix_intersection_ids(&mut value);

    abstutil::from_json(&value.to_string().into_bytes()).map_err(|x| x.to_string())
}

// eee179ce8a6c1e6133dc212b73c3f79b11603e82 added an offset_seconds field
fn fix_offset(value: &mut Value) {
    match value {
        Value::Array(list) => {
            for x in list {
                fix_offset(x);
            }
        }
        Value::Object(map) => {
            if map.len() == 1 && map.contains_key("TrafficSignal") {
                let ts = map
                    .get_mut("TrafficSignal")
                    .unwrap()
                    .as_object_mut()
                    .unwrap();
                if ts.get("offset_seconds").is_none() {
                    ts.insert("offset_seconds".to_string(), Value::Number(0.into()));
                }
            } else {
                for x in map.values_mut() {
                    fix_offset(x);
                }
            }
        }
        _ => {}
    }
}

// 11cefb118ab353d2e7fa5dceaab614a9b775e6ec changed { "osm_node_id": 123 } to just 123
fn fix_intersection_ids(value: &mut Value) {
    match value {
        Value::Array(list) => {
            for x in list {
                fix_intersection_ids(x);
            }
        }
        Value::Object(map) => {
            if map.len() == 1 && map.contains_key("osm_node_id") {
                *value = Value::Number(map["osm_node_id"].as_i64().unwrap().into());
            } else {
                for x in map.values_mut() {
                    fix_intersection_ids(x);
                }
            }
        }
        _ => {}
    }
}