openapi-client-generator 0.1.2

Generates a client library based on an OpenAPI spec.
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

#![allow(non_snake_case)]
#![allow(unused)]

use std::fs;
use std::fs::File;
use std::path::{Path, PathBuf};
use proc_macro2::TokenStream;
use anyhow::Result;
use derive_builder::Builder;
use openapiv3::OpenAPI;
use tokio::io::AsyncWriteExt;
use crate::codegen::client::generate_lib_rs;
use crate::codegen::format::format_code;
use crate::codegen::model::generate_model_rs;
use crate::util::open;
use std::io::Write;

pub mod codegen;
mod util;

#[derive(Builder)]
pub struct GenerateLibrary {
    pub name: String,
    pub yaml_path: PathBuf,
    pub dest_path: PathBuf,

    pub lib_rs_path: Option<PathBuf>,
    pub model_rs_path: Option<PathBuf>,
}


fn write_file(path: &Path, code: TokenStream, template: &str) -> Result<()> {
    let code = format_code(code)?;
    let mut f = open(&path)?;
    f.write(template.as_bytes())?;
    f.write("\n".as_bytes())?;
    f.write(code.as_bytes())?;
    Ok(())
}

pub fn generate_library(opts: GenerateLibrary) -> Result<()> {
    fs::create_dir_all(&opts.dest_path)?;

    let spec: OpenAPI = {
        let file = File::open(opts.yaml_path)?;
        serde_yaml::from_reader(file)?
    };

    let lib_rs_template = match opts.lib_rs_path {
        Some(path) => fs::read_to_string(path)?,
        None => include_str!("../template/lib.rs").to_string(),
    };
    let code = generate_lib_rs(&spec, &opts.name);
    write_file(&opts.dest_path.join("lib.rs"), code, &lib_rs_template)?;

    let lib_rs_template = match opts.model_rs_path {
        Some(path) => fs::read_to_string(path)?,
        None => include_str!("../template/model.rs").to_string(),
    };
    let code = generate_model_rs(&spec);
    write_file(&opts.dest_path.join("model.rs"), code, &lib_rs_template)?;

    Ok(())
}