Add function to update user agent from package.json

src/exports/index.ts

@@ -1,6 +1,6 @@
 export { getTitleIdFromEcUrl } from '../util/misc.js';
 export { ErrorResponse, ResponseSymbol } from '../api/util.js';
-export { addUserAgent } from '../util/useragent.js';
+export { addUserAgent, addUserAgentFromPackageJson } from '../util/useragent.js';
 
 export { version } from '../util/product.js';
 

src/util/useragent.ts

@@ -1,5 +1,6 @@
 import * as process from 'node:process';
 import * as os from 'node:os';
+import * as fs from 'node:fs/promises';
 import { docker, git, release, version } from '../util/product.js';
 
 const default_useragent = 'nxapi/' + version + ' (' +
@@ -19,6 +20,68 @@ export function addUserAgent(...useragent: string[]) {
     additional_useragents.push(...useragent);
 }
 
+export function addUserAgentFromPackageJson(pkg: string | URL, additional?: string): Promise<void>;
+export function addUserAgentFromPackageJson(pkg: object, additional?: string): void;
+export function addUserAgentFromPackageJson(pkg: string | URL | object, additional?: string) {
+    if (typeof pkg === 'string' || pkg instanceof URL) {
+        return fs.readFile(pkg, 'utf-8').then(pkg => addUserAgentFromPackageJson(JSON.parse(pkg)));
+    }
+
+    const name = 'name' in pkg && typeof pkg.name === 'string' ? pkg.name : null;
+    const version = 'version' in pkg && typeof pkg.version === 'string' ? pkg.version : null;
+    if (!name || !version) throw new Error('package.json does not contain valid name and version fields');
+
+    const homepage = 'homepage' in pkg ? pkg.homepage : null;
+    if (homepage != null && typeof homepage !== 'string') throw new Error('package.json contains an invalid homepage field');
+    const repository = 'repository' in pkg && pkg.repository != null ?
+        getPackageJsonRepository(pkg.repository) : null;
+
+    const end =
+        (homepage ? '+' + homepage : repository ? '+' + repository.url : '') +
+        (repository && additional ? '; ' : '') +
+        additional;
+
+    const useragent = name + '/' + version + (end ? ' (' + end + ')' : '');
+
+    addUserAgent(useragent);
+}
+
+function getPackageJsonRepository(repository: unknown): {
+    type: string; url: string; directory?: string | null;
+} {
+    if (typeof repository === 'string') {
+        if (repository.startsWith('github:')) {
+            return {type: 'git', url: 'https://github.com/' + repository.substr(7)};
+        }
+        if (repository.startsWith('gist:')) {
+            return {type: 'git', url: 'https://gist.github.com/' + repository.substr(5)};
+        }
+        if (repository.startsWith('bitbucket:')) {
+            return {type: 'git', url: 'https://bitbucket.org/' + repository.substr(10)};
+        }
+        if (repository.startsWith('gitlab:')) {
+            return {type: 'git', url: 'https://gitlab.com/' + repository.substr(7)};
+        }
+        if (repository.match(/^[0-9a-z-.]+\/[0-9a-z-.]+$/i)) {
+            return {type: 'git', url: 'https://github.com/' + repository};
+        }
+    }
+
+    if (typeof repository === 'object' && repository) {
+        if ('type' in repository && typeof repository.type === 'string' &&
+            'url' in repository && typeof repository.url === 'string' &&
+            (!('directory' in repository) || repository.directory == null || typeof repository.directory === 'string')
+        ) {
+            return {
+                type: repository.type, url: repository.url,
+                directory: 'directory' in repository ? repository.directory as string : null,
+            };
+        }
+    }
+
+    throw new Error('package.json contains an invalid repository field');
+}
+
 /**
  * Only used by cli/nso/http-server.ts.
  * This command is intended to be run automatically and doesn't make any requests itself, so this function removes