railties/lib/rails/command/helpers/pretty_credentials.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

# frozen_string_literal: true

require "fileutils"

module Rails
  module Command
    module Helpers
      module PrettyCredentials
        Error = Class.new(StandardError)

        def opt_in_pretty_credentials
          unless already_answered? || already_opted_in?
            answer = yes?("Would you like to make the credentials diff from git more readable in the future? [Y/n]")
          end

          opt_in! if answer
          FileUtils.touch(tracker) unless answer.nil?
        rescue Error
          say("Couldn't setup git to prettify the credentials diff")
        end

        private
          def already_answered?
            tracker.exist?
          end

          def already_opted_in?
            system_call("git config --get 'diff.rails_credentials.textconv'", accepted_codes: [0, 1])
          end

          def opt_in!
            system_call("git config diff.rails_credentials.textconv 'bin/rails credentials:show'", accepted_codes: [0])

            git_attributes = Rails.root.join(".gitattributes")
            File.open(git_attributes, "a+") do |file|
              file.write(<<~EOM)
                config/credentials/*.yml.enc diff=rails_credentials
                config/credentials.yml.enc diff=rails_credentials
              EOM
            end
          end

          def tracker
            Rails.root.join("tmp", "rails_pretty_credentials")
          end

          def system_call(command_line, accepted_codes:)
            result = system(command_line)
            raise(Error) if accepted_codes.exclude?($?.exitstatus)
            result
          end
      end
    end
  end
end