feat: support link that open with code to carry access token
This commit is contained in:
50
src/main.ts
50
src/main.ts
@@ -1,15 +1,17 @@
|
||||
import * as vscode from 'vscode';
|
||||
import QuickAccessTreeProvider from './views/quick-access-tree';
|
||||
import DSHome from './home';
|
||||
import {openProjectWithoutLogging} from './remote-container';
|
||||
import RemoteContainer, {openProjectWithoutLogging} from './remote-container';
|
||||
import User from './user';
|
||||
|
||||
export class DevStarExtension {
|
||||
user: User;
|
||||
remoteContainer: RemoteContainer;
|
||||
dsHome: DSHome;
|
||||
|
||||
constructor(private context: vscode.ExtensionContext) {
|
||||
this.dsHome = new DSHome(context);
|
||||
|
||||
this.registerGlobalCommands(context);
|
||||
this.user = new User(context);
|
||||
this.remoteContainer = new RemoteContainer(this.user);
|
||||
|
||||
const handler = vscode.window.registerUriHandler({
|
||||
handleUri: async (uri: vscode.Uri) => {
|
||||
@@ -19,10 +21,45 @@ export class DevStarExtension {
|
||||
const port = params.get('port');
|
||||
const username = params.get('username');
|
||||
const path = params.get('path');
|
||||
const access_token = params.get('access_token');
|
||||
const devstar_username = params.get('devstar_username');
|
||||
|
||||
if (host && port && username && path) {
|
||||
await openProjectWithoutLogging(host, parseInt(port, 10), username, decodeURIComponent(path))
|
||||
const container_host = host;
|
||||
const container_port = parseInt(port, 10);
|
||||
const container_username = username;
|
||||
const project_path = decodeURIComponent(path);
|
||||
|
||||
if (access_token && devstar_username) {
|
||||
if (!this.user.isLogged()) {
|
||||
// 如果没有用户登录,则直接登录;
|
||||
this.user.setUserTokenToLocal(access_token)
|
||||
this.user.setUsernameToLocal(devstar_username)
|
||||
} else if (devstar_username === this.user.getUsernameFromLocal()){
|
||||
// 如果同用户已经登录,则弃用,无操作;
|
||||
} else {
|
||||
// 如果不是同用户,可以选择切换用户,或者不切换登录用户,直接打开容器
|
||||
const selection = await vscode.window.showWarningMessage(`已登录用户:${this.user.getUsernameFromLocal()},是否切换用户?`,
|
||||
'Yes', 'No',);
|
||||
if (selection === 'Yes') {
|
||||
this.user.setUserTokenToLocal(access_token);
|
||||
this.user.setUsernameToLocal(devstar_username);
|
||||
} else if (selection === 'No') {
|
||||
await openProjectWithoutLogging(container_host, container_port, container_username, project_path);
|
||||
}
|
||||
}
|
||||
|
||||
// 使用登录已经登录过的容器连接方法
|
||||
await this.remoteContainer.firstConnect(container_host, container_username, container_port)
|
||||
.then((_res) => {
|
||||
if (_res == 'success') {
|
||||
// only success then open folder
|
||||
this.remoteContainer.openRemoteFolder(container_host, container_port, container_username, project_path);
|
||||
}
|
||||
})
|
||||
} else {
|
||||
await openProjectWithoutLogging(container_host, container_port, container_username, project_path);
|
||||
}
|
||||
} else {
|
||||
vscode.window.showErrorMessage('Missing required parameters.');
|
||||
}
|
||||
@@ -39,6 +76,9 @@ export class DevStarExtension {
|
||||
)
|
||||
);
|
||||
|
||||
this.dsHome = new DSHome(context);
|
||||
this.registerGlobalCommands(context);
|
||||
|
||||
this.startDevStarHome();
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user