feat(builtin): len, #amber-lang#441

src/modules/builtin/len.rs

@@ -0,0 +1,46 @@
+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: Expr,
+}
+
+impl SyntaxModule<ParserMetadata> for Len {
+    syntax_name!("Length");
+
+    fn new() -> Self {
+        Len {
+            value: 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()
+    }
+}

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/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/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"
+}