add `brew analytics` command (#173)

4 files changed, 54 insertions, 9 deletions

diff --git a/Library/Homebrew/cmd/analytics.rb b/Library/Homebrew/cmd/analytics.rb
new file mode 100644
index 000000000..649c53710
--- /dev/null
+++ b/Library/Homebrew/cmd/analytics.rb
@@ -0,0 +1,46 @@
+#:  * `analytics` [`state`]:
+#:    Display anonymous user behaviour analytics state.
+#:    Read more at <https://git.io/brew-analytics>.
+#:
+#:  * `analytics` (`on`|`off`):
+#:    Turn on/off Homebrew's analytics.
+#:
+#:  * `analytics` `regenerate-uuid`:
+#:    Regenerate UUID used in Homebrew's analytics.
+
+module Homebrew
+  def analytics
+    config_file = HOMEBREW_REPOSITORY/".git/config"
+
+    raise UsageError if ARGV.named.size > 1
+    case ARGV.named.first
+    when nil, "state"
+      analyticsdisabled = \
+        Utils.popen_read("git config --file=#{config_file} --get homebrew.analyticsdisabled").chuzzle
+      uuid = \
+        Utils.popen_read("git config --file=#{config_file} --get homebrew.analyticsuuid").chuzzle
+      if ENV["HOMEBREW_NO_ANALYTICS"]
+        puts "Analytics is disabled (by HOMEBREW_NO_ANALYTICS)."
+      elsif analyticsdisabled == "true"
+        puts "Analytics is disabled."
+      else
+        puts "Analytics is enabled."
+        puts "UUID: #{uuid}" if uuid
+      end
+    when "on"
+      safe_system "git", "config", "--file=#{config_file}",
+                                   "--replace-all", "homebrew.analyticsdisabled", "false"
+      safe_system "git", "config", "--file=#{config_file}",
+                                   "--replace-all", "homebrew.analyticsmessage", "true"
+    when "off"
+      safe_system "git", "config", "--file=#{config_file}",
+                                   "--replace-all", "homebrew.analyticsdisabled", "true"
+      system "git", "config", "--file=#{config_file}", "--unset-all", "homebrew.analyticsuuid"
+    when "regenerate-uuid"
+      # it will be regenerated in next run.
+      system "git", "config", "--file=#{config_file}", "--unset-all", "homebrew.analyticsuuid"
+    else
+      raise UsageError
+    end
+  end
+end
diff --git a/Library/Homebrew/cmd/update-report.rb b/Library/Homebrew/cmd/update-report.rb
index 53b9d639a..1d9822ae5 100644
--- a/Library/Homebrew/cmd/update-report.rb
+++ b/Library/Homebrew/cmd/update-report.rb
@@ -17,7 +17,7 @@ module Homebrew
         Utils.popen_read("git", "config", "--local", "--get", "homebrew.analyticsmessage").chuzzle
       analytics_disabled = \
         Utils.popen_read("git", "config", "--local", "--get", "homebrew.analyticsdisabled").chuzzle
-      if analytics_message_displayed != "true" && analytics_disabled != "true"
+      if analytics_message_displayed != "true" && analytics_disabled != "true" && !ENV["HOMEBREW_NO_ANALYTICS"]
         ENV["HOMEBREW_NO_ANALYTICS_THIS_RUN"] = "1"
         ohai "Homebrew has enabled anonymous aggregate user behaviour analytics"
         puts "Read the analytics documentation (and how to opt-out) here:"
diff --git a/Library/Homebrew/utils/analytics.sh b/Library/Homebrew/utils/analytics.sh
index 47716e5e5..2ca1c72e7 100644
--- a/Library/Homebrew/utils/analytics.sh
+++ b/Library/Homebrew/utils/analytics.sh
@@ -18,11 +18,9 @@ setup-analytics() {
 
   migrate-legacy-uuid-file
 
-  # Make disabling anlytics sticky
   if [[ -n "$HOMEBREW_NO_ANALYTICS" ]]
   then
-    git config --file="$git_config_file" --replace-all homebrew.analyticsdisabled true
-    git config --file="$git_config_file" --unset-all homebrew.analyticsuuid
+    return
   fi
 
   local message_seen="$(git config --file="$git_config_file" --get homebrew.analyticsmessage)"
@@ -72,10 +70,11 @@ report-analytics-screenview-command() {
   # Don't report commands used mostly by our scripts and not users.
   # TODO: list more e.g. shell completion things here perhaps using a single
   # script as a shell-completion entry point.
-  if [[ "$HOMEBREW_COMMAND" = "commands" ]]
-  then
-    return
-  fi
+  case "$HOMEBREW_COMMAND" in
+    --prefix|analytics|command|commands)
+      return
+      ;;
+  esac
 
   local args=(
     --max-time 3 \
diff --git a/Library/brew.sh b/Library/brew.sh
index 54f713dd9..33f446503 100644
--- a/Library/brew.sh
+++ b/Library/brew.sh
@@ -190,7 +190,7 @@ fi
 if [[ "$(id -u)" = "0" && "$(/usr/bin/stat -f%u "$HOMEBREW_BREW_FILE")" != "0" ]]
 then
   case "$HOMEBREW_COMMAND" in
-    install|reinstall|postinstall|link|pin|update|upgrade|create|migrate|tap|tap-pin|switch)
+    analytics|install|reinstall|postinstall|link|pin|update|upgrade|create|migrate|tap|tap-pin|switch)
       odie <<EOS
 Cowardly refusing to 'sudo brew $HOMEBREW_COMMAND'
 You can use brew with sudo, but only if the brew executable is owned by root.