iceshrimp-161sh/packages/backend/src/db/sonic.ts

import * as SonicChannel from "sonic-channel";
import { dbLogger } from "./logger.js";

import config from "@/config/index.js";

const logger = dbLogger.createSubLogger("sonic", "gray", false);

logger.info("Connecting to Sonic");

const handlers = (type: string): SonicChannel.Handlers => (
	{
		connected: () => {
			logger.succ(`Connected to Sonic ${type}`);
		},
		disconnected: (error) => {
			logger.warn(`Disconnected from Sonic ${type}, error: ${error}`);
		},
		error: (error) => {
			logger.warn(`Sonic ${type} error: ${error}`);
		},
		retrying: () => {
			logger.info(`Sonic ${type} retrying`);
		},
		timeout: () => {
			logger.warn(`Sonic ${type} timeout`);
		},
	}
)

const hasConfig =
	config.sonic
	&& ( config.sonic.host
		|| config.sonic.port
		|| config.sonic.auth
	)

const host = hasConfig ? config.sonic.host ?? "localhost" : "";
const port = hasConfig ? config.sonic.port ?? 1491 : 0;
const auth = hasConfig ? config.sonic.auth ?? "SecretPassword" : "";
const collection = hasConfig ? config.sonic.collection ?? "main" : "";
const bucket = hasConfig ? config.sonic.bucket ?? "default" : "";

export default hasConfig
	? {
			search: new SonicChannel.Search({host, port, auth}).connect(handlers("search")),
			ingest: new SonicChannel.Ingest({host, port, auth}).connect(handlers("ingest")),

			collection,
			bucket,
		}
	: null;