Implement str2nr()

Author: J-Fields

src/vimscript/expression/evaluate.ts

@@ -1193,7 +1193,16 @@ export class EvaluationContext {
         }
         return list(result.map(int));
       }
-      // TODO: str2nr()
+      case 'str2nr': {
+        const [s, _base] = getArgs(1, 2);
+        const base = _base ? toInt(_base) : 10
+        if (![2, 8, 10, 16].includes(base)) {
+          throw VimError.fromCode(ErrorCode.InvalidArgument474);
+        }
+        // TODO: Skip prefixes like 0x
+        const parsed = Number.parseInt(toString(s!), base)
+        return int(isNaN(parsed) ? 0 : parsed);
+      }
       // TODO: stridx()
       case 'string': {
         const [x] = getArgs(1);

test/vimscript/expression.test.ts

@@ -733,6 +733,16 @@ suite('Vimscript expressions', () => {
       exprTest('str2list("á")', { value: list([int(97), int(769)]) });
     });
 
+    suite('str2nr', () => {
+      exprTest('str2nr("123")', { value: int(123) });
+      exprTest('str2nr("1001010110", 2)', { value: int(598) });
+      exprTest('str2nr("123", 8)', { value: int(83) });
+      exprTest('str2nr("123", 10)', { value: int(123) });
+      exprTest('str2nr("DEADBEEF", 16)', { value: int(3735928559) });
+      exprTest('str2nr("DEADBEEF", 10)', { value: int(0) });
+      exprTest('str2nr("DEADBEEF", 9)', { error: ErrorCode.InvalidArgument474 });
+    });
+
     suite('string', () => {
       exprTest('string("")', { value: str('') });
       exprTest('string(123)', { value: str('123') });