/[sudobot]/trunk/src/client/Database.ts
ViewVC logotype

Diff of /trunk/src/client/Database.ts

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 86 by rakin, Mon Jul 29 17:28:32 2024 UTC revision 393 by rakin, Mon Jul 29 17:29:59 2024 UTC
# Line 1  Line 1 
1  import { Sequelize } from 'sequelize';  /**
2  import { Database as DB } from 'sqlite3';  * This file is part of SudoBot.
3    *
4    * Copyright (C) 2021-2022 OSN Inc.
5    *
6    * SudoBot is free software; you can redistribute it and/or modify it
7    * under the terms of the GNU Affero General Public License as published by
8    * the Free Software Foundation, either version 3 of the License, or
9    * (at your option) any later version.
10    *
11    * SudoBot is distributed in the hope that it will be useful, but
12    * WITHOUT ANY WARRANTY; without even the implied warranty of
13    * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14    * GNU Affero General Public License for more details.
15    *
16    * You should have received a copy of the GNU Affero General Public License
17    * along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
18    */
19    
20  import DiscordClient from './Client';  import DiscordClient from './Client';
21    import mongoose from "mongoose";
22    
23  export default class Database {  export default class Database {
24      client: DiscordClient;      client: DiscordClient;
     dbpath: string;  
     db: DB;  
     sequelize: Sequelize;  
25    
26      constructor(dbpath: string, client: DiscordClient) {      constructor(client: DiscordClient) {
27          this.client = client;          this.client = client;
         this.dbpath = dbpath;  
   
         this.sequelize = new Sequelize({  
             dialect: 'sqlite',  
             storage: dbpath,  
         });  
   
         this.db = new DB(dbpath, (err) => {  
             if (err) {  
                 console.log(err);  
             }  
         });  
     }  
   
     get(sql: string, params: any[] | Function, callback?: Function) {  
         return this.db.get(sql, params, callback);  
     }  
   
     all(sql: string, params: any[] | Function, callback?: Function) {  
         return this.db.all(sql, params, callback);  
     }  
   
     runAsync(sql: string, paramsOrCallback: any[] | Function = []) {  
         return new Promise<void>((resolve, reject) => {  
             this.db.run(sql, paramsOrCallback, err => {  
                 if (err) {  
                     reject(err);  
                     return;  
                 }  
   
                 resolve();  
             });  
         });  
     }  
   
     getAsync(sql: string, paramsOrCallback: any[] | Function = []): Promise <any> {  
         return new Promise((resolve, reject) => {  
             this.db.get(sql, paramsOrCallback, (err, data) => {  
                 if (err) {  
                     reject(err);  
                     return;  
                 }  
   
                 resolve(data);  
             });  
         });  
     }  
   
     allAsync(sql: string, paramsOrCallback: any[] | Function = []): Promise <any> {  
         return new Promise((resolve, reject) => {  
             this.db.all(sql, paramsOrCallback, (err, data) => {  
                 if (err) {  
                     reject(err);  
                     return;  
                 }  
   
                 resolve(data);  
             });  
         });  
     }  
28    
29      get s(): Sequelize {          mongoose.connect(process.env.MONGO_URI!)
30          return this.sequelize;              .then(() => console.log("Connected to MongoDB"))
31                .catch(console.error);
32      }      }
 };  
33    }
Legend:
Removed from v.86  
changed lines
  Added in v.393
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26