GitBot.hpp

Go to the documentation of this file.

#pragma once
#include <chrono>
#include <condition_variable>
#include <cstdint>
#include <tgbotxx/tgbotxx.hpp>
#include <type_traits>
#include "api/GitApi.hpp"
#include <cpr/threadpool.h>
 
class GitBot : public tgbotxx::Bot {
  using ChatId = decltype(tgbotxx::Chat::id);
  using UserId = decltype(tgbotxx::User::id);
 
public:
  GitBot();
  virtual ~GitBot() = default;
 
private:
  bool middleware(const tgbotxx::Ptr<tgbotxx::Message> &message);
 
  void onStart() override;
  void onStop() override;
  void onCommand(const tgbotxx::Ptr<tgbotxx::Message> &message) override;
  void onNonCommandMessage(const tgbotxx::Ptr<tgbotxx::Message> &message) override;
  void onCallbackQuery(const tgbotxx::Ptr<tgbotxx::CallbackQuery> &callbackQuery) override;
  void onLongPollError(const std::string &errorMessage, tgbotxx::ErrorCode errorCode) override;
 
  void onStartCommand(const tgbotxx::Ptr<tgbotxx::Message> message);
  void onWatchRepoCommand(const tgbotxx::Ptr<tgbotxx::Message> message);
  void onUnwatchRepoCommand(const tgbotxx::Ptr<tgbotxx::Message> message);
  void onMyReposCommand(const tgbotxx::Ptr<tgbotxx::Message> message);
 
  void onUserBlockedBot(const UserId userId);
 
private:
  void watchDog();
 
  void alertUserRepositoryStarsChange(UserId userId, const std::string& repositoryName, std::int64_t oldStarsCount, std::int64_t newStarsCount);
  void alertUserRepositoryWatchersChange(UserId userId, const std::string& repositoryName, std::int64_t oldWatchersCount, std::int64_t newWatchersCount);
  void alertUserRepositoryIssuesChange(UserId userId, const std::string& repositoryName, std::int64_t oldIssuesCount, std::int64_t newIssuesCount);
  void alertUserRepositoryForksChange(UserId userId, const std::string& repositoryName, std::int64_t oldForksCount, std::int64_t newForksCount);
  void alertUserRepositoryPullRequestsChange(UserId userId, const std::string& repositoryName, std::int64_t oldPullsCount, std::int64_t newPullsCount);
 
private:
  void notifyAdmin(const std::string& msg, const std::source_location& loc = std::source_location::current());
 
private:
  void safeSendMessage(UserId userId, std::string messageText,
                       std::int32_t messageThreadId = 0,
                       const std::string &parseMode = "",
                       const std::vector<tgbotxx::Ptr<tgbotxx::MessageEntity>> &entities = std::vector<tgbotxx::Ptr<tgbotxx::MessageEntity>>(),
                       bool disableWebPagePreview = false,
                       bool disableNotification = false,
                       bool protectContent = false,
                       std::int32_t replyToMessageId = 0,
                       bool allowSendingWithoutReply = false,
                       const tgbotxx::Ptr<tgbotxx::IReplyMarkup> &replyMarkup = nullptr);
 
  void safeSendLargeMessage(UserId userId, const std::string &messageText);
 
public:
  static bool isRepositoryFullName(const std::string &str);
 
  static bool isRepositoryFullURL(const std::string &str, std::string &fullname);
 
private:
  UserId m_adminUserId{}; 
  std::unique_ptr<GitApi> m_gitApi; 
  std::unique_ptr<std::thread> m_watchdogThread; 
  std::mutex m_sleepMutex; 
  std::atomic<bool> m_watchdogRunning; 
  std::condition_variable m_watchdogCv; 
  std::unique_ptr<cpr::ThreadPool> m_threadPool; 
 
  inline static constexpr std::size_t kTelegramMessageMax = 4096; 
  inline static constexpr std::size_t kMaxWatchListRepositories = 25; 
};