restore scan.features.rb

src/scan.features.rb

@@ -0,0 +1,61 @@
+#!/usr/bin/env ruby
+# encoding: utf-8
+
+require 'json'
+require 'optparse'
+
+OptionParser.new do |opts|
+    opts.banner = "Usage: scan.features.rb [options]"
+
+    opts.on("-s", "--src-dir PATH", "Specify gitlab source dir (required)") do |v|
+        GITLAB_SRC_DIR = File.expand_path(v)
+    end
+
+    opts.on("-o", "--output PATH", "Output to json file (required)") do |v|
+        EXPORT_JSON_FILE = File.expand_path(v)
+    end
+
+    opts.on("-h", "--help", "Prints this help") do
+        puts opts
+        exit
+    end
+end
+.parse!
+
+if GITLAB_SRC_DIR.nil? || EXPORT_JSON_FILE.nil?
+    puts "[!] missing required options"
+    puts "[!] use -h for help"
+    exit 1
+end
+
+def ignore_exception
+    begin
+      yield
+    rescue Exception
+    end
+end
+
+puts "[*] loading features.rb..."
+ignore_exception do
+    require_relative "#{GITLAB_SRC_DIR}/ee/app/models/gitlab_subscriptions/features.rb"
+end
+
+ALL_FEATURES = []
+GitlabSubscriptions::Features.constants.each do |const_name|
+    puts "[*] gathering features from #{const_name}"
+    if const_name.to_s.include? 'FEATURE'
+        ALL_FEATURES.concat(GitlabSubscriptions::Features.const_get(const_name))
+    else
+        puts "[?] unrecognized constant #{const_name}"
+    end
+end
+
+ALL_FEATURES.uniq!
+ALL_FEATURES.sort_by! { |feature| feature }
+
+puts "[*] total features: #{ALL_FEATURES.size}"
+
+puts "[*] writing to #{EXPORT_JSON_FILE}"
+File.write(EXPORT_JSON_FILE, JSON.pretty_generate(ALL_FEATURES))
+
+puts "[*] done"