feat: More directives support (#15)

* feat: part directive

* feat: library directive

* feat: library augmentation

* chore: update README

* feat: support augmentation import

* feat(test): part and part-of

* feat(test): library

* chore: format

* chore: remove augment test

* chore: format

README.md

@@ -35,7 +35,7 @@ compilationUnit = {
 
 directive = {
   // What kind of directive this is
-  "kind": "import" | "export",
+  "kind": "import" | "export" | "part" | "part-of" | "library",
 
   // The name of the file being imported or exported
   "uri": string,
@@ -46,6 +46,13 @@ directive = {
 
   // For "import" statements, the ID after the `as` keyword
   "as": string,
+
+  // For "library" statements, the name of the library
+  // For "part-of" statements, the name of the library that the file is a part of
+  "name": string,
+
+  // Whether the directive references an augmentation library
+  "augmentation": bool,
 }
 
 declaration = {

lib/src/compilation_unit.dart

@@ -6,7 +6,9 @@ import 'package:dartdoc_json/src/extension_declaration.dart';
 import 'package:dartdoc_json/src/function_declaration.dart';
 import 'package:dartdoc_json/src/generic_type_alias.dart';
 import 'package:dartdoc_json/src/import_directive.dart';
+import 'package:dartdoc_json/src/library_directive.dart';
 import 'package:dartdoc_json/src/mixin_declaration.dart';
+import 'package:dartdoc_json/src/part_directive.dart';
 import 'package:dartdoc_json/src/top_level_variable_declaration.dart';
 
 Map<String, dynamic> serializeCompilationUnit(CompilationUnit unit) {
@@ -41,11 +43,18 @@ Map<String, dynamic> serializeCompilationUnit(CompilationUnit unit) {
     late Map<String, dynamic> serialized;
     if (directive is ImportDirective) {
       serialized = serializeImportDirective(directive);
+    } else if (directive is AugmentationImportDirective) {
+      serialized = serializeAugmentationImportDirective(directive);
     } else if (directive is ExportDirective) {
       serialized = serializeExportDirective(directive);
-      // } else if (directive is LibraryDirective) {
-      // } else if (directive is PartDirective) {
-      // } else if (directive is PartOfDirective) {
+    } else if (directive is LibraryDirective) {
+      serialized = serializeLibraryDirective(directive);
+    } else if (directive is LibraryAugmentationDirective) {
+      serialized = serializeLibraryAugmentationDirective(directive);
+    } else if (directive is PartDirective) {
+      serialized = serializePartDirective(directive);
+    } else if (directive is PartOfDirective) {
+      serialized = serializePartOfDirective(directive);
     } else {
       throw AssertionError('Unknown directive type: ${directive.runtimeType}');
     }

lib/src/import_directive.dart

@@ -14,3 +14,13 @@ Map<String, dynamic> serializeImportDirective(ImportDirective import_) {
     'hide': hides.isEmpty ? null : hides,
   });
 }
+
+Map<String, dynamic> serializeAugmentationImportDirective(
+  AugmentationImportDirective import_,
+) {
+  return filterMap(<String, dynamic>{
+    'kind': 'import',
+    'uri': import_.uri.stringValue,
+    'augmentation': true,
+  });
+}

lib/src/library_directive.dart

@@ -0,0 +1,19 @@
+import 'package:analyzer/dart/ast/ast.dart';
+import 'package:dartdoc_json/src/utils.dart';
+
+Map<String, dynamic> serializeLibraryDirective(LibraryDirective library) {
+  return filterMap(<String, dynamic>{
+    'kind': 'library',
+    'name': library.name2?.name,
+  });
+}
+
+Map<String, dynamic> serializeLibraryAugmentationDirective(
+  LibraryAugmentationDirective libraryAgumentation,
+) {
+  return filterMap(<String, dynamic>{
+    'kind': 'library',
+    'augmentation': true,
+    'uri': libraryAgumentation.uri,
+  });
+}

lib/src/part_directive.dart

@@ -0,0 +1,17 @@
+import 'package:analyzer/dart/ast/ast.dart';
+import 'package:dartdoc_json/src/utils.dart';
+
+Map<String, dynamic> serializePartDirective(PartDirective part) {
+  return filterMap(<String, dynamic>{
+    'kind': 'part',
+    'uri': part.uri.stringValue,
+  });
+}
+
+Map<String, dynamic> serializePartOfDirective(PartOfDirective partOf) {
+  return filterMap(<String, dynamic>{
+    'kind': 'part-of',
+    'uri': partOf.uri?.stringValue,
+    'name': partOf.libraryName?.name,
+  });
+}

test/library_directive_test.dart

@@ -0,0 +1,14 @@
+import 'package:test/test.dart';
+
+import 'utils.dart';
+
+void main() {
+  group('LibraryDirective', () {
+    test('simple library', () {
+      expect(
+        parseAsJson('library foo;'),
+        {'kind': 'library', 'name': 'foo'},
+      );
+    });
+  });
+}

test/part_directive_test.dart

@@ -0,0 +1,45 @@
+import 'package:test/test.dart';
+
+import 'utils.dart';
+
+void main() {
+  group('PartDirective', () {
+    test('simple part', () {
+      expect(
+        parseAsJson('part "foo.dart";'),
+        {'kind': 'part', 'uri': 'foo.dart'},
+      );
+    });
+
+    test('multiple part', () {
+      expect(
+        parseAsJson('''
+          part 'foo.dart';
+          part 'boo.dart';
+        '''),
+        [
+          {'kind': 'part', 'uri': 'foo.dart'},
+          {'kind': 'part', 'uri': 'boo.dart'},
+        ],
+      );
+    });
+  });
+
+  group('PartOfDirective', () {
+    test('simple part', () {
+      expect(
+        parseAsJson('part of "foo.dart";'),
+        {'kind': 'part-of', 'uri': 'foo.dart'},
+      );
+    });
+
+    test('part of library', () {
+      expect(
+        parseAsJson('''
+          part of foo;
+        '''),
+        {'kind': 'part-of', 'name': 'foo'},
+      );
+    });
+  });
+}