// Copyright 2021 The Chromium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. 'use strict'; const child_process = require('child_process'); const fs = require('fs'); const os = require('os'); const path = require('path'); // QueryRunner uses the swarming binary to retrieve data from the swarming // servers. class QueryRunner { constructor(server) { this.swarming_server_ = server; this.bin_ = './tools/luci-go/swarming'; } // Retrieves the list of tasks matching `tag`. retrieveTasks(tag, fields = [ 'task_id', 'run_id', 'failure', 'state', 'tags' ]) { const cmds = [ this.bin_, 'tasks', '-S', this.swarming_server_, `-field="items(${fields.join(',')})"`, `-tag="${tag}"` ]; const output = child_process.execSync(cmds.join(' ')); return JSON.parse(output.toString()); } // Returns a promoise for a parsed JSON object from the contents in `filename` // for the specified `task_id`. retrieveJSONFile(task_id, filename) { // This is going to create some temp directory. So put this in a try/finally // block so that the temp directory gets cleaned up correctly. let temp_dir = undefined; try { temp_dir = fs.mkdtempSync(path.join(os.tmpdir(), 'cr-fetch')); return this.retrieveFile_(task_id, filename, temp_dir); } catch (e) { console.error(e); } finally { if (temp_dir) { try { fs.rmdirSync(temp_dir, {recursive: true, force: false}); } catch (e) { console.error(`Failed to cleanup temp directory at ${temp_dir}.`); console.error(e); } } } } retrieveFile_(task_id, filename, temp_dir) { return new Promise((resolve, reject) => { const cmds = [ this.bin_, 'collect', '-S', this.swarming_server_, `-output-dir="${temp_dir}"`, task_id ]; const output = child_process.exec(cmds.join(' '), () => { const filepath = path.join(temp_dir, task_id, filename); resolve(JSON.parse(fs.readFileSync(filepath).toString())); }); }); } }; module.exports = { QueryRunner, };