feat(overmind-vue): add typing support for connect

Author: christianalfoni

packages/node_modules/overmind-vue/src/Index.test.ts

@@ -2,6 +2,7 @@ import { JSDOM } from 'jsdom'
 import { Overmind } from 'overmind'
 
 import { createConnect } from './'
+
 const Vue = require('vue/dist/vue')
 
 const dom = new JSDOM()

packages/node_modules/overmind-vue/src/index.ts

@@ -1,4 +1,7 @@
-import { EventType, MODE_SSR } from 'overmind'
+import { EventType, IConfiguration, MODE_SSR, Overmind } from 'overmind'
+import { Component, ComponentOptions } from 'vue'
+
+type AnyComponent = ComponentOptions<any, any, any> | Component
 
 const OVERMIND = Symbol('OVERMIND')
 const IS_PRODUCTION = process.env.NODE_ENV === 'production'
@@ -143,21 +146,25 @@ export const createPlugin = (overmind) => ({
   },
 })
 
-export const createConnect = (overmind) => (...args) => {
-  let options = args.length === 1 ? args[0] : args[1]
-  let propsCallback = args.length === 1 ? null : args[0]
 
-  if (propsCallback && typeof propsCallback !== 'function') {
-    throw new Error(
-      `OVERMIND-VUE: When passing two arguments to "connect", the first has to be a function. You can alternatively only pass a single argument, which is the component`
-    )
-  }
 
-  return {
-    ...options,
-    mixins: (options.mixins ? options.mixins : []).concat(
-      createMixin(overmind, propsCallback, true)
-    ),
-    overmind,
-  } as any
+export function createConnect<Config extends IConfiguration>(overmind: Overmind<Config>) {
+  return <T extends {}>(cb: ((overmind: Overmind<Config>) => T) | AnyComponent, component?: AnyComponent) => {
+    let options: any = component || cb
+    let propsCallback = component ? cb : null
+
+    if (propsCallback && typeof propsCallback !== 'function') {
+      throw new Error(
+        `OVERMIND-VUE: When passing two arguments to "connect", the first has to be a function. You can alternatively only pass a single argument, which is the component`
+      )
+    }
+
+    return {
+      ...options,
+      mixins: (options.mixins ? options.mixins : []).concat(
+        createMixin(overmind, propsCallback, true)
+      ),
+      overmind,
+    } as any
+  }
 }