From bc552d07307d74a60b2e91ba474ad46192b5aee2 Mon Sep 17 00:00:00 2001
From: Vitalii Elenhaupt <velenhaupt@gmail.com>
Date: Thu, 23 Nov 2017 23:59:59 +0200
Subject: [PATCH] Generate config

closes #17
---
 src/ameba/cli/cmd.cr | 21 +++++++++++++++++++--
 src/ameba/config.cr  | 19 +++++++++++++++----
 src/ameba/runner.cr  |  2 +-
 3 files changed, 35 insertions(+), 7 deletions(-)

diff --git a/src/ameba/cli/cmd.cr b/src/ameba/cli/cmd.cr
index 25306829..cec3393f 100644
--- a/src/ameba/cli/cmd.cr
+++ b/src/ameba/cli/cmd.cr
@@ -6,8 +6,9 @@ module Ameba::Cli
   extend self
 
   private class Opts
-    property config : String?
+    property config = Ameba::Config::PATH
     property silent : Bool = false
+    property generate : Bool = false
     property files : Array(String)?
     property only : Array(String)?
     property except : Array(String)?
@@ -33,6 +34,10 @@ module Ameba::Cli
       parser.on("--except RULE1,RULE2,...", "Disable the given rules") do |rules|
         opts.except = rules.split ","
       end
+
+      parser.on("--gen-config", "Generate a configuration file acting as a TODO list") do
+        opts.generate = true
+      end
     end
 
     run_ameba opts
@@ -45,7 +50,9 @@ module Ameba::Cli
 
     configure_rules(config, opts)
 
-    exit 1 unless Ameba.run(config).success?
+    runner = Ameba.run(config)
+    generate_config_file(config, runner, opts) if opts.generate
+    exit 1 unless runner.success?
   rescue e
     puts "Error: #{e.message}"
     exit 255
@@ -64,6 +71,16 @@ module Ameba::Cli
     end
   end
 
+  private def generate_config_file(config, runner, opts)
+    failed_rules =
+      runner.sources
+            .map { |s| s.errors.map &.rule.name }
+            .flatten
+            .uniq!
+    failed_rules.each { |rule| config.update_rule rule, enabled: false }
+    File.write(opts.config, config.to_yaml)
+  end
+
   private def print_version
     puts Ameba::VERSION
     exit 0
diff --git a/src/ameba/config.cr b/src/ameba/config.cr
index dd5a6a49..1167a13c 100644
--- a/src/ameba/config.cr
+++ b/src/ameba/config.cr
@@ -12,6 +12,7 @@ require "yaml"
 # By default config loads `.ameba.yml` file in a current directory.
 #
 class Ameba::Config
+  PATH = ".ameba.yml"
   setter formatter : Formatter::BaseFormatter?
   setter files : Array(String)?
   getter rules : Array(Rule::Base)
@@ -19,7 +20,7 @@ class Ameba::Config
   # Creates a new instance of `Ameba::Config` based on YAML parameters.
   #
   # `Config.load` uses this constructor to instantiate new config by YAML file.
-  protected def initialize(config : YAML::Any)
+  protected def initialize(@config : Hash(YAML::Type, YAML::Type))
     @rules = Rule.rules.map &.new(config)
   end
 
@@ -29,10 +30,11 @@ class Ameba::Config
   # config = Ameba::Config.load
   # ```
   #
-  def self.load(path = nil)
-    path ||= ".ameba.yml"
+  def self.load(path = PATH)
     content = File.exists?(path) ? File.read path : "{}"
-    Config.new YAML.parse(content)
+    Config.new YAML.parse(content).as_h
+  rescue e
+    raise "Config file is invalid"
   end
 
   # Returns a list of paths (with wildcards) to files.
@@ -78,6 +80,15 @@ class Ameba::Config
     @rules[index] = rule
   end
 
+  def to_yaml(yaml : YAML::Builder)
+    yaml.mapping do
+      rules.each do |rule|
+        rule.name.to_yaml(yaml)
+        rule.to_yaml(yaml)
+      end
+    end
+  end
+
   private def default_files
     Dir["**/*.cr"].reject(&.starts_with? "lib/")
   end
diff --git a/src/ameba/runner.cr b/src/ameba/runner.cr
index 1293c2cf..5ae2d9f5 100644
--- a/src/ameba/runner.cr
+++ b/src/ameba/runner.cr
@@ -15,7 +15,7 @@ module Ameba
     @rules : Array(Rule::Base)
 
     # A list of sources to run inspection on.
-    @sources : Array(Source)
+    getter sources : Array(Source)
 
     # A formatter to prepare report.
     @formatter : Formatter::BaseFormatter