How to type-safely handle specific action returns in $onAction?

[zhristophe]

Hey Pinia team! 👋

I'm wondering about a TypeScript type inference challenge with $onAction.

When using something like:

assetStore.$onAction(({ name, after }) => {  
  after((ret) => {  
    if (name === "importAssetsFromPath") {  
      // How can I make TypeScript understand 'ret' is AssetImportResult here?  
      handleAssetsImported(ret as AssetImportResult);  
    }  
  });  
});  

Right now, I'm using a type assertion, but it feels a bit hacky.

Any elegant TypeScript solutions or patterns you'd recommend for this scenario?

Curious to hear your thoughts! 🤔

[posva]

Putting the if outside of the after should constrain the type as expected