/*
 * Copyright (c) 2011-2016 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#pragma once

#include <string>
#include <vector>

#include <boost/optional.hpp>

#include <Swiften/Base/API.h>
#include <Swiften/Elements/JingleDescription.h>
#include <Swiften/Elements/JingleTransportPayload.h>
#include <Swiften/Elements/Payload.h>
#include <Swiften/JID/JID.h>

namespace Swift {
    class SWIFTEN_API JingleContentPayload : public Payload {
        public:
            typedef std::shared_ptr<JingleContentPayload> ref;

            enum Creator {
                UnknownCreator,
                InitiatorCreator,
                ResponderCreator
            };

            JingleContentPayload() : creator(UnknownCreator) {
            }

            /*enum Senders {
                NoSenders,
                InitiatorSender,
                ResponderSender,
                BothSenders,
            };*/

            Creator getCreator() const {
                return creator;
            }

            void setCreator(Creator creator) {
                this->creator = creator;
            }

            const std::string& getName() const {
                return name;
            }

            void setName(const std::string& name) {
                this->name = name;
            }

            const std::vector<JingleDescription::ref>& getDescriptions() const {
                return descriptions;
            }

            void addDescription(JingleDescription::ref description) {
                descriptions.push_back(description);
            }

            const std::vector<std::shared_ptr<JingleTransportPayload> >& getTransports() const {
                return transports;
            }

            void addTransport(std::shared_ptr<JingleTransportPayload>  transport) {
                transports.push_back(transport);
            }

            template<typename T>
            std::shared_ptr<T> getDescription() const {
                for (size_t i = 0; i < descriptions.size(); ++i) {
                    std::shared_ptr<T> result(std::dynamic_pointer_cast<T>(descriptions[i]));
                    if (result) {
                        return result;
                    }
                }
                return std::shared_ptr<T>();
            }

            template<typename T>
            std::shared_ptr<T> getTransport() const {
                for (size_t i = 0; i < transports.size(); ++i) {
                    std::shared_ptr<T> result(std::dynamic_pointer_cast<T>(transports[i]));
                    if (result) {
                        return result;
                    }
                }
                return std::shared_ptr<T>();
            }

        private:
            Creator creator;
            std::string name;
            //Senders senders;
            std::vector<JingleDescription::ref> descriptions;
            std::vector<std::shared_ptr<JingleTransportPayload> > transports;
    };
}