/**
 *
 *  @file
 *  @author An Tao
 *  @section LICENSE
 *
 *  Copyright 2018, An Tao.  All rights reserved.
 *  Use of this source code is governed by a MIT license
 *  that can be found in the License file.
 *
 *  @section DESCRIPTION
 *
 */

#pragma once

#include <drogon/DrObject.h>
#include "CommandHandler.h"
using namespace drogon;
namespace drogon_ctl
{
    class create_controller:public DrObject<create_controller>,public CommandHandler
    {
    public:
        virtual void handleCommand(std::vector<std::string> &parameters) override;
        virtual std::string script() override {return "create controller files";}
    protected:
        enum ControllerType{
            Simple=0,
            API,
            WebSocket
        };
        void createSimpleController(std::vector<std::string> &ctlNames,const std::string &namespaceName="");
        void createSimpleController(const std::string &ctlName,const std::string &namespaceName="");
        void createWebsockController(std::vector<std::string> &ctlNames,const std::string &namespaceName="");
        void createWebsockController(const std::string &ctlName,const std::string &namespaceName="");

        void createApiController(std::vector<std::string> &apiClasses);
        void createApiController(const std::string &className);
        void newSimpleControllerHeaderFile(std::ofstream &file,const std::string &ctlName,const std::string &namespaceName="");
        void newSimpleControllerSourceFile(std::ofstream &file,const std::string &ctlName,const std::string &namespaceName="");
        void newWebsockControllerHeaderFile(std::ofstream &file,const std::string &ctlName,const std::string &namespaceName="");
        void newWebsockControllerSourceFile(std::ofstream &file,const std::string &ctlName,const std::string &namespaceName="");

        void newApiControllerHeaderFile(std::ofstream &file,const std::string &className);
        void newApiControllerSourceFile(std::ofstream &file,const std::string &className,const std::string &filename);

    };
}