66 lines
1.5 KiB
C++
66 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include <cstdint>
|
|
#include <codecvt>
|
|
#include <fstream>
|
|
#include <locale>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "usenetsearch/SSLConnection.h"
|
|
#include "usenetsearch/TcpConnection.h"
|
|
|
|
namespace usenetsearch {
|
|
|
|
struct UsenetClientException: public UsenetSearchException
|
|
{
|
|
UsenetClientException(int errorCode, const std::string& message):
|
|
UsenetSearchException(errorCode, message){}
|
|
|
|
virtual ~UsenetClientException() = default;
|
|
};
|
|
|
|
struct NntpMessage
|
|
{
|
|
std::uint16_t code;
|
|
std::wstring message;
|
|
};
|
|
|
|
struct NntpListEntry
|
|
{
|
|
std::wstring name;
|
|
std::uint64_t high;
|
|
std::uint64_t low;
|
|
std::uint64_t count;
|
|
std::wstring status;
|
|
};
|
|
std::ostream& operator<<(std::ofstream& out, const NntpListEntry& obj);
|
|
std::ifstream& operator>>(std::ifstream& in, NntpListEntry& obj);
|
|
|
|
class UsenetClient
|
|
{
|
|
std::wstring_convert<std::codecvt_utf8_utf16<wchar_t>> m_conv;
|
|
std::unique_ptr<SSLConnection> m_ssl;
|
|
std::unique_ptr<TcpConnection> m_tcp;
|
|
bool m_useSSL{false};
|
|
|
|
bool IsError(const NntpMessage& msg) const;
|
|
NntpMessage ReadLine();
|
|
void Write(const std::wstring& message);
|
|
std::wstring ReadUntil(const std::wstring& deliminator);
|
|
|
|
public:
|
|
|
|
void Authenticate(const std::wstring& user, const std::wstring& password);
|
|
void Connect(
|
|
const std::string& host,
|
|
std::uint16_t port,
|
|
bool useSSL = false
|
|
);
|
|
std::vector<NntpListEntry> List();
|
|
|
|
};
|
|
|
|
} // namespace usenetsearch
|