[Proposal] Add Namespace Source Literals

docs/source-literals.md

@@ -142,6 +142,50 @@ function Human_Versus_Zombie_Eat()
 end function
 ```
 
+## SOURCE_NAMESPACE
+
+This works the same as SOURCE_FUNCTION_NAME except that it does not include the function name.
+
+The SOURCE_NAMESPACE can be a useful string for debug logs and logging filters.
+
+```BrighterScript
+namespace Human.Versus.Zombie
+    function Eat()
+        print SOURCE_NAMESPACE
+    end function
+end namespace
+```
+
+transpiles to:
+
+```BrightScript
+function Human_Versus_Zombie_Eat()
+    print "Human.Versus.Zombie"
+end function
+```
+
+## SOURCE_NAMESPACE_ROOT
+
+This works the same as SOURCE_NAMESPACE except that it only includes the lowest-level namespace.
+
+The SOURCE_NAMESPACE_ROOT can be a useful string for debug logs and logging filters.
+
+```BrighterScript
+namespace Human.Versus.Zombie
+    function Eat()
+        print SOURCE_NAMESPACE_ROOT
+    end function
+end namespace
+```
+
+transpiles to:
+
+```BrightScript
+function Human_Versus_Zombie_Eat()
+    print "Human"
+end function
+```
+
 ## SOURCE_LOCATION
 A combination of SOURCE_FILE_PATH and SOURCE_LINE_NUM.
 

src/astUtils/creators.ts

@@ -74,6 +74,8 @@ const tokenDefaults = {
     [TokenKind.Semicolon]: ';',
     [TokenKind.SourceFilePathLiteral]: 'SOURCE_FILE_PATH',
     [TokenKind.SourceFunctionNameLiteral]: 'SOURCE_FUNCTION_NAME',
+    [TokenKind.SourceNamespaceRootLiteral]: 'SOURCE_NAMESPACE_ROOT',
+    [TokenKind.SourceNamespaceLiteral]: 'SOURCE_NAMESPACE',
     [TokenKind.SourceLineNumLiteral]: 'SOURCE_LINE_NUM',
     [TokenKind.SourceLocationLiteral]: 'SOURCE_LOCATION',
     [TokenKind.Star]: '*',

src/lexer/Lexer.spec.ts

@@ -1332,13 +1332,15 @@ describe('lexer', () => {
     });
 
     it('identifies brighterscript source literals', () => {
-        let { tokens } = Lexer.scan('LINE_NUM SOURCE_FILE_PATH SOURCE_LINE_NUM FUNCTION_NAME SOURCE_FUNCTION_NAME SOURCE_LOCATION PKG_PATH PKG_LOCATION');
+        let { tokens } = Lexer.scan('LINE_NUM SOURCE_FILE_PATH SOURCE_LINE_NUM FUNCTION_NAME SOURCE_FUNCTION_NAME SOURCE_NAMESPACE SOURCE_NAMESPACE_ROOT SOURCE_LOCATION PKG_PATH PKG_LOCATION');
         expect(tokens.map(x => x.kind)).to.eql([
             TokenKind.LineNumLiteral,
             TokenKind.SourceFilePathLiteral,
             TokenKind.SourceLineNumLiteral,
             TokenKind.FunctionNameLiteral,
             TokenKind.SourceFunctionNameLiteral,
+            TokenKind.SourceNamespaceLiteral,
+            TokenKind.SourceNamespaceRootLiteral,
             TokenKind.SourceLocationLiteral,
             TokenKind.PkgPathLiteral,
             TokenKind.PkgLocationLiteral,

src/lexer/TokenKind.ts

@@ -170,6 +170,8 @@ export enum TokenKind {
     SourceLineNumLiteral = 'SourceLineNumLiteral',
     FunctionNameLiteral = 'FunctionNameLiteral',
     SourceFunctionNameLiteral = 'SourceFunctionNameLiteral',
+    SourceNamespaceLiteral = 'SourceNamespaceLiteral',
+    SourceNamespaceRootLiteral = 'SourceNamespaceRootLiteral',
     SourceLocationLiteral = 'SourceLocationLiteral',
     PkgPathLiteral = 'PkgPathLiteral',
     PkgLocationLiteral = 'PkgLocationLiteral',
@@ -312,6 +314,8 @@ export const Keywords: Record<string, TokenKind> = {
     'source_line_num': TokenKind.SourceLineNumLiteral,
     'function_name': TokenKind.FunctionNameLiteral,
     'source_function_name': TokenKind.SourceFunctionNameLiteral,
+    'source_namespace': TokenKind.SourceNamespaceLiteral,
+    'source_namespace_root': TokenKind.SourceNamespaceRootLiteral,
     'source_location': TokenKind.SourceLocationLiteral,
     'pkg_path': TokenKind.PkgPathLiteral,
     'pkg_location': TokenKind.PkgLocationLiteral,
@@ -438,6 +442,8 @@ export const AllowedProperties = [
     TokenKind.SourceLineNumLiteral,
     TokenKind.FunctionNameLiteral,
     TokenKind.SourceFunctionNameLiteral,
+    TokenKind.SourceNamespaceLiteral,
+    TokenKind.SourceNamespaceRootLiteral,
     TokenKind.SourceLocationLiteral,
     TokenKind.PkgPathLiteral,
     TokenKind.PkgLocationLiteral,
@@ -490,6 +496,8 @@ export const BrighterScriptSourceLiterals = [
     TokenKind.SourceLineNumLiteral,
     TokenKind.FunctionNameLiteral,
     TokenKind.SourceFunctionNameLiteral,
+    TokenKind.SourceNamespaceLiteral,
+    TokenKind.SourceNamespaceRootLiteral,
     TokenKind.SourceLocationLiteral,
     TokenKind.PkgPathLiteral,
     TokenKind.PkgLocationLiteral
@@ -546,6 +554,8 @@ export const DisallowedLocalIdentifiers = [
     TokenKind.SourceLineNumLiteral,
     TokenKind.FunctionNameLiteral,
     TokenKind.SourceFunctionNameLiteral,
+    TokenKind.SourceNamespaceLiteral,
+    TokenKind.SourceNamespaceRootLiteral,
     TokenKind.SourceLocationLiteral,
     TokenKind.PkgPathLiteral,
     TokenKind.PkgLocationLiteral,

src/parser/Expression.ts

@@ -1163,6 +1163,19 @@ export class SourceLiteralExpression extends Expression {
             case TokenKind.SourceFunctionNameLiteral:
                 text = `"${this.getFunctionName(state, ParseMode.BrighterScript)}"`;
                 break;
+            case TokenKind.SourceNamespaceLiteral:
+                let namespaceParts = this.getFunctionName(state, ParseMode.BrighterScript).split('.');
+                namespaceParts.pop(); // remove the function name
+
+                text = `"${namespaceParts.join('.')}"`;
+                break;
+            case TokenKind.SourceNamespaceRootLiteral:
+                let namespaceRootParts = this.getFunctionName(state, ParseMode.BrighterScript).split('.');
+                namespaceRootParts.pop(); // remove the function name
+
+                let rootNamespace = namespaceRootParts.shift() ?? '';
+                text = `"${rootNamespace}"`;
+                break;
             case TokenKind.SourceLocationLiteral:
                 const locationUrl = fileUrl(state.srcPath);
                 //TODO find first parent that has range, or default to -1