TransferShares.h

#ifndef TRANSFERSHARES_H_
#define TRANSFERSHARES_H_

#include <map>

#include <arc/credential/VOMSUtil.h>

#include "DTR.h"

namespace DataStaging {


  class TransferSharesConf {

   public:

    enum ShareType {
      USER,
      VO,
      GROUP,
      ROLE,
      NONE
    };

  private:

    std::map<std::string, int> ReferenceShares;

    ShareType shareType;

    std::string extract_user_share(const Arc::Credential& cred){
      return getCredentialProperty(cred, "dn");
    }

    std::string extract_vo_share(const Arc::Credential& cred){
      return getCredentialProperty(cred, "voms:vo");
    }

    std::string extract_group_share(const Arc::Credential& cred){
      return getCredentialProperty(cred, "voms:group");
    }

    std::string extract_role_share(const Arc::Credential& cred){
      return getCredentialProperty(cred, "voms:role");
    }

  public:

    TransferSharesConf(const std::string& type, const std::map<std::string, int>& ref_shares);

    TransferSharesConf();

    void set_share_type(const std::string& type);

    void set_reference_share(const std::string& RefShare, int Priority);

    void set_reference_shares(const std::map<std::string, int>& shares);

    bool is_configured(const std::string& ShareToCheck);

    int get_basic_priority(const std::string& ShareToCheck);

    std::string conf() const;

    std::string extract_share_info(DTR_ptr DTRToExtract);
  };



  class TransferShares {

   private:

    TransferSharesConf conf;

    std::map<std::string, int> ActiveShares;

    std::map<std::string, int> ActiveSharesSlots;

   public:

    TransferShares() {};

    TransferShares(const TransferSharesConf& shares_conf);

    ~TransferShares(){};

    void set_shares_conf(const TransferSharesConf& share_conf);


    void calculate_shares(int TotalNumberOfSlots);

    void increase_transfer_share(const std::string& ShareToIncrease);
    void decrease_transfer_share(const std::string& ShareToDecrease);


    void decrease_number_of_slots(const std::string& ShareToDecrease);

    bool can_start(const std::string& ShareToStart);

    std::map<std::string, int> active_shares() const;

  }; // class TransferShares

} // namespace DataStaging

#endif /* TRANSFERSHARES_H_ */