import DB from "../database";
import { ModelDataBase } from "@hibas123/safe_mongo/lib/model";
import { ObjectID } from "mongodb";
import moment = require("moment");

export interface ILoginToken extends ModelDataBase {
   token: string;
   special: boolean;
   user: ObjectID;
   validTill: Date;
   valid: boolean;
   validated: boolean;
   data: any;
   ip: string;
   browser: string;
}
const LoginToken = DB.addModel<ILoginToken>({
   name: "login_token",
   versions: [
      {
         migration: () => {},
         schema: {
            token: { type: String },
            special: { type: Boolean, default: () => false },
            user: { type: ObjectID },
            validTill: { type: Date },
            valid: { type: Boolean },
         },
      },
      {
         migration: (doc: ILoginToken) => {
            doc.validated = true;
         },
         schema: {
            token: { type: String },
            special: { type: Boolean, default: () => false },
            user: { type: ObjectID },
            validTill: { type: Date },
            valid: { type: Boolean },
            validated: { type: Boolean, default: false },
         },
      },
      {
         migration: (doc: ILoginToken) => {
            doc.validated = true;
         },
         schema: {
            token: { type: String },
            special: { type: Boolean, default: () => false },
            user: { type: ObjectID },
            validTill: { type: Date },
            valid: { type: Boolean },
            validated: { type: Boolean, default: false },
            data: { type: "any", optional: true },
            ip: { type: String, optional: true },
            browser: { type: String, optional: true },
         },
      },
   ],
});

export async function CheckToken(
   token: ILoginToken,
   validated: boolean = true
): Promise<boolean> {
   if (!token || !token.valid) return false;
   if (validated && !token.validated) return false;
   if (moment().isAfter(token.validTill)) {
      token.valid = false;
      await LoginToken.save(token);
      return false;
   }
   return true;
}

export default LoginToken;