From 0d159d04ce5683d92e9ed508ff9b70e3d131d5ff Mon Sep 17 00:00:00 2001
From: Andrew Lisowski <lisowski54@gmail.com>
Date: Thu, 3 Dec 2020 09:42:49 -0800
Subject: [PATCH] actually use new code

---
 src/generateDocgenCodeBlock.ts | 4 +---
 src/plugin.ts                  | 3 ++-
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/generateDocgenCodeBlock.ts b/src/generateDocgenCodeBlock.ts
index 0eed5bd..b8abe8a 100644
--- a/src/generateDocgenCodeBlock.ts
+++ b/src/generateDocgenCodeBlock.ts
@@ -321,9 +321,7 @@ export interface GeneratorOptions {
   typePropName: string;
 }
 
-export default function generateDocgenCodeBlock(
-  options: GeneratorOptions
-): string {
+export function generateDocgenCodeBlock(options: GeneratorOptions): string {
   const sourceFile = ts.createSourceFile(
     options.filename,
     options.source,
diff --git a/src/plugin.ts b/src/plugin.ts
index 581fbb7..d8dd2af 100644
--- a/src/plugin.ts
+++ b/src/plugin.ts
@@ -3,9 +3,10 @@ import createDebug from "debug";
 import * as webpack from "webpack";
 import ts from "typescript";
 import * as docGen from "react-docgen-typescript";
-import generateDocgenCodeBlock from "react-docgen-typescript-loader/dist/generateDocgenCodeBlock";
 import { matcher } from "micromatch";
 
+import { generateDocgenCodeBlock } from "./generateDocgenCodeBlock";
+
 const debugExclude = createDebug("docgen:exclude");
 const debugInclude = createDebug("docgen:include");
 const debugDocs = createDebug("docgen:docs");