@nozbe/watermelondb/src/react/withDatabase.js

Build powerful React Native and React web apps that scale from hundreds to tens of thousands of records and remain fast

// @flow
import React from 'react'
import hoistNonReactStatics from 'hoist-non-react-statics'
import type Database from '../Database'
import { DatabaseConsumer } from './DatabaseContext'

type WithDatabaseProps<T: {}> = {
  ...T,
  database: Database,
}
// HoC to inject the database into the props of consumers
export default function withDatabase<T: {}>(
  Component: React$ComponentType<WithDatabaseProps<T>>,
): React$ComponentType<T> {
  function DatabaseComponent(props: any): React$Element<any> {
    return (
      <DatabaseConsumer>
        {(database: Database) => <Component {...props} database={database} />}
      </DatabaseConsumer>
    )
  }

  return hoistNonReactStatics(DatabaseComponent, Component)
}