SQL Operations Studio Public Preview 1 (0.23) release source code

build/tfs/common/enqueue.ts

@@ -0,0 +1,85 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+'use strict';
+
+import { execSync } from 'child_process';
+import { DocumentClient } from 'documentdb';
+import * as azure from 'azure-storage';
+
+interface Asset {
+	platform: string;
+	type: string;
+	url: string;
+	mooncakeUrl: string;
+	hash: string;
+}
+
+function queueSigningRequest(quality: string, commit: string): Promise<void> {
+	const retryOperations = new azure.ExponentialRetryPolicyFilter();
+	const queueSvc = azure
+		.createQueueService(process.env['AZURE_STORAGE_ACCOUNT_2'], process.env['AZURE_STORAGE_ACCESS_KEY_2'])
+		.withFilter(retryOperations);
+
+	queueSvc.messageEncoder = new azure.QueueMessageEncoder.TextBase64QueueMessageEncoder();
+
+	const message = `${quality}/${commit}`;
+
+	return new Promise<void>((c, e) => queueSvc.createMessage('sign-darwin', message, err => err ? e(err) : c()));
+}
+
+function isBuildSigned(quality: string, commit: string): Promise<boolean> {
+	const client = new DocumentClient(process.env['AZURE_DOCUMENTDB_ENDPOINT'], { masterKey: process.env['AZURE_DOCUMENTDB_MASTERKEY'] });
+	const collection = 'dbs/builds/colls/' + quality;
+	const updateQuery = {
+		query: 'SELECT TOP 1 * FROM c WHERE c.id = @id',
+		parameters: [{ name: '@id', value: commit }]
+	};
+
+	return new Promise<boolean>((c, e) => {
+		client.queryDocuments(collection, updateQuery).toArray((err, results) => {
+			if (err) { return e(err); }
+			if (results.length !== 1) { return c(false); }
+
+			const [release] = results;
+			const assets: Asset[] = release.assets;
+			const isSigned = assets.some(a => a.platform === 'darwin' && a.type === 'archive');
+
+			c(isSigned);
+		});
+	});
+}
+
+async function waitForSignedBuild(quality: string, commit: string): Promise<void> {
+	let retries = 0;
+
+	while (retries < 180) {
+		if (await isBuildSigned(quality, commit)) {
+			return;
+		}
+
+		await new Promise<void>(c => setTimeout(c, 10000));
+		retries++;
+	}
+
+	throw new Error('Timed out waiting for signed build');
+}
+
+async function main(quality: string): Promise<void> {
+	const commit = execSync('git rev-parse HEAD', { encoding: 'utf8' }).trim();
+
+	console.log(`Queueing signing request for '${quality}/${commit}'...`);
+	await queueSigningRequest(quality, commit);
+
+	console.log('Waiting on signed build...');
+	await waitForSignedBuild(quality, commit);
+
+	console.log('Found signed build!');
+}
+
+main(process.argv[2]).catch(err => {
+	console.error(err);
+	process.exit(1);
+});