yatemime.h

#ifndef __YATEMIME_H
#define __YATEMIME_H

#ifndef __cplusplus
#error C++ is required
#endif

#include <yateclass.h>

namespace TelEngine {

class YATE_API MimeHeaderLine : public NamedString
{
public:
    MimeHeaderLine(const char* name, const String& value, char sep = 0);

    MimeHeaderLine(const MimeHeaderLine& original, const char* newName = 0);

    virtual ~MimeHeaderLine();

    virtual void* getObject(const String& name) const;

    virtual MimeHeaderLine* clone(const char* newName = 0) const;

    virtual void buildLine(String& line, bool header = true) const;

    inline MimeHeaderLine& operator=(const char* value)
        { NamedString::operator=(value); return *this; }

    inline const ObjList& params() const
        { return m_params; }

    inline char separator() const
        { return m_separator; }

    void setParam(const char* name, const char* value = 0);

    void delParam(const char* name);

    const NamedString* getParam(const char* name) const;

    static void addQuotes(String& str, bool force = false);

    static void delQuotes(String& str, bool force = false);

    static String quote(const String& str, bool force = false);

    static String unquote(const String& str, bool force = false);

    static int findSep(const char* str, char sep, int offs = 0);

    static void buildHeaders(String& buf, const ObjList& headers);

protected:
    ObjList m_params;                    // Header list of parameters
    char m_separator;                    // Parameter separator
private:
    void operator=(const MimeHeaderLine&); // no assignment
};

class YATE_API MimeAuthLine : public MimeHeaderLine
{
public:
    MimeAuthLine(const char* name, const String& value);

    MimeAuthLine(const MimeAuthLine& original, const char* newName = 0);

    virtual void* getObject(const String& name) const;

    virtual MimeHeaderLine* clone(const char* newName = 0) const;

    virtual void buildLine(String& line, bool header = true) const;

private:
    void operator=(const MimeAuthLine&); // no assignment
};

class YATE_API MimeBody : public GenObject
{
    YNOCOPY(MimeBody); // no automatic copies please
public:
    virtual ~MimeBody();

    virtual void* getObject(const String& name) const;

    inline const MimeHeaderLine& getType() const
        { return m_type; }

    MimeBody* getFirst(const String& type) const;

    inline const ObjList& headers() const
        { return m_headers; }

    inline void appendHdr(MimeHeaderLine* hdr)
        { if (hdr) m_headers.append(hdr); }

    inline void removeHdr(MimeHeaderLine* hdr, bool delobj = true)
        { if (hdr) m_headers.remove(hdr,delobj); }

    MimeHeaderLine* findHdr(const String& name, const MimeHeaderLine* start = 0) const;

    inline void buildHeaders(String& buf) {
            m_type.buildLine(buf);
            buf << "\r\n";
            MimeHeaderLine::buildHeaders(buf,m_headers);
        }

    bool setParam(const char* name, const char* value = 0, const char* header = 0);

    bool delParam(const char* name, const char* header = 0);

    const NamedString* getParam(const char* name, const char* header = 0) const;

    const DataBlock& getBody() const;

    inline const DataBlock& body() const
        { return m_body; }

    virtual bool isSDP() const
        { return false; }

    virtual bool isMultipart() const
        { return false; }

    virtual MimeBody* clone() const = 0;

    static MimeBody* build(const char* buf, int len, const MimeHeaderLine& type);

    static String* getUnfoldedLine(const char*& buf, int& len);

protected:
    MimeBody(const String& type);

    MimeBody(const MimeHeaderLine& type);

    virtual void buildBody() const = 0;

    mutable DataBlock m_body;

    ObjList m_headers;

private:
    MimeHeaderLine m_type;               // Content type header line
};

class YATE_API MimeMultipartBody : public MimeBody
{
public:
    explicit MimeMultipartBody(const char* subtype = "mixed", const char* boundary = 0);

    MimeMultipartBody(const String& type, const char* buf, int len);

    MimeMultipartBody(const MimeHeaderLine& type, const char* buf, int len);

    virtual ~MimeMultipartBody();

    inline const ObjList& bodies() const
        { return m_bodies; }

    inline void appendBody(MimeBody* body)
        { if (body) m_bodies.append(body); }

    inline void removeBody(MimeBody* body, bool delobj = true)
        { if (body) m_bodies.remove(body,delobj); }

    MimeBody* findBody(const String& content, MimeBody** start = 0) const;

    virtual void* getObject(const String& name) const;

    virtual bool isMultipart() const
        { return true; }

    virtual MimeBody* clone() const;

protected:
    MimeMultipartBody(const MimeMultipartBody& original);

    virtual void buildBody() const;

    void parse(const char* buf, int len);

private:
    // Parse input buffer for first body boundary or data end
    // Advance buffer pass the boundary line and decrease the buffer length
    // Set endBody to true if the last boundary was found
    // Return the length of data before the found boundary
    int findBoundary(const char*& buf, int& len,
        const char* boundary, unsigned int bLen, bool& endBody);
    // Build a boundary string to be used when parsing or building body
    // Remove quotes if present. Trim blanks
    // Insert CRLF and boundary marks ('--') before parameter
    // @param boundary Destination string
    // @return False if the parameter is missing or the boundary is empty
    bool getBoundary(String& boundary) const;


    ObjList m_bodies;                    // The list of bodies contained in this multipart
};

class YATE_API MimeSdpBody : public MimeBody
{
public:
    MimeSdpBody(bool hashing = false);

    MimeSdpBody(const String& type, const char* buf, int len);

    MimeSdpBody(const MimeHeaderLine& type, const char* buf, int len);

    virtual ~MimeSdpBody();

    virtual void* getObject(const String& name) const;

    virtual bool isSDP() const
        { return true; }

    virtual MimeBody* clone() const;

    inline const ObjList& lines() const
        { return m_lines; }

    inline unsigned int hash() const
        { return m_hash; }

    NamedString* addLine(const char* name, const char* value = 0);

    const NamedString* getLine(const char* name) const;

    const NamedString* getNextLine(const NamedString* line) const;

protected:
    MimeSdpBody(const MimeSdpBody& original);

    virtual void buildBody() const;

private:
    // Build the lines from a data buffer
    void buildLines(const char* buf, int len);

    ObjList m_lines;
    ObjList* m_lineAppend;
    unsigned int m_hash;
    bool m_hashing;
};

class YATE_API MimeBinaryBody : public MimeBody
{
public:
    MimeBinaryBody(const String& type, const char* buf, int len);

    MimeBinaryBody(const MimeHeaderLine& type, const char* buf, int len);

    virtual ~MimeBinaryBody();

    virtual void* getObject(const String& name) const;

    virtual MimeBody* clone() const;

protected:
    MimeBinaryBody(const MimeBinaryBody& original);

    virtual void buildBody() const;
};

class YATE_API MimeStringBody : public MimeBody
{
public:
    MimeStringBody(const String& type, const char* buf, int len = -1);

    MimeStringBody(const MimeHeaderLine& type, const char* buf, int len = -1);

    virtual ~MimeStringBody();

    virtual void* getObject(const String& name) const;

    virtual MimeBody* clone() const;

    inline const String& text() const
        { return m_text; }

protected:
    MimeStringBody(const MimeStringBody& original);

    virtual void buildBody() const;

private:
    String m_text;
};

class YATE_API MimeLinesBody : public MimeBody
{
public:
    MimeLinesBody(const String& type, const char* buf, int len);

    MimeLinesBody(const MimeHeaderLine& type, const char* buf, int len);

    virtual ~MimeLinesBody();

    virtual void* getObject(const String& name) const;

    virtual MimeBody* clone() const;

    inline const ObjList& lines() const
        { return m_lines; }

    inline void addLine(const char* line)
        { m_lines.append(new String(line)); }

protected:
    MimeLinesBody(const MimeLinesBody& original);

    virtual void buildBody() const;

private:
    ObjList m_lines;
};

}; // namespace TelEngine

#endif /* __YATEMIME_H */

/* vi: set ts=8 sw=4 sts=4 noet: */