Addon-docs: Apply transformSource to any SourceType

addons/docs/src/blocks/Source.tsx

@@ -81,12 +81,12 @@ const getSnippet = (snippet: string, storyContext?: StoryContext): string => {
 
   // if user has explicitly set this as dynamic, use snippet
   if (type === SourceType.DYNAMIC) {
-    return snippet;
+    return parameters.docs?.transformSource?.(snippet, storyContext) || snippet;
   }
 
   // if this is an args story and there's a snippet
   if (type === SourceType.AUTO && snippet && isArgsStory) {
-    return snippet;
+    return parameters.docs?.transformSource?.(snippet, storyContext) || snippet;
   }
 
   // otherwise, use the source code logic

examples/official-storybook/stories/addon-docs/transform-source.stories.js

@@ -0,0 +1,25 @@
+export default {
+  title: 'Addons/Docs/transformSource',
+  parameters: {
+    docs: {
+      transformSource(src, ctx) {
+        return `// We transformed this!\n const example = ${src};`;
+      },
+    },
+  },
+};
+
+export const code = () => 'StoryType "CODE" story which has source transformed';
+code.parameters = {
+  docs: { source: { type: 'code' } },
+};
+
+export const dynamic = () => 'StoryType "DYNAMIC" story which has source transformed';
+dynamic.parameters = {
+  docs: { source: { type: 'dynamic' } },
+};
+
+export const auto = () => 'StoryType "AUTO" story which has source transformed';
+dynamic.parameters = {
+  docs: { source: { type: 'auto' } },
+};