serial/include/serial.h

#ifndef SERIAL_H
#define SERIAL_H

#include <algorithm>
#include <chrono>
#include <expected>
#include <format>
#include <functional>
#include <iostream>
#include <ranges>
#include <sstream>
#include <string>
#include <string_view>
#include <thread>
#include <type_traits>
#include <vector>

#include "serialib.h"

using namespace std::literals::chrono_literals;
namespace ranges = std::ranges;
namespace views = std::views;

namespace serial {
template <typename... Types> struct _StrongType {};

template <typename T> struct _StrongType<T> {
    using Type = T;
};

template <typename Ta, typename Tb> struct _StrongType<Ta, Tb> {
    using Type = decltype(true ? std::declval<Ta>() : std::declval<Tb>());
};

template <typename T, typename... Types> struct _StrongType<T, Types...> {
    using Type =
        typename _StrongType<T, typename _StrongType<Types...>::Type>::Type;
};

template <typename... Types>
using _strongType_t = typename _StrongType<Types...>::Type;

template <typename... Args> struct _IsCastable {};

template <typename T> struct _IsCastable<T> {
    static constexpr bool value = true;
    using Type = T;
};

template <typename T, typename U> struct _IsCastable<T, U> {
    using __TRUE = char;
    using __FALSE = struct {
        char _[2];
    };
    static consteval __TRUE __TEST(U);
    static consteval __FALSE __TEST(...);
    static constexpr bool value =
        sizeof(__TEST(std::declval<T>())) == sizeof(__TRUE);
    using Type = std::conditional_t<value, U, void>;
};

template <typename T, typename... Args> struct _IsCastable<T, Args...> {
    static constexpr bool value =
        _IsCastable<T, typename _IsCastable<Args...>::Type>::value;
    using Type =
        std::conditional_t<value, typename _IsCastable<Args...>::Type, void>;
};

template <typename T>
concept _SupportString = requires {
    _IsCastable<T, const char *, char *, std::string, std::string_view>::value;
};

template <_SupportString T> std::string _to_string(T &&str) {
    if constexpr (std::is_same_v<std::decay_t<T>, std::string>) {
        return str;
    } else if constexpr (std::is_same_v<std::decay_t<T>, std::string_view>) {
        return std::move(std::string(str.data()));
    } else if constexpr (std::is_same_v<std::decay_t<T>,  const char*>) {
        return std::string(str);
    }
}

static std::vector<std::string> GetUsbPorts() {
    std::vector<std::string> portArray;
    std::string comname;
    std::string showname;
    ranges::for_each(views::iota(1, 256), [&](int i) {
        std::format_to(std::back_inserter(comname), "\\\\.\\COM{}", i);
        std::format_to(std::back_inserter(showname), "COM{}", i);
        const HANDLE m_handle = ::CreateFileA(
            comname.c_str(), static_cast<DWORD>(GENERIC_WRITE) | GENERIC_READ,
            0U, nullptr, OPEN_EXISTING, 0U, nullptr);
        if (m_handle != INVALID_HANDLE_VALUE) {
            portArray.push_back(showname);
            CloseHandle(m_handle);
        }
        comname.clear();
        showname.clear();
    });
    return portArray;
}

enum class [[maybe_unused]] SerialErrorCode {
    SUCCESS,
    TIMEOUT,
    SETTIMEOUTERROR,
    WRITEINGERROR,
    READINGERROR,
};

class Serial {
  private:
    serialib ser;
    const char *endChar = "\r\n";
    std::function<void(const std::string &)> logCallBack;
    bool removeEcho = true; // 剔除回显

  public:
    Serial() = default;
    ~Serial() { CloseDevice(); }
    Serial(const Serial &other) = delete;
    Serial(Serial &&other) = delete;
    Serial &operator=(const Serial &other) = delete;
    Serial &operator=(Serial &&other) = delete;

    auto SetRemoveEcho(bool remove) { removeEcho = remove; }
    auto SetDtr(bool flag) { return flag ? ser.setDTR() : ser.clearDTR(); }
    auto SetRts(bool flag) { return flag ? ser.setRTS() : ser.clearRTS(); }

    static std::string GetTimeNow() {
        auto now = std::chrono::system_clock::now();
        auto now_c = std::chrono::system_clock::to_time_t(now);
        char buffer[32];
        auto _ = ctime_s(buffer, 32, &now_c);
        return buffer;
    }

    bool IsOpen() { return ser.isDeviceOpen(); }

    template <_SupportString T>
    bool OpenDevice(T portName, unsigned int bauds = 115200,
                    int delayTime = 0) {
        std::string reallyPortName;
        std::format_to(std::back_inserter(reallyPortName), "\\\\.\\{}",
                       portName);
        std::this_thread::sleep_for(std::chrono::milliseconds(delayTime));
        if (ser.isDeviceOpen())
            return true;
        int code = ser.openDevice(reallyPortName.c_str(), bauds);
        if (code == 1) {
            return true;
        } else {
            return false;
        }
    }

    void Log(const std::string &log) const {
        if (logCallBack) {
            auto msg = GetTimeNow() + " " + log;
            logCallBack(msg);
        }
    }

    void
    SetLogCallBack(const std::function<void(const std::string &)> &callBack) {
        logCallBack = callBack;
    }
    void CloseDevice() {
        if (!ser.isDeviceOpen())
            return;
        ser.closeDevice();
    }

    template <_SupportString T>
    std::expected<std::string, SerialErrorCode>
    DelayGetResponse(int delayTime, T command, int timeout = 50) {
        std::this_thread::sleep_for(std::chrono::milliseconds(delayTime));
        return GetAtResponse(command, timeout);
    }

    template <int repeatTime = 5, int delayTime = 200, int timeout = 200,
              _SupportString T, _SupportString... Args>
    bool GetAtUntilRepeat(T command, Args... args) {
        std::stringstream ss;
        int i = 0;
        while (i++ < repeatTime) {
            ss.str("");
            ss << "Count: " << i << "\n";
            Log(ss.str());
            std::this_thread::sleep_for(std::chrono::milliseconds(delayTime));
            if (GetAtUntil<timeout>(command, args...))
                return true;
        }
        return false;
    }

    template <_SupportString T>
    std::expected<std::string, SerialErrorCode>
    GetAtResponse(T command, int timeout = 50) {
        ser.flushReceiver();
        std::string reallyCommand = std::string(command) + endChar;
        ser.writeString(reallyCommand.c_str());
        Log("Send: " + reallyCommand);
        std::this_thread::sleep_for(10ms);
        auto availableSize = ser.available();
        auto buffer = std::make_unique<char[]>(availableSize + 1);
        std::memset(buffer.get(), 0, availableSize);
        auto size = ser.readBytes(buffer.get(), availableSize, timeout);

        if (size > 0) {
            buffer[size] = '\0';
            std::string response = std::string(buffer.get());
            Log("Receive: " + response);
            if (removeEcho)
                response.replace(0, reallyCommand.length(), "");
            return response;
        }
        return std::unexpected(SerialErrorCode::TIMEOUT);
    }

    template <_SupportString T>
    auto GetAtResponseRepeat(T command, int timeout = 200, int repeatTime = 1)
        -> void {
        for (int i = 0; i <= repeatTime; i++) {
            auto _ = GetAtResponse(command, timeout);
        }
    }

    template <int timeout = 200, _SupportString T, _SupportString... Args>
    bool GetAtUntil(T command, Args... expect) {
        auto endTime = std::chrono::system_clock::now() +
                       std::chrono::milliseconds(timeout);
        ser.flushReceiver();
        std::string reallyCommand = std::string(command) + endChar;
        ser.writeString(reallyCommand.c_str());
        Log("Send : " + reallyCommand);
        while (std::chrono::system_clock::now() < endTime) {
            std::this_thread::sleep_for(10ms);
            auto availableSize = ser.available();
            auto buffer = std::make_unique<char[]>(availableSize + 1);
            auto size = ser.readBytes(buffer.get(), availableSize, timeout);
            buffer[size] = '\0';
            auto str = std::string(buffer.get());
            if (size > 0)
                Log("Receive: " + str);
            if (((str.find(expect) != std::string::npos) && ...)) {
                return true;
            }
        }
        return false;
    }
};
} // namespace serial
#endif // SERIAL_H