2019-04-07 21:50:36 +09:00
|
|
|
import { createConnection, Logger, getConnection } from 'typeorm';
|
|
|
|
import config from '../config';
|
|
|
|
import { entities as charts } from '../services/chart/entities';
|
|
|
|
import { dbLogger } from './logger';
|
|
|
|
import * as highlight from 'cli-highlight';
|
|
|
|
|
|
|
|
import { Log } from '../models/entities/log';
|
|
|
|
import { User } from '../models/entities/user';
|
|
|
|
import { DriveFile } from '../models/entities/drive-file';
|
|
|
|
import { DriveFolder } from '../models/entities/drive-folder';
|
|
|
|
import { AccessToken } from '../models/entities/access-token';
|
|
|
|
import { App } from '../models/entities/app';
|
|
|
|
import { PollVote } from '../models/entities/poll-vote';
|
|
|
|
import { Note } from '../models/entities/note';
|
|
|
|
import { NoteReaction } from '../models/entities/note-reaction';
|
|
|
|
import { NoteWatching } from '../models/entities/note-watching';
|
|
|
|
import { NoteUnread } from '../models/entities/note-unread';
|
|
|
|
import { Notification } from '../models/entities/notification';
|
|
|
|
import { Meta } from '../models/entities/meta';
|
|
|
|
import { Following } from '../models/entities/following';
|
|
|
|
import { Instance } from '../models/entities/instance';
|
|
|
|
import { Muting } from '../models/entities/muting';
|
|
|
|
import { SwSubscription } from '../models/entities/sw-subscription';
|
|
|
|
import { Blocking } from '../models/entities/blocking';
|
|
|
|
import { UserList } from '../models/entities/user-list';
|
|
|
|
import { UserListJoining } from '../models/entities/user-list-joining';
|
2019-05-18 20:36:33 +09:00
|
|
|
import { UserGroup } from '../models/entities/user-group';
|
|
|
|
import { UserGroupJoining } from '../models/entities/user-group-joining';
|
2020-02-13 02:17:54 +09:00
|
|
|
import { UserGroupInvitation } from '../models/entities/user-group-invitation';
|
2019-04-07 21:50:36 +09:00
|
|
|
import { Hashtag } from '../models/entities/hashtag';
|
|
|
|
import { NoteFavorite } from '../models/entities/note-favorite';
|
|
|
|
import { AbuseUserReport } from '../models/entities/abuse-user-report';
|
|
|
|
import { RegistrationTicket } from '../models/entities/registration-tickets';
|
|
|
|
import { MessagingMessage } from '../models/entities/messaging-message';
|
|
|
|
import { Signin } from '../models/entities/signin';
|
|
|
|
import { AuthSession } from '../models/entities/auth-session';
|
|
|
|
import { FollowRequest } from '../models/entities/follow-request';
|
|
|
|
import { Emoji } from '../models/entities/emoji';
|
|
|
|
import { ReversiGame } from '../models/entities/games/reversi/game';
|
|
|
|
import { ReversiMatching } from '../models/entities/games/reversi/matching';
|
|
|
|
import { UserNotePining } from '../models/entities/user-note-pinings';
|
|
|
|
import { Poll } from '../models/entities/poll';
|
|
|
|
import { UserKeypair } from '../models/entities/user-keypair';
|
|
|
|
import { UserPublickey } from '../models/entities/user-publickey';
|
2019-04-10 15:04:27 +09:00
|
|
|
import { UserProfile } from '../models/entities/user-profile';
|
2019-07-03 20:18:07 +09:00
|
|
|
import { UserSecurityKey } from '../models/entities/user-security-key';
|
|
|
|
import { AttestationChallenge } from '../models/entities/attestation-challenge';
|
2019-04-29 09:11:57 +09:00
|
|
|
import { Page } from '../models/entities/page';
|
2019-05-17 19:56:47 +09:00
|
|
|
import { PageLike } from '../models/entities/page-like';
|
2019-07-14 03:18:45 +09:00
|
|
|
import { ModerationLog } from '../models/entities/moderation-log';
|
2019-07-22 10:15:00 +09:00
|
|
|
import { UsedUsername } from '../models/entities/used-username';
|
2020-01-30 04:37:25 +09:00
|
|
|
import { Announcement } from '../models/entities/announcement';
|
|
|
|
import { AnnouncementRead } from '../models/entities/announcement-read';
|
|
|
|
import { Clip } from '../models/entities/clip';
|
|
|
|
import { ClipNote } from '../models/entities/clip-note';
|
|
|
|
import { Antenna } from '../models/entities/antenna';
|
|
|
|
import { AntennaNote } from '../models/entities/antenna-note';
|
2020-02-18 08:41:32 +09:00
|
|
|
import { PromoNote } from '../models/entities/promo-note';
|
|
|
|
import { PromoRead } from '../models/entities/promo-read';
|
2020-03-28 18:28:21 +09:00
|
|
|
import { program } from '../argv';
|
2020-05-10 18:42:31 +09:00
|
|
|
import { Relay } from '../models/entities/relay';
|
2020-07-27 13:34:20 +09:00
|
|
|
import { MutedNote } from '../models/entities/muted-note';
|
2019-04-07 21:50:36 +09:00
|
|
|
|
|
|
|
const sqlLogger = dbLogger.createSubLogger('sql', 'white', false);
|
|
|
|
|
|
|
|
class MyCustomLogger implements Logger {
|
|
|
|
private highlight(sql: string) {
|
|
|
|
return highlight.highlight(sql, {
|
|
|
|
language: 'sql', ignoreIllegals: true,
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
public logQuery(query: string, parameters?: any[]) {
|
2020-03-29 14:08:19 +09:00
|
|
|
if (program.verbose) {
|
|
|
|
sqlLogger.info(this.highlight(query));
|
|
|
|
}
|
2019-04-07 21:50:36 +09:00
|
|
|
}
|
|
|
|
|
|
|
|
public logQueryError(error: string, query: string, parameters?: any[]) {
|
|
|
|
sqlLogger.error(this.highlight(query));
|
|
|
|
}
|
|
|
|
|
|
|
|
public logQuerySlow(time: number, query: string, parameters?: any[]) {
|
|
|
|
sqlLogger.warn(this.highlight(query));
|
|
|
|
}
|
|
|
|
|
|
|
|
public logSchemaBuild(message: string) {
|
|
|
|
sqlLogger.info(message);
|
|
|
|
}
|
|
|
|
|
|
|
|
public log(message: string) {
|
|
|
|
sqlLogger.info(message);
|
|
|
|
}
|
|
|
|
|
|
|
|
public logMigration(message: string) {
|
|
|
|
sqlLogger.info(message);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-06-29 23:12:00 +09:00
|
|
|
export const entities = [
|
2020-01-30 04:37:25 +09:00
|
|
|
Announcement,
|
|
|
|
AnnouncementRead,
|
2019-06-29 23:12:00 +09:00
|
|
|
Meta,
|
|
|
|
Instance,
|
|
|
|
App,
|
|
|
|
AuthSession,
|
|
|
|
AccessToken,
|
|
|
|
User,
|
|
|
|
UserProfile,
|
|
|
|
UserKeypair,
|
|
|
|
UserPublickey,
|
|
|
|
UserList,
|
|
|
|
UserListJoining,
|
|
|
|
UserGroup,
|
|
|
|
UserGroupJoining,
|
2020-02-13 02:17:54 +09:00
|
|
|
UserGroupInvitation,
|
2019-06-29 23:12:00 +09:00
|
|
|
UserNotePining,
|
2019-07-03 20:18:07 +09:00
|
|
|
UserSecurityKey,
|
2019-07-22 10:15:00 +09:00
|
|
|
UsedUsername,
|
2019-07-03 20:18:07 +09:00
|
|
|
AttestationChallenge,
|
2019-06-29 23:12:00 +09:00
|
|
|
Following,
|
|
|
|
FollowRequest,
|
|
|
|
Muting,
|
|
|
|
Blocking,
|
|
|
|
Note,
|
|
|
|
NoteFavorite,
|
|
|
|
NoteReaction,
|
|
|
|
NoteWatching,
|
|
|
|
NoteUnread,
|
|
|
|
Page,
|
|
|
|
PageLike,
|
|
|
|
Log,
|
|
|
|
DriveFile,
|
|
|
|
DriveFolder,
|
|
|
|
Poll,
|
|
|
|
PollVote,
|
|
|
|
Notification,
|
|
|
|
Emoji,
|
|
|
|
Hashtag,
|
|
|
|
SwSubscription,
|
|
|
|
AbuseUserReport,
|
|
|
|
RegistrationTicket,
|
|
|
|
MessagingMessage,
|
|
|
|
Signin,
|
2019-07-14 03:18:45 +09:00
|
|
|
ModerationLog,
|
2020-01-30 04:37:25 +09:00
|
|
|
Clip,
|
|
|
|
ClipNote,
|
|
|
|
Antenna,
|
|
|
|
AntennaNote,
|
2020-02-18 08:41:32 +09:00
|
|
|
PromoNote,
|
|
|
|
PromoRead,
|
2019-06-29 23:12:00 +09:00
|
|
|
ReversiGame,
|
|
|
|
ReversiMatching,
|
2020-05-10 18:42:31 +09:00
|
|
|
Relay,
|
2020-07-27 13:34:20 +09:00
|
|
|
MutedNote,
|
2019-06-29 23:12:00 +09:00
|
|
|
...charts as any
|
|
|
|
];
|
|
|
|
|
2020-03-28 18:28:21 +09:00
|
|
|
export function initDb(justBorrow = false, sync = false, forceRecreate = false) {
|
2020-01-09 14:35:04 +09:00
|
|
|
if (!forceRecreate) {
|
|
|
|
try {
|
|
|
|
const conn = getConnection();
|
|
|
|
return Promise.resolve(conn);
|
|
|
|
} catch (e) {}
|
|
|
|
}
|
2019-04-07 21:50:36 +09:00
|
|
|
|
2020-03-29 14:08:19 +09:00
|
|
|
const log = process.env.NODE_ENV != 'production';
|
2020-03-28 18:28:21 +09:00
|
|
|
|
2019-04-07 21:50:36 +09:00
|
|
|
return createConnection({
|
|
|
|
type: 'postgres',
|
|
|
|
host: config.db.host,
|
|
|
|
port: config.db.port,
|
|
|
|
username: config.db.user,
|
|
|
|
password: config.db.pass,
|
|
|
|
database: config.db.db,
|
2019-05-24 03:26:56 +09:00
|
|
|
extra: config.db.extra,
|
2019-04-07 21:50:36 +09:00
|
|
|
synchronize: process.env.NODE_ENV === 'test' || sync,
|
|
|
|
dropSchema: process.env.NODE_ENV === 'test' && !justBorrow,
|
2019-06-11 20:49:08 +09:00
|
|
|
cache: !config.db.disableCache ? {
|
2019-05-25 08:35:16 +09:00
|
|
|
type: 'redis',
|
|
|
|
options: {
|
|
|
|
host: config.redis.host,
|
|
|
|
port: config.redis.port,
|
2019-07-07 07:59:59 +09:00
|
|
|
password: config.redis.pass,
|
2019-12-15 03:34:11 +09:00
|
|
|
prefix: `${config.redis.prefix}:query:`,
|
2019-07-07 07:59:59 +09:00
|
|
|
db: config.redis.db || 0
|
2019-05-25 08:35:16 +09:00
|
|
|
}
|
2019-06-11 20:49:08 +09:00
|
|
|
} : false,
|
2019-04-17 16:45:31 +09:00
|
|
|
logging: log,
|
|
|
|
logger: log ? new MyCustomLogger() : undefined,
|
2019-06-29 23:12:00 +09:00
|
|
|
entities: entities
|
2019-04-07 21:50:36 +09:00
|
|
|
});
|
|
|
|
}
|