198 lines
7.2 KiB
Plaintext
198 lines
7.2 KiB
Plaintext
![]() |
#[cfg(test)]
|
||
|
mod test_examples_my_func {
|
||
|
#[tokio::test(flavor = "multi_thread", worker_threads = 5)]
|
||
|
async fn serial_test_example_my_func0() {
|
||
|
let user_agent = concat!(env!("CARGO_PKG_NAME"), ".rs/", env!("CARGO_PKG_VERSION"),);
|
||
|
let http_client = reqwest::Client::builder()
|
||
|
.user_agent(user_agent)
|
||
|
.timeout(std::time::Duration::from_secs(600))
|
||
|
.connect_timeout(std::time::Duration::from_secs(60));
|
||
|
let ws_client = reqwest::Client::builder()
|
||
|
.user_agent(user_agent)
|
||
|
.timeout(std::time::Duration::from_secs(600))
|
||
|
.connect_timeout(std::time::Duration::from_secs(60))
|
||
|
.connection_verbose(true)
|
||
|
.tcp_keepalive(std::time::Duration::from_secs(600))
|
||
|
.http1_only();
|
||
|
let token = std::env::var("KITTYCAD_API_TOKEN").expect("KITTYCAD_API_TOKEN not set");
|
||
|
let client = kittycad::Client::new_from_reqwest(token, http_client, ws_client);
|
||
|
let ws = client
|
||
|
.modeling()
|
||
|
.commands_ws(None, None, None, None, None, Some(false))
|
||
|
.await
|
||
|
.unwrap();
|
||
|
let tokens = crate::token::lexer("This is another code block.\nyes sirrr.\nmyFunc");
|
||
|
let parser = crate::parser::Parser::new(tokens);
|
||
|
let program = parser.ast().unwrap();
|
||
|
let mut mem: crate::executor::ProgramMemory = Default::default();
|
||
|
let ctx = crate::executor::ExecutorContext {
|
||
|
engine: std::sync::Arc::new(Box::new(
|
||
|
crate::engine::conn::EngineConnection::new(ws)
|
||
|
.await
|
||
|
.unwrap(),
|
||
|
)),
|
||
|
fs: crate::fs::FileManager::new(),
|
||
|
stdlib: std::sync::Arc::new(crate::std::StdLib::new()),
|
||
|
units: kittycad::types::UnitLength::Mm,
|
||
|
};
|
||
|
crate::executor::execute(program, &mut mem, crate::executor::BodyType::Root, &ctx)
|
||
|
.await
|
||
|
.unwrap();
|
||
|
}
|
||
|
|
||
|
#[tokio::test(flavor = "multi_thread", worker_threads = 5)]
|
||
|
async fn serial_test_example_my_func1() {
|
||
|
let user_agent = concat!(env!("CARGO_PKG_NAME"), ".rs/", env!("CARGO_PKG_VERSION"),);
|
||
|
let http_client = reqwest::Client::builder()
|
||
|
.user_agent(user_agent)
|
||
|
.timeout(std::time::Duration::from_secs(600))
|
||
|
.connect_timeout(std::time::Duration::from_secs(60));
|
||
|
let ws_client = reqwest::Client::builder()
|
||
|
.user_agent(user_agent)
|
||
|
.timeout(std::time::Duration::from_secs(600))
|
||
|
.connect_timeout(std::time::Duration::from_secs(60))
|
||
|
.connection_verbose(true)
|
||
|
.tcp_keepalive(std::time::Duration::from_secs(600))
|
||
|
.http1_only();
|
||
|
let token = std::env::var("KITTYCAD_API_TOKEN").expect("KITTYCAD_API_TOKEN not set");
|
||
|
let client = kittycad::Client::new_from_reqwest(token, http_client, ws_client);
|
||
|
let ws = client
|
||
|
.modeling()
|
||
|
.commands_ws(None, None, None, None, None, Some(false))
|
||
|
.await
|
||
|
.unwrap();
|
||
|
let tokens = crate::token::lexer("This is code.\nIt does other shit.\nmyFunc");
|
||
|
let parser = crate::parser::Parser::new(tokens);
|
||
|
let program = parser.ast().unwrap();
|
||
|
let mut mem: crate::executor::ProgramMemory = Default::default();
|
||
|
let ctx = crate::executor::ExecutorContext {
|
||
|
engine: std::sync::Arc::new(Box::new(
|
||
|
crate::engine::conn::EngineConnection::new(ws)
|
||
|
.await
|
||
|
.unwrap(),
|
||
|
)),
|
||
|
fs: crate::fs::FileManager::new(),
|
||
|
stdlib: std::sync::Arc::new(crate::std::StdLib::new()),
|
||
|
units: kittycad::types::UnitLength::Mm,
|
||
|
};
|
||
|
crate::executor::execute(program, &mut mem, crate::executor::BodyType::Root, &ctx)
|
||
|
.await
|
||
|
.unwrap();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[allow(non_camel_case_types, missing_docs)]
|
||
|
#[doc = "Std lib function: myFunc\nThis is some function.\nIt does shit."]
|
||
|
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, schemars :: JsonSchema, ts_rs :: TS)]
|
||
|
#[ts(export)]
|
||
|
pub(crate) struct MyFunc {}
|
||
|
|
||
|
#[allow(non_upper_case_globals, missing_docs)]
|
||
|
#[doc = "Std lib function: myFunc\nThis is some function.\nIt does shit."]
|
||
|
pub(crate) const MyFunc: MyFunc = MyFunc {};
|
||
|
fn boxed_my_func(
|
||
|
args: crate::std::Args,
|
||
|
) -> std::pin::Pin<
|
||
|
Box<
|
||
|
dyn std::future::Future<
|
||
|
Output = anyhow::Result<crate::executor::MemoryItem, crate::errors::KclError>,
|
||
|
>,
|
||
|
>,
|
||
|
> {
|
||
|
Box::pin(my_func(args))
|
||
|
}
|
||
|
|
||
|
impl crate::docs::StdLibFn for MyFunc {
|
||
|
fn name(&self) -> String {
|
||
|
"myFunc".to_string()
|
||
|
}
|
||
|
|
||
|
fn summary(&self) -> String {
|
||
|
"This is some function.".to_string()
|
||
|
}
|
||
|
|
||
|
fn description(&self) -> String {
|
||
|
"It does shit.".to_string()
|
||
|
}
|
||
|
|
||
|
fn tags(&self) -> Vec<String> {
|
||
|
vec![]
|
||
|
}
|
||
|
|
||
|
fn args(&self) -> Vec<crate::docs::StdLibFnArg> {
|
||
|
let mut settings = schemars::gen::SchemaSettings::openapi3();
|
||
|
settings.inline_subschemas = true;
|
||
|
let mut generator = schemars::gen::SchemaGenerator::new(settings);
|
||
|
vec![crate::docs::StdLibFnArg {
|
||
|
name: "args".to_string(),
|
||
|
type_: "kittycad::types::InputFormat".to_string(),
|
||
|
schema: <Option<kittycad::types::InputFormat>>::json_schema(&mut generator),
|
||
|
required: false,
|
||
|
}]
|
||
|
}
|
||
|
|
||
|
fn return_value(&self) -> Option<crate::docs::StdLibFnArg> {
|
||
|
let mut settings = schemars::gen::SchemaSettings::openapi3();
|
||
|
settings.inline_subschemas = true;
|
||
|
let mut generator = schemars::gen::SchemaGenerator::new(settings);
|
||
|
Some(crate::docs::StdLibFnArg {
|
||
|
name: "".to_string(),
|
||
|
type_: "[SketchGroup]".to_string(),
|
||
|
schema: <Vec<SketchGroup>>::json_schema(&mut generator),
|
||
|
required: true,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
fn unpublished(&self) -> bool {
|
||
|
false
|
||
|
}
|
||
|
|
||
|
fn deprecated(&self) -> bool {
|
||
|
false
|
||
|
}
|
||
|
|
||
|
fn examples(&self) -> Vec<String> {
|
||
|
let code_blocks = vec![
|
||
|
"This is another code block.\nyes sirrr.\nmyFunc",
|
||
|
"This is code.\nIt does other shit.\nmyFunc",
|
||
|
];
|
||
|
code_blocks
|
||
|
.iter()
|
||
|
.map(|cb| {
|
||
|
let tokens = crate::token::lexer(cb);
|
||
|
let parser = crate::parser::Parser::new(tokens);
|
||
|
let program = parser.ast().unwrap();
|
||
|
let mut options: crate::ast::types::FormatOptions = Default::default();
|
||
|
options.insert_final_newline = false;
|
||
|
program.recast(&options, 0)
|
||
|
})
|
||
|
.collect::<Vec<String>>()
|
||
|
}
|
||
|
|
||
|
fn std_lib_fn(&self) -> crate::std::StdFn {
|
||
|
boxed_my_func
|
||
|
}
|
||
|
|
||
|
fn clone_box(&self) -> Box<dyn crate::docs::StdLibFn> {
|
||
|
Box::new(self.clone())
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[doc = r" This is some function."]
|
||
|
#[doc = r" It does shit."]
|
||
|
#[doc = r""]
|
||
|
#[doc = r" This is code."]
|
||
|
#[doc = r" It does other shit."]
|
||
|
#[doc = r" myFunc"]
|
||
|
#[doc = r""]
|
||
|
#[doc = r" ```"]
|
||
|
#[doc = r" This is another code block."]
|
||
|
#[doc = r" yes sirrr."]
|
||
|
#[doc = r" myFunc"]
|
||
|
#[doc = r" ```"]
|
||
|
fn inner_my_func(
|
||
|
#[doc = r" The args to do shit to."] args: Option<kittycad::types::InputFormat>,
|
||
|
) -> Result<Vec<Box<SketchGroup>>> {
|
||
|
args
|
||
|
}
|