#pragma once

#include <sstream>

#include "Layer.h"

/// @file

/**

 * \namespace pcpp

 * \brief The main namespace for the PcapPlusPlus lib

 */

namespace pcpp

{

  /**

   * Class for single command text based protocol (FTP, SMTP) messages

   */

  class SingleCommandTextProtocol : public Layer

  {

  private:

    size_t getArgumentFieldOffset() const;

    void setDelimiter(bool hyphen);

    bool hyphenRequired(const std::string& value);

  protected:

    SingleCommandTextProtocol(uint8_t *data, size_t dataLen, Layer *prevLayer, Packet *packet) : Layer(data, dataLen, prevLayer, packet) {};

    SingleCommandTextProtocol(const std::string &command, const std::string &option);

    bool setCommandInternal(std::string value);

    bool setCommandOptionInternal(std::string value);

    std::string getCommandInternal() const;

    std::string getCommandOptionInternal() const;

  public:

    /**

     * Checks if the current message is a multi-line reply. Multi-line messages are indicated with a Hyphen (-) immediately after reply code.

     * @return true If this is a multi-line reply

     * @return false Otherwise

     */

    bool isMultiLine() const;

    /**

     * A static method that takes a byte array and detects whether it is a single command text based message.

     * All single command text based message terminated with single "\r\n".

     * @param[in] data A byte array

     * @param[in] dataSize The byte array size (in bytes)

     * @return True if the data is identified as single command text based message

     */

    static bool isDataValid(const uint8_t *data, size_t dataSize);

  };

} // namespace pcpp