feat: allow message.getTypeUrl provide custom tyepUrlPrefix

kairlec · Jul 7, 2022 · 597ddc2 · 597ddc2
1 parent a1d6029
commit 597ddc2
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/cli/targets/static.js b/cli/targets/static.js
@@ -597,11 +597,17 @@ function buildType(ref, type) {
             "@function getTypeUrl",
             "@memberof " + exportName(type),
             "@static",
+            "@param {string|undefined} [typeUrlPrefix] your custom typeUrlPrefix(default \"type.googleapis.com\")",
             "@returns {string} The default type url"
         ]);
-        push(escapeName(type.name) + ".getTypeUrl = function getTypeUrl() {");
+        push(escapeName(type.name) + ".getTypeUrl = function getTypeUrl(typeUrlPrefix) {");
         ++indent;
-            push("return \"type.googleapis.com/" + exportName(type) + "\";");
+            push("if (typeUrlPrefix === undefined) {");
+            ++indent;
+                push("typeUrlPrefix = \"type.googleapis.com\";");
+            --indent;
+            push("}");
+            push("return typeUrlPrefix + \"/" + exportName(type) + "\";");
         --indent;
         push("};");
     }