67 lines
2 KiB
TypeScript
67 lines
2 KiB
TypeScript
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
import React, {ComponentType, useEffect, useState} from 'react';
|
|
import {Database} from '@nozbe/watermelondb';
|
|
import DatabaseProvider from '@nozbe/watermelondb/DatabaseProvider';
|
|
|
|
import {MM_TABLES} from '@constants/database';
|
|
import ServerUrlProvider from '@context/server_url';
|
|
import ThemeProvider from '@context/theme';
|
|
import DatabaseManager from '@database/manager';
|
|
import Servers from '@database/models/app/servers';
|
|
|
|
type State = {
|
|
database: Database;
|
|
serverUrl: string;
|
|
}
|
|
|
|
const {SERVERS} = MM_TABLES.APP;
|
|
|
|
export function withServerDatabase<T>(
|
|
Component: ComponentType<T>,
|
|
): ComponentType<T> {
|
|
return function ServerDatabaseComponent(props) {
|
|
const [state, setState] = useState<State|undefined>();
|
|
const db = DatabaseManager.appDatabase?.database;
|
|
|
|
const observer = async (servers: Servers[]) => {
|
|
const server = servers.reduce((a, b) => (b.lastActiveAt > a.lastActiveAt ? b : a));
|
|
|
|
const serverDatabase = DatabaseManager.serverDatabases[server?.url]?.database;
|
|
|
|
setState({
|
|
database: serverDatabase,
|
|
serverUrl: server?.url,
|
|
});
|
|
};
|
|
|
|
useEffect(() => {
|
|
const subscription = db?.collections.
|
|
get(SERVERS).
|
|
query().
|
|
observeWithColumns(['last_active_at']).
|
|
subscribe(observer);
|
|
|
|
return () => {
|
|
subscription?.unsubscribe();
|
|
};
|
|
}, []);
|
|
|
|
if (!state?.database) {
|
|
return null;
|
|
}
|
|
|
|
return (
|
|
<DatabaseProvider
|
|
database={(state.database)}
|
|
>
|
|
<ServerUrlProvider url={state.serverUrl}>
|
|
<ThemeProvider database={state.database}>
|
|
<Component {...props}/>
|
|
</ThemeProvider>
|
|
</ServerUrlProvider>
|
|
</DatabaseProvider>
|
|
);
|
|
};
|
|
}
|