mirror of
https://github.com/hcengineering/platform.git
synced 2025-03-31 12:38:27 +00:00
80 lines
2.9 KiB
TypeScript
80 lines
2.9 KiB
TypeScript
//
|
|
// Copyright © 2024 Hardcore Engineering Inc.
|
|
//
|
|
// Licensed under the Eclipse Public License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License. You may
|
|
// obtain a copy of the License at https://www.eclipse.org/legal/epl-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
//
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
import { MongoClientReference, getMongoClient } from '@hcengineering/mongo'
|
|
import { Collection, Db, MongoClient, ObjectId, UpdateFilter, WithId } from 'mongodb'
|
|
import { WorkspaceInfoRecord } from '@hcengineering/server-ai-bot'
|
|
import { Doc, Ref, SortingOrder } from '@hcengineering/core'
|
|
|
|
import config from './config'
|
|
import { HistoryRecord } from './types'
|
|
|
|
const clientRef: MongoClientReference = getMongoClient(config.MongoURL)
|
|
let client: MongoClient | undefined
|
|
|
|
export const getDB = (() => {
|
|
return async () => {
|
|
if (client === undefined) {
|
|
client = await clientRef.getClient()
|
|
}
|
|
|
|
return client.db(config.ConfigurationDB)
|
|
}
|
|
})()
|
|
|
|
export const closeDB: () => Promise<void> = async () => {
|
|
clientRef.close()
|
|
}
|
|
|
|
export class DbStorage {
|
|
private readonly workspacesInfoCollection: Collection<WorkspaceInfoRecord>
|
|
private readonly historyCollection: Collection<HistoryRecord>
|
|
|
|
constructor (private readonly db: Db) {
|
|
this.workspacesInfoCollection = this.db.collection<WorkspaceInfoRecord>('workspacesInfo')
|
|
this.historyCollection = this.db.collection<HistoryRecord>('history')
|
|
}
|
|
|
|
async addHistoryRecord (record: HistoryRecord): Promise<ObjectId> {
|
|
return (await this.historyCollection.insertOne(record)).insertedId
|
|
}
|
|
|
|
async getHistoryRecords (workspace: string, objectId: Ref<Doc>): Promise<WithId<HistoryRecord>[]> {
|
|
return await this.historyCollection
|
|
.find({ workspace, objectId }, { sort: { timestamp: SortingOrder.Ascending } })
|
|
.toArray()
|
|
}
|
|
|
|
async removeHistoryRecords (_ids: ObjectId[]): Promise<void> {
|
|
await this.historyCollection.deleteMany({ _id: { $in: _ids } })
|
|
}
|
|
|
|
async getActiveWorkspaces (): Promise<WorkspaceInfoRecord[]> {
|
|
return await this.workspacesInfoCollection.find({ active: true }).toArray()
|
|
}
|
|
|
|
async inactiveWorkspace (workspace: string): Promise<void> {
|
|
await this.workspacesInfoCollection.updateOne({ workspace }, { $set: { active: false } })
|
|
}
|
|
|
|
async getWorkspace (workspace: string): Promise<WorkspaceInfoRecord | undefined> {
|
|
return (await this.workspacesInfoCollection.findOne({ workspace })) ?? undefined
|
|
}
|
|
|
|
async updateWorkspace (workspace: string, update: UpdateFilter<WorkspaceInfoRecord>): Promise<void> {
|
|
await this.workspacesInfoCollection.updateOne({ workspace }, update)
|
|
}
|
|
}
|