userrepository.h

Go to the documentation of this file.

/********************************************************************************************
* Twitch Broadcasting SDK
*
* This software is supplied under the terms of a license agreement with Twitch Interactive, Inc. and
* may not be copied or used except in accordance with the terms of that agreement
* Copyright (c) 2012-2016 Twitch Interactive, Inc.
*********************************************************************************************/

#pragma once

#include "twitchsdk/core/types/coretypes.h"
#include "twitchsdk/core/types/errortypes.h"
#include "twitchsdk/core/user/user.h"
#include "twitchsdk/core/timer.h"
#include "twitchsdk/core/coreutilities.h"

#include <memory>
#include <map>


namespace ttv
{
    class TaskRunner;
    class User;
    class UserRepository;
    class GetUserTask;
}


class ttv::UserRepository : public Component
{
public:
    using UserInfoLookupCallback = std::function<void(TTV_ErrorCode ec, const UserInfo& userInfo)>;

public:
    UserRepository();

    std::shared_ptr<User> GetAnonymousUser() const { return mAnonymousUser; }

    std::shared_ptr<User> RegisterUser(UserId userId);

    std::shared_ptr<User> GetUser(UserId userId);

    TTV_ErrorCode GetUsers(std::vector<std::shared_ptr<User>>& users);

    TTV_ErrorCode UnRegisterUser(UserId userId);
    
    TTV_ErrorCode FetchUserInfoById(UserId userId, UserInfoLookupCallback callback);

    TTV_ErrorCode FetchUserInfoByName(const std::string& username, UserInfoLookupCallback callback);
    
    TTV_ErrorCode FetchUserInfoByAuthToken(const std::shared_ptr<OAuthToken>& authToken, UserInfoLookupCallback callback);
    
    TTV_ErrorCode GetUserInfoById(UserId userId, UserInfo& result);
    
    TTV_ErrorCode GetUserInfoByName(const std::string& username, UserInfo& result);
    
    // Component overrides
    virtual TTV_ErrorCode Initialize() override;
    virtual void Update() override;
    virtual TTV_ErrorCode Shutdown() override;
    virtual std::string GetLoggerName() const override;

private:
    struct UserInfoLookupEntry
    {
        std::string username;
        std::shared_ptr<OAuthToken> authToken;
        UserId userId;
        UserInfoLookupCallback callback;
        std::shared_ptr<GetUserTask> task;
        bool forceRefetch;
    };

    // Component overrides
    virtual bool CheckShutdown() override;
    virtual void CompleteShutdown() override;

    TTV_ErrorCode ScheduleUserInfoLookup();
    void CompleteUserInfoLookup(const UserInfo& userInfo, TTV_ErrorCode ec);

    TTV_ErrorCode CacheUserInfo(const UserInfo& userInfo);

    std::map<UserId, std::shared_ptr<User>> mUsers;
    std::vector<std::shared_ptr<User>> mCleanupUsers; 
    std::map<UserId, UserInfo> mUserInfo; 
    std::vector<UserInfoLookupEntry> mPendingUserInfoLookups;
    std::shared_ptr<User> mAnonymousUser;

    bool mLookupInProgress;
};