☰

Open in VSCode - OpenPAI

This script enables to open a SSH remote window in VSCode with the SSH info in OpenPAI task details.

// ==UserScript==
// @name         Open in VSCode - OpenPAI
// @namespace    http://example.com/
// @version      0.2
// @description  This script enables to open a SSH remote window in VSCode with the SSH info in OpenPAI task details.
// @author       chenyaofo
// @match        http://*/job-detail.html*
// @grant        none
// @run-at       document-start
// ==/UserScript==

(function() {

var originOpen = XMLHttpRequest.prototype.open;
var originSend = XMLHttpRequest.prototype.send;
var user = 'root';
var defaultpath = '/mnt/cephfs'
var start=new Date();
window.flag = false;

document.addEventListener('DOMContentLoaded',function(){
    setTimeout(()=>{
        var container = document.getElementsByClassName('tachyons-mt4--1DSMA tachyons-flex--1fEIE tachyons-justify-between--3nb4N tachyons-items-center--2_Sdr');
        var box = container[0].children[0];
        var line = document.createElement('div');
        line.setAttribute('class',"tachyons-bl--26VLQ tachyons-mh3--2_GwA");
        box.append(line);
        var node = document.createElement('a');
        node.setAttribute('class',"ms-Link root-151");
        node.innerText = 'Open in VSCode';
        node.setAttribute('href',`vscode://vscode-remote/ssh-remote+${user}@${window.server}:${window.port}${defaultpath}`);
        box.append(node);
    },1000);
});
// 重写open
XMLHttpRequest.prototype.open = function () {
  this.addEventListener('load', function (obj) {
    var url = obj.target.responseURL; // obj.target -> this
   if(url.indexOf('rest-server/api/v2/jobs/' > -1) && !url.endsWith('ssh') && !url.endsWith('config') && !url.endsWith('healthz')){
       //console.log('load===============', url, obj, obj.target.response);
       if(obj.target.response.length>10 && !window.flag){
           var data=JSON.parse(obj.target.response);
           window.server = data.taskRoles.taskrole.taskStatuses[0].containerIp;
           window.port = data.taskRoles.taskrole.taskStatuses[0].containerPorts.ssh;
           // console.log(data.taskRoles.taskrole.taskStatuses[0].containerIp,data.taskRoles.taskrole.taskStatuses[0].containerPorts.ssh);
           window.flag = true;
       }
   }

  });

  originOpen.apply(this, arguments);
};

})();