yaaiisDatabase.js

const SQLite = require('sqlite3').verbose();
// import SQLite from 'sqlite3';

const { Sequelize, Model, DataTypes } = require('sequelize');

// 'sqlite:yaaiis-database.sqlite'
const sequelize = new Sequelize('database', 'username', 'password', {
    dialect: 'sqlite',
    storage: './yaaiis-database.sqlite',
    dialectOptions: {
        // Your sqlite3 options here
        // for instance, this is how you can configure the database opening mode:
        mode: SQLite.OPEN_READWRITE | SQLite.OPEN_CREATE | SQLite.OPEN_FULLMUTEX,
    },
});

const Image = sequelize.define('Image',
    {
        hash: {
            type:DataTypes.TEXT,
            primaryKey: true
        },
        model: DataTypes.TEXT,
        modelHash: DataTypes.TEXT,
        sampler: DataTypes.TEXT,
        prompt: DataTypes.TEXT,
        mtime: DataTypes.DATE,
        ctime: DataTypes.DATE,
        generationMetadata: DataTypes.JSON,
        paths:  DataTypes.JSON,
        stats:  DataTypes.JSON,
    });

const Tag = sequelize.define('Tag',
    {
        uuid: {
            type: DataTypes.UUID,
            defaultValue: DataTypes.UUIDV4,
            primaryKey: true
        },
        name: {
            type: DataTypes.TEXT,
            allowNull: false,
            unique: true
        },
    });

const ImageTag = sequelize.define('ImageTag',
    {
        uuid: {
            type: DataTypes.UUID,
            primaryKey: true
        },
        hash: {
            type: DataTypes.TEXT,
            primaryKey: true
        },
    });

Tag.belongsToMany(Image, {through:'ImageTag', foreignKey: "uuid",})
Image.belongsToMany(Tag, { through: 'ImageTag', foreignKey: "hash", });

const models = {Image, Tag, ImageTag};


let _me;
const yaaiisDatabase = {

    get : async () => {
        if (_me) return _me;

        try {
            await Image.sync();
            await Tag.sync();
            await sequelize.sync();

            _me = models;
            _me.get = this.get;
            return _me;
        } catch (e) {
            console.error(e);
        }

    }
}


module.exports = {yaaiisDatabase};