aboutsummaryrefslogtreecommitdiff
path: root/macros/src/syntax/parse/software_task.rs
diff options
context:
space:
mode:
Diffstat (limited to 'macros/src/syntax/parse/software_task.rs')
-rw-r--r--macros/src/syntax/parse/software_task.rs86
1 files changed, 86 insertions, 0 deletions
diff --git a/macros/src/syntax/parse/software_task.rs b/macros/src/syntax/parse/software_task.rs
new file mode 100644
index 00000000..2b1ac4a5
--- /dev/null
+++ b/macros/src/syntax/parse/software_task.rs
@@ -0,0 +1,86 @@
+use syn::{parse, ForeignItemFn, ItemFn, Stmt};
+
+use crate::syntax::parse::util::FilterAttrs;
+use crate::syntax::{
+ ast::{SoftwareTask, SoftwareTaskArgs},
+ parse::util,
+};
+
+impl SoftwareTask {
+ pub(crate) fn parse(args: SoftwareTaskArgs, item: ItemFn) -> parse::Result<Self> {
+ let valid_signature =
+ util::check_fn_signature(&item, true) && util::type_is_unit(&item.sig.output);
+
+ let span = item.sig.ident.span();
+
+ let name = item.sig.ident.to_string();
+
+ let is_async = item.sig.asyncness.is_some();
+
+ if valid_signature {
+ if let Some((context, Ok(inputs))) = util::parse_inputs(item.sig.inputs, &name) {
+ let FilterAttrs { cfgs, attrs, .. } = util::filter_attributes(item.attrs);
+
+ return Ok(SoftwareTask {
+ args,
+ attrs,
+ cfgs,
+ context,
+ inputs,
+ stmts: item.block.stmts,
+ is_extern: false,
+ is_async,
+ });
+ }
+ }
+
+ Err(parse::Error::new(
+ span,
+ &format!(
+ "this task handler must have type signature `(async) fn({}::Context, ..)`",
+ name
+ ),
+ ))
+ }
+}
+
+impl SoftwareTask {
+ pub(crate) fn parse_foreign(
+ args: SoftwareTaskArgs,
+ item: ForeignItemFn,
+ ) -> parse::Result<Self> {
+ let valid_signature =
+ util::check_foreign_fn_signature(&item, true) && util::type_is_unit(&item.sig.output);
+
+ let span = item.sig.ident.span();
+
+ let name = item.sig.ident.to_string();
+
+ let is_async = item.sig.asyncness.is_some();
+
+ if valid_signature {
+ if let Some((context, Ok(inputs))) = util::parse_inputs(item.sig.inputs, &name) {
+ let FilterAttrs { cfgs, attrs, .. } = util::filter_attributes(item.attrs);
+
+ return Ok(SoftwareTask {
+ args,
+ attrs,
+ cfgs,
+ context,
+ inputs,
+ stmts: Vec::<Stmt>::new(),
+ is_extern: true,
+ is_async,
+ });
+ }
+ }
+
+ Err(parse::Error::new(
+ span,
+ &format!(
+ "this task handler must have type signature `(async) fn({}::Context, ..)`",
+ name
+ ),
+ ))
+ }
+}