dabih 2.2.5

Command Line Interface for dabih
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

use openapiv3::{OpenAPI, ReferenceOr};
use std::path::PathBuf;

// remove multipart/form-data from  put "/upload/{mnemonic}/chunk" requestBody
// because progenitor does not support it yet
fn remove_multipart_form_data(spec: &mut OpenAPI) {
    if let Some(path_item) = spec.paths.paths.get_mut("/upload/{mnemonic}/chunk") {
        if let ReferenceOr::Item(item) = path_item {
            if let Some(put_op) = &mut item.put {
                if let Some(request_body) = &mut put_op.request_body {
                    if let ReferenceOr::Item(rb) = request_body {
                        rb.content.swap_remove("multipart/form-data");
                    }
                }
            }
        }
    }
}

// progenitor has a problem with merged types when additionalProperties is false
// so we set it to true for all object schemas
fn allow_additional_properties(spec: &mut OpenAPI) {
    if let Some(components) = &mut spec.components {
        for (_, schema) in &mut components.schemas {
            if let ReferenceOr::Item(item) = schema {
                if let openapiv3::SchemaKind::Type(openapiv3::Type::Object(obj)) =
                    &mut item.schema_kind
                {
                    obj.additional_properties = Some(openapiv3::AdditionalProperties::Any(true));
                }
            }
        }
    }
}

fn main() {
    let src = "spec.json";
    println!("cargo:rerun-if-changed={}", src);
    let file = std::fs::File::open(src).unwrap();
    let mut spec = serde_json::from_reader(file).unwrap();
    remove_multipart_form_data(&mut spec);
    allow_additional_properties(&mut spec);

    let mut generator = progenitor::Generator::default();

    let tokens = generator.generate_tokens(&spec).unwrap();
    let ast = syn::parse2(tokens).unwrap();
    let content = prettyplease::unparse(&ast);

    let mut out_file = std::path::Path::new(&std::env::var("OUT_DIR").unwrap()).to_path_buf();
    out_file.push("codegen.rs");

    let tmp_out_path = PathBuf::from("./codegen.rs");

    std::fs::write(tmp_out_path, &content).unwrap();
    std::fs::write(out_file, content).unwrap();
}