provide stack mgmt info for a resource

Author: Zee2413

src/handlers/ResourceHandler.ts

@@ -8,7 +8,9 @@ import {
     ResourceStateParams,
     ResourceStateResult,
     ResourceSummary,
+    ResourceIdentifier,
 } from '../resourceState/ResourceStateTypes';
+import { ResourceStackManagementResult } from '../resourceState/StackManagementInfoProvider';
 import { ServerComponents } from '../server/ServerComponents';
 import { LoggerFactory } from '../telemetry/LoggerFactory';
 import { extractErrorMessage } from '../utils/Errors';
@@ -86,3 +88,11 @@ export function refreshResourceListHandler(
         }
     };
 }
+
+export function getStackMgmtInfo(
+    components: ServerComponents,
+): ServerRequestHandler<ResourceIdentifier, ResourceStackManagementResult, never, void> {
+    return async (id) => {
+        return await components.stackManagementInfoProvider.getResourceManagementState(id);
+    };
+}

src/protocol/LspResourceHandlers.ts

@@ -11,7 +11,10 @@ import {
     RefreshResourceListRequest,
     RefreshResourcesParams,
     RefreshResourcesResult,
+    StackMgmtInfoRequest,
+    ResourceIdentifier,
 } from '../resourceState/ResourceStateTypes';
+import { ResourceStackManagementResult } from '../resourceState/StackManagementInfoProvider';
 
 export class LspResourceHandlers {
     constructor(private readonly connection: Connection) {}
@@ -31,4 +34,8 @@ export class LspResourceHandlers {
     onResourceStateImport(handler: ServerRequestHandler<ResourceStateParams, ResourceStateResult, never, void>) {
         this.connection.onRequest(ResourceStateRequest.method, handler);
     }
+
+    onStackMgmtInfo(handler: ServerRequestHandler<ResourceIdentifier, ResourceStackManagementResult, never, void>) {
+        this.connection.onRequest(StackMgmtInfoRequest.method, handler);
+    }
 }

src/resourceState/ResourceStateTypes.ts

@@ -1,5 +1,6 @@
 import { CodeAction, CodeActionParams } from 'vscode-languageserver';
 import { RequestType } from 'vscode-languageserver-protocol';
+import { ResourceStackManagementResult } from './StackManagementInfoProvider';
 
 // eslint-disable-next-line @typescript-eslint/no-empty-object-type
 export type ResourceTypesParams = {};
@@ -86,3 +87,7 @@ export interface ResourceTemplateFormat {
         };
     };
 }
+
+export const StackMgmtInfoRequest = new RequestType<ResourceIdentifier, ResourceStackManagementResult, void>(
+    'aws/cfn/resources/stackMgmtInfo',
+);

src/resourceState/StackManagementInfoProvider.ts

@@ -2,7 +2,7 @@ import { CloudFormationServiceException } from '@aws-sdk/client-cloudformation';
 import { CfnService } from '../services/CfnService';
 import { LoggerFactory } from '../telemetry/LoggerFactory';
 
-type ResourceStackManagementResult = {
+export type ResourceStackManagementResult = {
     physicalResourceId: string;
     managedByStack: boolean | undefined;
     stackName?: string;

src/server/CfnServer.ts

@@ -22,6 +22,7 @@ import {
     getResourceTypesHandler,
     importResourceStateHandler,
     refreshResourceListHandler,
+    getStackMgmtInfo,
 } from '../handlers/ResourceHandler';
 import { listStacksHandler } from '../handlers/StackHandler';
 import {
@@ -82,6 +83,7 @@ export class CfnServer {
         this.features.resourceHandlers.onRefreshResourceList(refreshResourceListHandler(this.components));
         this.features.resourceHandlers.onGetResourceTypes(getResourceTypesHandler(this.components));
         this.features.resourceHandlers.onResourceStateImport(importResourceStateHandler(this.components));
+        this.features.resourceHandlers.onStackMgmtInfo(getStackMgmtInfo(this.components));
     }
 
     async close(): Promise<void> {