asbru 0.0.12

A Data-oriented service mesh generator
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

use std::fs;
use std::io::Write;
use std::{
    cell::{RefCell, RefMut},
    path::{Path, PathBuf},
    rc::Rc,
};

use codegen::{Function, Scope};

use crate::codegen::render::render::Render;

/// Structure to manage the main.rs generated file
pub struct MainFile {
    path: PathBuf,
    scope: Rc<RefCell<Scope>>,
    main_function: Rc<RefCell<Function>>,
}

impl MainFile {
    pub fn new<P: AsRef<Path>>(path: &P) -> Self {
        let main_scope = Rc::new(RefCell::new(Scope::new()));
        let main_function = Function::new("main")
            .set_async(true)
            .attr("tokio::main(flavor = \"multi_thread\")")
            .ret("anyhow::Result<()>")
            .doc("Asbru auto-generated project")
            .clone();

        let path = path.as_ref().to_owned();
        MainFile {
            path,
            scope: main_scope,
            main_function: Rc::new(RefCell::new(main_function)),
        }
    }

    /// Builder arround the main file
    pub fn main_scope(&self) -> RefMut<'_, Scope> {
        self.scope.borrow_mut()
    }

    /// Builder arround the main function
    pub fn main_function(&self) -> RefMut<'_, Function> {
        self.main_function.borrow_mut()
    }

    /// Finalize the main file
    pub fn finalize(&self) -> String {
        let mut scope = self.scope.borrow_mut();
        scope.push_fn(self.main_function.borrow_mut().clone());

        scope.to_string()
    }
}

impl Render for MainFile {
    // TODO: Use a shared config
    fn generate(&self) -> Result<(), crate::codegen::generate::GenericErrors> {
        let output = &self.path;
        self.main_scope().import("async_graphql", "Schema");
        self.main_scope().import("async_graphql", "EmptyMutation");
        self.main_scope()
            .import("async_graphql", "EmptySubscription");
        self.main_scope().import("warp", "Filter");
        self.main_scope().import("std", "env");
        self.main_scope().import("tower::make", "Shared");
        self.main_scope().import("tower", "ServiceBuilder");

        self.main_scope().import("domain::query", "Query");
        self.main_function().line(
            r#"
    let schema = Schema::build(Query::default(), EmptyMutation, EmptySubscription)
        .finish();

    let env_port = env::var("PORT")
        .expect("No PORT provided in env variables.");
    let env_port: u16 = env::var("PORT")
        .expect("No PORT provided in env variables.").parse::<u16>().expect("No valid PORT provided.");

    let cors = warp::cors()
        .allow_methods(vec!["POST"])
        .allow_header("content-type")
        .allow_any_origin()
        .build();

    let graphql_post = warp::post()
        .and(warp::path("graphql"))
        .and(async_graphql_warp::graphql(schema))
        .and_then(
            |(schema, request): (
                Schema<Query, EmptyMutation, EmptySubscription>,
                async_graphql::Request,
            )| async move {
                Ok::<_, std::convert::Infallible>(async_graphql_warp::Response::from(
                    schema
                        .execute(request)
                        .await,
                ))
            },
        );

    let filters = graphql_post
            .with(cors)
            .with(warp::trace::request());

    let service = ServiceBuilder::new().timeout(std::time::Duration::from_secs(10)).service(warp::service(filters));

    let service = Shared::new(service);

    let addr = std::net::SocketAddr::from(([0, 0, 0, 0], env_port));
    let listener = std::net::TcpListener::bind(addr).unwrap();

    warp::hyper::Server::from_tcp(listener).unwrap().serve(service).await?;

    Ok(())
        "#,
        );
        let content = self.finalize();

        let mut f = fs::File::create(&output)?;
        f.write_all(&(content.as_bytes()))?;
        Ok(())
    }
}