class CodesRepository
{
  constructor(store)
  {
    var self = this;
    self.store = store;
  }


  async init()
  {
    var self = this;

    return new Promise((resolve, reject) =>
    {
      // Initialize database if empty
      self.store.count({}, (err, count) =>
      {
        if (err)
        {
          reject(err);
          return;
        }

        if (count == 0)
          self.addUser('admin', null, 'test', null);

        resolve();
      });
    });
  }


  async findCodeUserId(code)
  {
    var self = this;

    return new Promise((resolve, reject) =>
    {
      self.store.findOne({ _id: code }, (err, doc) =>
      {
        if (err)
        {
          reject(err);
          return;
        }

        resolve(doc !== null ? doc.userId : null);
      });
    });
  }


  async addCode(userId, expiration)
  {
    var self = this;

    return await retry(async bail =>
    {
      var code = shortid.generate();

      if ((await self.findCodeUserId(code)) !== null)
        throw new Error('Code ' + code + ' already exists');

      self.store.insert({
        _id: code,
        userId: userId,
        created: new Date(),
        expiration: expiration
      })

      return code;
    }, {
      retries: 100,
      minTimeout: 0,
      maxTimeout: 0
    });
  }
}


module.exports = CodesRepository;