kong--kong/kong/cmd/config.lua

local DB = require "kong.db"
local log = require "kong.cmd.utils.log"
local pl_path = require "pl.path"
local pl_file = require "pl.file"
local kong_global = require "kong.global"
local declarative = require "kong.db.declarative"
local conf_loader = require "kong.conf_loader"
local kong_yml = require "kong.templates.kong_yml"


local DEFAULT_FILE = "./kong.yml"


local function db_export(filename, conf)
  if pl_file.access_time(filename) then
    error(filename .. " already exists. Will not overwrite it.")
  end

  local fd, err = io.open(filename, "w")
  if not fd then
    return nil, err
  end

  local ok, err = declarative.export_from_db(fd)
  if not ok then
    error(err)
  end

  fd:close()

  os.exit(0)
end


local function generate_init(filename)
  if pl_file.access_time(filename) then
    error(filename .. " already exists.\nWill not overwrite it.")
  end
  pl_file.write(filename, kong_yml)
end


local function execute(args)
  if args.command == "init" then
    generate_init(pl_path.abspath(args[1] or DEFAULT_FILE))
    os.exit(0)
  end

  log.disable()
  -- retrieve default prefix or use given one
  local conf = assert(conf_loader(args.conf, {
    prefix = args.prefix
  }))
  log.enable()

  if pl_path.exists(conf.kong_env) then
    -- load <PREFIX>/kong.conf containing running node's config
    conf = assert(conf_loader(conf.kong_env))
  end

  args.command = args.command:gsub("%-", "_")

  if args.command == "db_import" and conf.database == "off" then
    error("'kong config db_import' only works with a database.\n" ..
          "When using database=off, reload your declarative configuration\n" ..
          "using the /config endpoint.")
  end

  if args.command == "db_export" and conf.database == "off" then
    error("'kong config db_export' only works with a database.")
  end

  package.path = conf.lua_package_path .. ";" .. package.path

  _G.kong = kong_global.new()
  kong_global.init_pdk(_G.kong, conf)

  local dc, err = declarative.new_config(conf, true)
  if not dc then
    error(err)
  end

  local db = assert(DB.new(conf))
  assert(db:init_connector())
  assert(db:connect())
  assert(db.vaults:load_vault_schemas(conf.loaded_vaults))
  assert(db.plugins:load_plugin_schemas(conf.loaded_plugins))

  _G.kong.db = db

  if args.command == "db_export" then
    return db_export(pl_path.abspath(args[1] or DEFAULT_FILE), conf)
  end

  if args.command == "db_import" or args.command == "parse" then
    local filename = args[1]
    if not filename then
      error("expected a declarative configuration file; see `kong config --help`")
    end
    filename = pl_path.abspath(filename)

    if pl_path.extension(filename) == ".lua" then
      log.deprecation("db_import of .lua files is deprecated; please convert your file into .yaml or .json")
    end

    local entities, err, _, meta = dc:parse_file(filename)
    if not entities then
      error("Failed parsing:\n" .. err)
    end

    if args.command == "db_import" then
      log("parse successful, beginning import")

      local ok, err = declarative.load_into_db(entities, meta)
      if not ok then
        error("Failed importing:\n" .. err)
      end

      log("import successful")

      -- send anonymous report if reporting is not disabled
      if conf.anonymous_reports then
        local kong_reports = require "kong.reports"
        kong_reports.init(conf)

        local report = {
          decl_fmt_version = meta._format_version,
          file_ext = pl_path.extension(filename),
        }
        kong_reports.send("config-db-import", report)
      end

    else -- parse
      log("parse successful")
    end

    os.exit(0)
  end

  error("unknown command '" .. args.command .. "'")
end

local lapp = [[
Usage: kong config COMMAND [OPTIONS]

Use declarative configuration files with Kong.

The available commands are:
  init [<file>]                       Generate an example config file to
                                      get you started. If a filename
                                      is not given, ]] .. DEFAULT_FILE .. [[ is used
                                      by default.

  db_import <file>                    Import a declarative config file into
                                      the Kong database.

  db_export [<file>]                  Export the Kong database into a
                                      declarative config file. If a filename
                                      is not given, ]] .. DEFAULT_FILE .. [[ is used
                                      by default.

  parse <file>                        Parse a declarative config file (check
                                      its syntax) but do not load it into Kong.

Options:
 -c,--conf        (optional string)   Configuration file.
 -p,--prefix      (optional string)   Override prefix directory.
]]

return {
  lapp = lapp,
  execute = execute,
  sub_commands = {
    init = true,
    db_import = true,
    db_export = true,
    parse = true,
  },
}