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'},
+ );
+ });
+ });
+}