index.js

const SQLiteCrud = require('sqlite3-promisify');
const ChromeCookiePath = require('chrome-cookie-path');
const { ValuesPointer, SelectFields, QueryValues } = require('genql');

const { CONSTANTS, Util } = require('./lib');


function Chrome(path) {
  this.tableName = CONSTANTS['TABLE_NAME'];
  this.cookies = require(CONSTANTS['MODEL_PATH']);
  this.fields =  SelectFields.get(this.cookies);
  this.sqliteCrud = new SQLiteCrud(path || ChromeCookiePath.get());
}


Chrome.prototype.setCookie = async function(cookies) {
  const columnsCount = this.cookies.columns.length;
  const insertQuery = `INSERT INTO ${this.tableName} (${this.fields.join(', ')}) VALUES (${ValuesPointer.get(columnsCount)})`;

  const queries = [];
  for (let cookie of cookies) {
    const values = QueryValues.get(this.cookies, cookie);
    queries.push(this.sqliteCrud.run(insertQuery, values));
  }
  
  return await Promise.all(queries);
};


Chrome.prototype.getCookie = async function(domain) {
  const query = `SELECT * from ${this.tableName} WHERE host_key like '%${domain}%'`;
  return Util.parse(this.cookies, await this.sqliteCrud.all(query));
};


Chrome.prototype.removeCookie = async function(domain) {
  const query = `DELETE from ${this.tableName} WHERE host_key like '%${domain}%'`;
  return await this.sqliteCrud.run(query);
};


module.exports = Chrome;