feat(builtin): len, #441

src/modules/builtin/len.rs

@@ -0,0 +1,52 @@
+use heraclitus_compiler::prelude::*;
+use crate::modules::expression::expr::Expr;
+use crate::translate::module::TranslateModule;
+use crate::docs::module::DocumentationModule;
+use crate::modules::types::{Type, Typed};
+use crate::utils::{ParserMetadata, TranslateMetadata};
+
+#[derive(Debug, Clone)]
+pub struct Len {
+    value: Box<Expr>,
+}
+
+impl SyntaxModule<ParserMetadata> for Len {
+    syntax_name!("Length");
+
+    fn new() -> Self {
+        Len {
+            value: Box::new(Expr::new()),
+        }
+    }
+
+    fn parse(&mut self, meta: &mut ParserMetadata) -> SyntaxResult {
+        token(meta, "len")?;
+        syntax(meta, &mut *self.value)?;
+
+        Ok(())
+    }
+}
+
+impl TranslateModule for Len {
+    fn translate(&self, meta: &mut TranslateMetadata) -> String {
+        let path_type = self.value.get_type();
+        let value = self.value.translate(meta);
+        if path_type == Type::Text {
+            format!("echo \"${{#{}}}\"", value).trim_end().to_string()
+        } else {
+            format!("echo \"${{#{}[@]}}\"", value).trim_end().to_string()
+        }
+    }
+}
+
+impl DocumentationModule for Len {
+    fn document(&self, _meta: &ParserMetadata) -> String {
+        "".to_string()
+    }
+}
+
+impl Typed for Len {
+    fn get_type(&self) -> Type {
+        Type::Num
+    }
+}

src/modules/builtin/mod.rs

@@ -3,3 +3,4 @@ pub mod echo;
 pub mod mv;
 pub mod nameof;
 pub mod exit;
+pub mod len;

src/modules/expression/expr.rs

@@ -46,6 +46,7 @@ use crate::modules::variable::get::VariableGet;
 use super::ternop::ternary::Ternary;
 use crate::modules::function::invocation::FunctionInvocation;
 use crate::modules::builtin::nameof::Nameof;
+use crate::modules::builtin::len::Len;
 use crate::{document_expression, parse_expr, parse_expr_group, translate_expression};
 
 #[derive(Debug, Clone)]
@@ -80,6 +81,7 @@ pub enum ExprType {
     Status(Status),
     Nameof(Nameof),
     Is(Is),
+    Len(Len),
 }
 
 #[derive(Debug, Clone, Default)]
@@ -145,6 +147,8 @@ impl SyntaxModule<ParserMetadata> for Expr {
             types @ TypeOp => [ Is, Cast ],
             unops @ UnOp => [ Neg, Not ],
             literals @ Literal => [
+                // Builtins
+                Len,
                 // Literals
                 Parentheses, Bool, Number, Text,
                 Array, Null, Nameof, Status,
@@ -162,6 +166,8 @@ impl SyntaxModule<ParserMetadata> for Expr {
 impl TranslateModule for Expr {
     fn translate(&self, meta: &mut TranslateMetadata) -> String {
         translate_expression!(meta, self.value.as_ref().unwrap(), [
+            // Builtin
+            Len,
             // Ternary conditional
             Ternary,
             // Logical operators
@@ -187,6 +193,8 @@ impl TranslateModule for Expr {
 impl DocumentationModule for Expr {
     fn document(&self, meta: &ParserMetadata) -> String {
         document_expression!(meta, self.value.as_ref().unwrap(), [
+            // Builtin
+            Len,
             // Ternary conditional
             Ternary,
             // Logical operators

src/modules/statement/stmt.rs

@@ -39,6 +39,7 @@ use crate::modules::builtin::{
     mv::Mv,
     cd::Cd,
     exit::Exit,
+    len::Len,
 };
 use super::comment_doc::CommentDoc;
 use super::comment::Comment;
@@ -68,6 +69,7 @@ pub enum StatementType {
     Echo(Echo),
     Mv(Mv),
     Exit(Exit),
+    Len(Len),
     CommandModifier(CommandModifier),
     Comment(Comment),
     CommentDoc(CommentDoc),
@@ -95,7 +97,7 @@ impl Statement {
         ShorthandMul, ShorthandDiv,
         ShorthandModulo,
         // Command
-        CommandModifier, Echo, Mv, Cd, Exit,
+        CommandModifier, Echo, Mv, Cd, Exit, Len,
         // Comment doc
         CommentDoc, Comment,
         // Expression

src/modules/variable/mod.rs

@@ -32,7 +32,7 @@ pub fn variable_name_keywords() -> Vec<&'static str> {
         // Command Modifiers
         "silent", "unsafe",
         // Misc
-        "echo", "status", "nameof", "mv", "cd", "exit",
+        "echo", "status", "nameof", "mv", "cd", "exit", "len"
     ]
 }
 

src/std/text.ab

@@ -94,6 +94,7 @@ pub fun chars(text: Text): [Text] {
 /// Gets the length of provided text or array.
 #[allow_absurd_cast]
 pub fun len(value): Num {
+    echo "The len stdlib is deprecated, use the builtin!"
     unsafe {
         if value is Text:
             return $echo "\$\{#{nameof value}}"$ as Num
@@ -140,7 +141,7 @@ pub fun ends_with(text: Text, suffix: Text): Bool {
 /// If `length` is provided, the substring will include `length` characters; otherwise, it slices to the end of `text`.
 /// If `length` is negative, an empty string is returned.
 pub fun slice(text: Text, index: Num, length: Num = 0): Text {
-    if length == 0: length = len(text) - index
+    if length == 0: length = len text - index
     if length <= 0: return ""
     return unsafe $printf "%.{length}s" "\$\{text:{index}}"$
 }
@@ -159,16 +160,18 @@ pub fun capitalize(text: Text): Text {
 
 /// Pads `text` with the specified `pad` character on left until it reaches the desired `length`.
 pub fun lpad(text: Text, pad: Text, length: Num): Text {
-    if length <= len(text): return text
-    length = len(text) - length
+    if length <= len text: return text
+    length = len text
+    length -= length
     pad = unsafe $printf "%{length}s" "" | tr " " "{pad}"$
     return pad + text
 }
 
 /// Pads `text` with the specified `pad` character on the right until it reaches the desired `length`.
 pub fun rpad(text: Text, pad: Text, length: Num): Text {
     if length <= len(text): return text
-    length = len(text) - length
+    length = len text
+    length -= length
     pad = unsafe $printf "%{length}s" "" | tr " " "{pad}"$
     return text + pad
 }

src/tests/validity/len_list.ab

@@ -0,0 +1,6 @@
+// Output
+// 4
+
+main {
+    echo len [1, 2, 3, 4]
+} 

src/tests/validity/len_string.ab

@@ -0,0 +1,6 @@
+// Output
+// 5
+
+main {
+    echo len "hello"
+}