fix: supported for github's mermaid

shoumoji · shoumoji · commit 5bf2fd8fe510 · 2022-12-17T19:51:19.000+09:00
Signed-off-by: ayumin &lt;kosyu79@gmail.com&gt;
diff --git a/db/column.go b/db/column.go
@@ -31,8 +31,8 @@ func (c *Column) ToMermaid() string {
 	mermaidType := c.Type
 
 	// mermaid cannot display Type Column "()" and "unsigned"
-	mermaidType = strings.Replace(mermaidType, "(", "[", -1)
-	mermaidType = strings.Replace(mermaidType, ")", "]", -1)
+	mermaidType = strings.Replace(mermaidType, "(", "_", -1)
+	mermaidType = strings.Replace(mermaidType, ")", "", -1)
 
 	mermaidType = strings.Replace(mermaidType, " ", "_", -1)
 
diff --git a/db/schema_test.go b/db/schema_test.go
@@ -194,7 +194,7 @@ func TestSchema_ToMermaid(t *testing.T) {
 			want: `erDiagram
 
 articles {
-  integer[10]_unsigned id PK "not null"
+  integer_10_unsigned id PK "not null"
   integer user_id FK "not null"
 }
 
@@ -240,7 +240,7 @@ users ||--o{ articles : owns`,
 			want: `erDiagram
 
 articles {
-  integer[10]_unsigned id PK "not null"
+  integer_10_unsigned id PK "not null"
   integer user_id FK "not null"
 }`,
 		},
diff --git a/db/table_test.go b/db/table_test.go
@@ -266,7 +266,7 @@ func TestTable_ToMermaid(t *testing.T) {
 				showComment: true,
 			},
 			want: `articles {
-  integer[10]_unsigned id PK "not null"
+  integer_10_unsigned id PK "not null"
   integer user_id FK "not null"
   text title
 }`,
@@ -304,7 +304,7 @@ func TestTable_ToMermaid(t *testing.T) {
 				showComment: false,
 			},
 			want: `articles {
-  integer[10]_unsigned id
+  integer_10_unsigned id
   integer user_id
   text title
 }`,
