drogon/drogon_ctl/create_model.h

/**
 *
 *  create_model.h
 *  An Tao
 *
 *  Copyright 2018, An Tao.  All rights reserved.
 *  https://github.com/an-tao/drogon
 *  Use of this source code is governed by a MIT license
 *  that can be found in the License file.
 *
 *  Drogon
 *
 */

#pragma once

#include <drogon/config.h>
#include <json/json.h>
#if USE_ORM
#include <drogon/orm/DbClient.h>
using namespace drogon::orm;
#endif
#include <drogon/DrObject.h>
#include "CommandHandler.h"
#include <string>

using namespace drogon;


namespace drogon_ctl
{

struct ColumnInfo
{
  std::string _colName;
  std::string _colValName;
  std::string _colTypeName;
  std::string _colType;
  std::string _colDatabaseType;
  std::string _dbType;
  ssize_t _colLength = 0;
  size_t _index = 0;
  bool _isAutoVal = false;
  bool _isPrimaryKey = false;
  bool _notNull = false;
  bool _hasDefaultVal = false;
};

class create_model : public DrObject<create_model>, public CommandHandler
{
public:
  virtual void handleCommand(std::vector<std::string> &parameters) override;
  virtual std::string script() override { return "create Model classes files"; }

protected:
  void createModel(const std::string &path);
  void createModel(const std::string &path, const Json::Value &config);
#if USE_POSTGRESQL
  void createModelClassFromPG(const std::string &path, const DbClientPtr &client, const std::string &tableName);
  void createModelFromPG(const std::string &path, const DbClientPtr &client);
#endif
#if USE_MYSQL
  void createModelClassFromMysql(const std::string &path, const DbClientPtr &client, const std::string &tableName);
  void createModelFromMysql(const std::string &path, const DbClientPtr &client);
#endif
#if USE_SQLITE3
  void createModelClassFromSqlite3(const std::string &path, const DbClientPtr &client, const std::string &tableName);
  void createModelFromSqlite3(const std::string &path, const DbClientPtr &client);
#endif
  std::string _dbname;
};
} // namespace drogon_ctl