feat: update types and generation script (#1336)

alvarowolfx · web-flow · commit 496f52c3816d · 2024-02-16T13:33:10.000-04:00
diff --git a/package.json b/package.json
@@ -43,7 +43,7 @@
     "pretest": "npm run compile",
     "docs-test": "linkinator docs",
     "predocs-test": "npm run docs",
-    "types": "dtsd bigquery v2 > ./src/types.d.ts",
+    "types": "node scripts/gen-types.js",
     "prelint": "cd samples; npm link ../; npm install",
     "precompile": "gts clean"
   },
diff --git a/scripts/gen-types.js b/scripts/gen-types.js
@@ -0,0 +1,65 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+const {fetch} = require('discovery-tsd');
+const TypeGenerator = require('discovery-tsd/src/generator');
+const prettier = require('prettier');
+const fs = require('fs');
+const {promisify} = require('util');
+
+const writeFile = promisify(fs.writeFile);
+
+const LICENSE = `// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.`;
+
+function overridedRender() {
+  const source = this.template({
+    title: this.title ? this.converter.toJSDoc(this.title) : '',
+    name: this.name,
+    schemas: this.schemas.map(schema => this.converter.createType(schema)),
+    resources: this.resources.map(resource => resource.render()),
+  });
+
+  const patched = source.replaceAll(
+    'formatOptions.useInt64Timestamp',
+    "'formatOptions.useInt64Timestamp'"
+  );
+  const sourceWithLicense = LICENSE + '\n' + patched;
+
+  return prettier.format(sourceWithLicense, {
+    parser: 'typescript',
+    singleQuote: true,
+  });
+}
+
+async function genTypes() {
+  const json = await fetch('bigquery', 'v2');
+  const generator = new TypeGenerator(json);
+  generator.render = overridedRender.bind(generator);
+  const types = await generator.render();
+  await writeFile('./src/types.d.ts', types);
+}
+
+genTypes();
diff --git a/src/types.d.ts b/src/types.d.ts
