Skip to content

Commit 7343b3e

Browse files
committed
deps: remove imports of v2 ec2
1 parent 15a37f5 commit 7343b3e

File tree

4 files changed

+9
-11
lines changed

4 files changed

+9
-11
lines changed

packages/core/src/awsService/ec2/explorer/ec2InstanceNode.ts

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ import globals from '../../../shared/extensionGlobals'
1111
import { getIconCode } from '../utils'
1212
import { Ec2Selection } from '../prompter'
1313
import { Ec2Node, Ec2ParentNode } from './ec2ParentNode'
14-
import { EC2 } from 'aws-sdk'
1514
import { getLogger } from '../../../shared/logger/logger'
1615
import { InstanceStateName } from '@aws-sdk/client-ec2'
1716

@@ -80,7 +79,7 @@ export class Ec2InstanceNode extends AWSTreeNodeBase implements AWSResourceNode
8079
}
8180
}
8281

83-
public getStatus(): EC2.InstanceStateName {
82+
public getStatus(): InstanceStateName {
8483
return this.instance.LastSeenStatus
8584
}
8685

packages/core/src/awsService/ec2/model.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
* SPDX-License-Identifier: Apache-2.0
44
*/
55
import * as vscode from 'vscode'
6-
import { EC2, IAM } from 'aws-sdk'
6+
import { IAM } from 'aws-sdk'
77
import { Ec2Selection } from './prompter'
88
import { getOrInstallCli } from '../../shared/utilities/cliUtils'
99
import { isCloud9 } from '../../shared/extensionUtilities'
@@ -83,7 +83,7 @@ export class Ec2Connecter implements vscode.Disposable {
8383
return new DefaultIamClient(this.regionCode)
8484
}
8585

86-
public async addActiveSession(sessionId: string, instanceId: EC2.InstanceId): Promise<void> {
86+
public async addActiveSession(sessionId: string, instanceId: string): Promise<void> {
8787
await this.sessionManager.addSession(instanceId, sessionId)
8888
}
8989

packages/core/src/awsService/ec2/remoteSessionManager.ts

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,19 +3,18 @@
33
* SPDX-License-Identifier: Apache-2.0
44
*/
55

6-
import { EC2 } from 'aws-sdk'
76
import { SsmClient } from '../../shared/clients/ssm'
87
import { Disposable } from 'vscode'
98

10-
export class Ec2SessionTracker extends Map<EC2.InstanceId, string> implements Disposable {
9+
export class Ec2SessionTracker extends Map<string, string> implements Disposable {
1110
public constructor(
1211
readonly regionCode: string,
1312
protected ssm: SsmClient
1413
) {
1514
super()
1615
}
1716

18-
public async addSession(instanceId: EC2.InstanceId, sessionId: string): Promise<void> {
17+
public async addSession(instanceId: string, sessionId: string): Promise<void> {
1918
if (this.isConnectedTo(instanceId)) {
2019
const existingSessionId = this.get(instanceId)!
2120
await this.ssm.terminateSessionFromId(existingSessionId)
@@ -25,7 +24,7 @@ export class Ec2SessionTracker extends Map<EC2.InstanceId, string> implements Di
2524
}
2625
}
2726

28-
private async disconnectEnv(instanceId: EC2.InstanceId): Promise<void> {
27+
private async disconnectEnv(instanceId: string): Promise<void> {
2928
await this.ssm.terminateSessionFromId(this.get(instanceId)!)
3029
this.delete(instanceId)
3130
}
@@ -35,7 +34,7 @@ export class Ec2SessionTracker extends Map<EC2.InstanceId, string> implements Di
3534
this.forEach(async (_sessionId, instanceId) => await this.disconnectEnv(instanceId))
3635
}
3736

38-
public isConnectedTo(instanceId: EC2.InstanceId): boolean {
37+
public isConnectedTo(instanceId: string): boolean {
3938
return this.has(instanceId)
4039
}
4140
}

packages/core/src/test/awsService/ec2/explorer/ec2ParentNode.test.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,9 @@ import {
1212
assertNodeListOnlyHasPlaceholderNode,
1313
} from '../../../utilities/explorerNodeAssertions'
1414
import { Ec2InstanceNode } from '../../../../awsService/ec2/explorer/ec2InstanceNode'
15-
import { EC2 } from 'aws-sdk'
1615
import * as FakeTimers from '@sinonjs/fake-timers'
1716
import { installFakeClock } from '../../../testUtil'
17+
import { Filter } from '@aws-sdk/client-ec2'
1818

1919
export const testInstance = {
2020
InstanceId: 'testId',
@@ -32,7 +32,7 @@ export const testParentNode = new Ec2ParentNode('fake-region', 'testPartition',
3232
describe('ec2ParentNode', function () {
3333
let testNode: Ec2ParentNode
3434
let client: Ec2Client
35-
let getInstanceStub: sinon.SinonStub<[filters?: EC2.Filter[] | undefined], Promise<SafeEc2Instance[]>>
35+
let getInstanceStub: sinon.SinonStub<[filters?: Filter[] | undefined], Promise<SafeEc2Instance[]>>
3636
let clock: FakeTimers.InstalledClock
3737
let refreshStub: sinon.SinonStub<[], Promise<void>>
3838
let statusUpdateStub: sinon.SinonStub<[status: string], Promise<string>>

0 commit comments

Comments
 (0)