Rocket.Chat.ReactNative/app/lib/database/index.js

133 lines
2.9 KiB
JavaScript

import { Database } from '@nozbe/watermelondb';
import SQLiteAdapter from '@nozbe/watermelondb/adapters/sqlite';
import logger from '@nozbe/watermelondb/utils/common/logger';
import Subscription from './model/Subscription';
import Room from './model/Room';
import Message from './model/Message';
import Thread from './model/Thread';
import ThreadMessage from './model/ThreadMessage';
import CustomEmoji from './model/CustomEmoji';
import FrequentlyUsedEmoji from './model/FrequentlyUsedEmoji';
import Upload from './model/Upload';
import Setting from './model/Setting';
import Role from './model/Role';
import Permission from './model/Permission';
import SlashCommand from './model/SlashCommand';
import User from './model/User';
import Server from './model/Server';
import ServersHistory from './model/ServersHistory';
import serversSchema from './schema/servers';
import appSchema from './schema/app';
import migrations from './model/migrations';
import serversMigrations from './model/serversMigrations';
import { isIOS } from '../../utils/deviceInfo';
import appGroup from '../../utils/appGroup';
const appGroupPath = isIOS ? appGroup.path : '';
if (__DEV__ && isIOS) {
console.log(appGroupPath);
}
export const getDatabase = (database = '') => {
const path = database.replace(/(^\w+:|^)\/\//, '').replace(/\//g, '.');
const dbName = `${ appGroupPath }${ path }-experimental.db`;
const adapter = new SQLiteAdapter({
dbName,
schema: appSchema,
migrations
});
return new Database({
adapter,
modelClasses: [
Subscription,
Room,
Message,
Thread,
ThreadMessage,
CustomEmoji,
FrequentlyUsedEmoji,
Upload,
Setting,
Role,
Permission,
SlashCommand
],
actionsEnabled: true
});
};
class DB {
databases = {
serversDB: new Database({
adapter: new SQLiteAdapter({
dbName: `${ appGroupPath }default-experimental.db`,
schema: serversSchema,
migrations: serversMigrations
}),
modelClasses: [Server, User, ServersHistory],
actionsEnabled: true
})
}
get active() {
return this.databases.shareDB || this.databases.activeDB;
}
get share() {
return this.databases.shareDB;
}
set share(db) {
this.databases.shareDB = db;
}
get servers() {
return this.databases.serversDB;
}
setShareDB(database = '') {
const path = database.replace(/(^\w+:|^)\/\//, '').replace(/\//g, '.');
const dbName = `${ appGroupPath }${ path }-experimental.db`;
const adapter = new SQLiteAdapter({
dbName,
schema: appSchema,
migrations
});
this.databases.shareDB = new Database({
adapter,
modelClasses: [
Subscription,
Message,
Thread,
ThreadMessage,
Upload,
Permission,
CustomEmoji,
FrequentlyUsedEmoji
],
actionsEnabled: true
});
}
setActiveDB(database) {
this.databases.activeDB = getDatabase(database);
}
}
const db = new DB();
export default db;
if (!__DEV__) {
logger.silence();
}