export default function mongooseDeleteFix(schema, options) {
  schema.methods.restore = async function (callback) {
    this.deleted = false;
    this.deletedAt = undefined;
    this.deletedBy = undefined;

    if (options.validateBeforeRestore === false) {
      try {
        await this.save({ validateBeforeSave: false });
      } catch (err) {
        return callback(err);
      }

      return this;
    }

    try {
      await this.save();
    } catch (err) {
      return callback(err);
    }

    return this;
  };

  schema.methods.delete = async function (deletedBy, cb) {
    if (typeof deletedBy === "function") {
      cb = deletedBy;
      deletedBy = null;
    }

    this.deleted = true;

    if (schema.path("deletedAt")) {
      this.deletedAt = new Date();
    }

    if (schema.path("deletedBy")) {
      this.deletedBy = deletedBy;
    }

    if (options.validateBeforeDelete === false) {
      try {
        await this.save({ validateBeforeSave: false });
      } catch (err) {
        return cb(err);
      }

      return this;
    }

    try {
      await this.save();
    } catch (err) {
      return cb(err);
    }

    return this;
  };
}