user.h

Go to the documentation of this file.

#ifndef USER_H_
#define USER_H_

#include <string>
#include <vector>

#include "utils.h"
#include "interfaces.h"
#include "job.h"
#include "bytebuffer.h"
#include "menu.h"

enum UserType
{
    USER_TYPE_ACADEMIC,
    USER_TYPE_ENTERPRISE,
};

static char* UserTypeStr[] = { "Academic", "Enterprise" };


class User : public ISave, public IPrint
{
public:
    virtual ~User() { }; 

    uint GetId() const { return _id; } 
    void SetId(uint val) { _id = val; } 

    void SetName(const std::string& name); 
    const std::string& GetName() const { return _name; } 

    virtual bool Save(ByteBuffer& bb) const override; 
    static User* Load(ByteBuffer& bb); 

    virtual bool CanCreateJob(const Job* job) = 0; 
    virtual void CreatedJob(const Job* job) = 0; 

    virtual void Print(std::ostream& os = std::cout) const override;
    static void PrintHeader(std::ostream& os = std::cout);

    static Menu* GetMenu() { return _menu; } 

protected:
    User(uint id) : _id(id) { }
    User(const std::string& name); 

    static Menu* _menu; 
    static uint _maxNameLength; 

private:
    std::string _name; 
    uint _id; 
};


class AcademicUser : public User
{
public:
    AcademicUser(uint id) : User(id) {}
    AcademicUser(const std::string& name) : User( name), _jobCount(0) {}
    AcademicUser(const std::string& name, uint jobCount) : User(name), _jobCount(jobCount) {}

    uint GetJobCount() const { return _jobCount; } 

    bool Save(ByteBuffer& bb) const override;

    void CreatedJob(const Job* job) override { if (job) _jobCount++; }

    bool CanCreateJob(const Job* job) override { return !!job; }

    void Print(std::ostream& os = std::cout) const override;

private:
    uint _jobCount; 
};


class EnterpriseUser : public User
{
public:
    EnterpriseUser(const std::string& name, double budget) : User(name), _budget(budget) {}

    double GetBudget() const { return _budget; } 

    void SetBudget(double val) { _budget = val; } 

    bool Save(ByteBuffer& bb) const override;

    void CreatedJob(const Job* job) override;

    bool CanCreateJob(const Job* job) override;

    void Print(std::ostream& os = std::cout) const override;

private:
    double _budget; 
};

#endif // USER_H_