fix: add react import to SFC typings

Author: unstubbable

package-lock.json

@@ -69,7 +69,7 @@
     "babel-generator": "^6.26.0",
     "babylon": "7.0.0-beta.34",
     "chalk": "^2.3.0",
-    "dts-dom": "^0.1.21",
+    "dts-dom": "^0.1.24",
     "get-stdin": "^5.0.1",
     "meow": "^4.0.0",
     "pascal-case": "2.0.1",

package.json

@@ -44,11 +44,10 @@ export function createTypings(moduleName: string|null, programAst: any, options:
     ...getComponentNamesByStaticPropTypeAttribute(ast),
     ...getComponentNamesByJsxInBody(ast)
   ]);
-
   const m = dom.create.module(moduleName || 'moduleName');
-  if (hasReactClass(ast, reactComponentName)) {
-    m.members.push(dom.create.importNamed('Component', reactImport));
-  }
+
+  createReactImport(m, ast, reactImport, reactComponentName);
+
   if (importStatements.length > 0) {
     importStatements.forEach(importStatement => {
       if (importStatement.name === undefined) {
@@ -76,6 +75,15 @@ export function createTypings(moduleName: string|null, programAst: any, options:
   }
 }
 
+function createReactImport(m: dom.ModuleDeclaration, ast: AstQuery, reactImport: string,
+    reactComponentName: string|undefined): void {
+  if (hasReactClass(ast, reactComponentName)) {
+    m.members.push(dom.create.importNamed('Component', reactImport));
+  } else {
+    m.members.push(dom.create.import(reactImport));
+  }
+}
+
 function createExportedTypes(m: dom.ModuleDeclaration, ast: AstQuery, componentName: string,
     reactComponentName: string|undefined, propTypes: any, importedPropTypes: ImportedPropTypes,
       exportType: dom.DeclarationFlags, options: IOptions): void {

src/typings.ts

@@ -1,4 +1,6 @@
 declare module 'component' {
+  import 'react';
+
   export interface ComponentProps {
     optionalAny?: any;
   }

tests/named-export-specifiers.d.ts

@@ -1,4 +1,6 @@
 declare module 'component' {
+  import 'react';
+
   export interface ComponentProps {
     text: string;
     className?: string;

tests/stateless-export-as-default.d.ts

@@ -1,4 +1,6 @@
 declare module 'component' {
+  import 'react';
+
   export interface ComponentProps {
     optionalAny?: any;
   }