Way to retain only whitelisted export names

[brijeshb42]

Hi Everyone. I am working on library that uses oxc to parse and replace some of the ast nodes in a code. I had a specific use-case, where I want the code to only retain an export named something specific, for example,

import colors from './colors';
import React from 'react';

export const a = React.createContext();

function keyFn() {}

export const whitelistedItem = {
  color: colors.red,
  key1: keyFn,
};

The output I am expecting is -

import colors from './colors';

function keyFn() {}

export const whitelistedItem = {
  color: colors.red,
  key1: keyFn,
};

It removed React import as well as the local variable a since these are not directly/indirectly referred in whiteListedItem.

It's kind of like tree-shaking but for specific exports. I was going about doing this in one of two ways -

1. Construct a new ast from scratch and copy over the relevant ast nodes from the source code to the new ast.
   I was trying this method but I got stuck on identifier/symbol tracking. I was able to track the identifiers being used in a specific exports but could not find a way to get the original statement where the symbol is declared. For ex, for the above input code, I get the colors's symbol but could not find a way to get to the import statement just from the symbol id. Any pointer here might help me make this work as well.
2. Or, just tree-shake the original code to get what I want. This seems a bit fool-proof (at-least in theory) because, in point 1, there may be case where some statement is missed coying over to the new AST but with step 2, we are only removing undesired code.

Any help or pointer is appreciated.