fix: remove invalid inheritance #4

src/with-fork.tsx

@@ -1,3 +1,4 @@
+import * as React from "react";
 import NextDocument, { DocumentContext, DocumentProps } from "next/document";
 import { fork, serialize, allSettled } from "effector/fork";
 import cookies from "next-cookies";
@@ -24,7 +25,12 @@ export interface WithForkConfig {
 
 export function withFork({ debug }: WithForkConfig = {}) {
   return (Document: typeof NextDocument) =>
-    class WithForkDocument extends Document {
+    class WithForkDocument extends React.Component<WrappedDocumentProps> {
+      static renderDocument = Document.renderDocument;
+      static headTagsMiddleware = Document.headTagsMiddleware;
+      static bodyTagsMiddleware = Document.bodyTagsMiddleware;
+      static htmlPropsMiddleware = Document.htmlPropsMiddleware;
+
       static async getInitialProps(ctx: DocumentContext) {
         const originalRenderPage = ctx.renderPage;
         const startUnits = getStartUnits(originalRenderPage);
@@ -79,5 +85,9 @@ export function withFork({ debug }: WithForkConfig = {}) {
 
         props.__NEXT_DATA__[INITIAL_STATE_KEY] = props.initialState;
       }
+
+      render() {
+        return <Document {...this.props} />;
+      }
     };
 }

src/with-hydrate.tsx

@@ -1,3 +1,4 @@
+import * as React from "react";
 import NextApp, { AppProps } from "next/app";
 import { hydrate } from "effector/fork";
 
@@ -13,15 +14,21 @@ declare global {
 export function withHydrate() {
   const isServer = typeof window === "undefined";
 
-  return function (App: typeof NextApp) {
-    return class WithHydrateApp extends App {
+  return (App: typeof NextApp) =>
+    class WithHydrateApp extends React.Component<AppProps> {
+      static origGetInitialProps = App.origGetInitialProps;
+      static getInitialProps = App.getInitialProps;
+
       constructor(props: AppProps) {
         super(props);
 
         if (isServer) return;
 
         hydrate(domain, { values: window.__NEXT_DATA__[INITIAL_STATE_KEY] });
       }
+
+      render() {
+        return <App {...this.props} />;
+      }
     };
-  };
 }

src/with-start.ts

@@ -1,14 +1,16 @@
-import { NextPage } from "next";
+import { NextPageContext, NextComponentType } from "next";
+import { AppContext } from "next/app";
 import { Unit } from "effector";
-import App from "next/app";
 
 import { PageContext } from "./types";
 import { START_UNIT_KEY } from "./lib";
 
 /* eslint-disable @typescript-eslint/no-explicit-any */
 
+type TargetComponentType<C> = NextComponentType<C, any, any>;
+
 export function withStart(unit: Unit<PageContext>) {
-  return (component: typeof App | NextPage<any>) => {
+  return (component: TargetComponentType<AppContext> | TargetComponentType<NextPageContext>) => {
     const originalGetInitialProps = component.getInitialProps;
 
     // We cancel static optimization for a component