drogon/drogon_ctl/create_controller.h

51 lines
2.0 KiB
C++
Executable File

/**
*
* @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);
};
} // namespace drogon_ctl