Merge pull request #26 from zakhenry/feat/suppot-plantuml

feat(Language Support): Add Plant UML .puml support
zakhenry · Jul 13, 2019 · b2fa8b0 · b2fa8b0
2 parents b400f05 + 0c47eee
commit b2fa8b0
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 0 deletions.
diff --git a/src/embedme.lib.ts b/src/embedme.lib.ts
@@ -68,13 +68,15 @@ enum SupportedFileType {
   KOTLIN = 'kotlin',
   SCALA = 'scala',
   CRYSTAL = 'cr',
+  PLANT_UML = 'puml',
 }
 
 enum CommentFamily {
   NONE, // some languages do not support comments, e.g. JSON
   C,
   XML,
   HASH,
+  SINGLE_QUOTE,
 }
 
 const languageMap: Record<CommentFamily, SupportedFileType[]> = {
@@ -106,6 +108,7 @@ const languageMap: Record<CommentFamily, SupportedFileType[]> = {
     SupportedFileType.RUBY,
     SupportedFileType.CRYSTAL,
   ],
+  [CommentFamily.SINGLE_QUOTE]: [SupportedFileType.PLANT_UML],
 };
 
 const filetypeCommentReaders: Record<CommentFamily, FilenameFromCommentReader> = {
@@ -132,6 +135,14 @@ const filetypeCommentReaders: Record<CommentFamily, FilenameFromCommentReader> =
       return null;
     }
 
+    return match[1];
+  },
+  [CommentFamily.SINGLE_QUOTE]: line => {
+    const match = line.match(/'\s*?(\S*?)$/);
+    if (!match) {
+      return null;
+    }
+
     return match[1];
   },
 };

diff --git a/test/fixture.md b/test/fixture.md
@@ -290,6 +290,20 @@ object HelloWorld {
 
 ```
 
+Plant UML
+
+```puml
+' sample.puml
+
+@startuml
+Bob -> Alice : hello
+activate Alice
+Alice -> Alice : some action
+return bye
+@enduml
+
+```
+
 ## Extension-less selection
 
 ```sh

diff --git a/test/sample.puml b/test/sample.puml
@@ -0,0 +1,6 @@
+@startuml
+Bob -> Alice : hello
+activate Alice
+Alice -> Alice : some action
+return bye
+@enduml