pigallery2/backend/model/memory/UserManager.ts

///<reference path="flat-file-db.ts"/>
import {UserDTO, UserRoles} from "../../../common/entities/UserDTO";
import {IUserManager} from "../interfaces/IUserManager";
import {ProjectPath} from "../../ProjectPath";
import {Utils} from "../../../common/Utils";
import * as flatfile from "flat-file-db";
import * as path from "path";
import {PasswordHelper} from "../PasswordHelper";


export class UserManager implements IUserManager {
  private db: any = null;

  generateId(): string {
    function s4() {
      return Math.floor((1 + Math.random()) * 0x10000)
        .toString(16)
        .substring(1);
    }

    return s4() + s4() + s4() + s4();
  }

  constructor() {
    this.db = flatfile.sync(path.join(ProjectPath.Root, 'users.db'));

    if (!this.db.has("idCounter")) {
      console.log("creating counter");
      this.db.put("idCounter", 1);
    }

    if (!this.db.has("users")) {
      this.db.put("users", []);
      //TODO: remove defaults
      this.createUser(<UserDTO>{name: "developer", password: "developer", role: UserRoles.Developer});
      this.createUser(<UserDTO>{name: "admin", password: "admin", role: UserRoles.Admin});
      this.createUser(<UserDTO>{name: "user", password: "user", role: UserRoles.User});
      this.createUser(<UserDTO>{name: "guest", password: "guest", role: UserRoles.LimitedGuest});
    }


  }


  public async findOne(filter: any) {
    const result = await this.find(filter);

    if (result.length == 0) {
      throw "UserDTO not found";
    }
    return result[0];
  }

  public async find(filter: any) {
    let pass = filter.password;
    delete filter.password;
    const users = await this.db.get("users");
    let i = users.length;
    while (i--) {
      if (pass && !(await PasswordHelper.comparePassword(pass, users[i].password))) {
        users.splice(i, 1);
        continue;
      }
      if (Utils.equalsFilter(users[i], filter) == false) {
        users.splice(i, 1);
      }
    }
    return users;
  }

  public async createUser(user: UserDTO) {
    user.id = parseInt(this.db.get("idCounter")) + 1;
    this.db.put("idCounter", user.id);
    let users = this.db.get("users");
    user.password = await PasswordHelper.cryptPassword(user.password);
    users.push(user);
    this.db.put("users", users);

    return user;
  }

  public deleteUser(id: number) {
    let deleted = this.db.get("users").filter((u: UserDTO) => u.id == id);
    let users = this.db.get("users").filter((u: UserDTO) => u.id != id);
    this.db.put("users", users);
    if (deleted.length > 0) {
      return deleted[0];
    }
    return null;
  }

  public async changeRole(id: number, newRole: UserRoles): Promise<UserDTO> {

    let users: Array<UserDTO> = this.db.get("users");

    for (let i = 0; i < users.length; i++) {
      if (users[i].id == id) {
        users[i].role = newRole;
        this.db.put("users", users);
        return users[i];
      }
    }
  }

  public async changePassword(request: any) {
    throw new Error("not implemented"); //TODO: implement
  }

}