Fix milestone extraction, Fixes stephencelis#289

Author: shubhamshuklaer

ghi

@@ -1568,9 +1568,24 @@ GitHub repository or by appending your command with the user/repo:
         end
         @issue
       end
+
       alias extract_issue     issue
-      alias milestone         issue
-      alias extract_milestone issue
+
+      def extract_milestone
+        return @milestone if defined? @milestone
+        extract_issue
+        if @issue
+          i = throb { api.get "/repos/#{repo}/issues/#{@issue}" }.body
+          if i["milestone"]
+            @milestone = i["milestone"]["number"]
+          else
+            @milestone = nil
+          end
+        end
+        @milestone
+      end
+
+      alias milestone extract_milestone
 
       def infer_issue_from_branch_prefix
         @issue = `git symbolic-ref --short HEAD 2>/dev/null`[/^\d+/];
@@ -2686,7 +2701,6 @@ EOF
       def execute
         self.action = 'index'
         require_repo
-        extract_milestone
 
         begin
           options.parse! args

lib/ghi/commands/command.rb

@@ -108,9 +108,24 @@ def issue
         end
         @issue
       end
+
       alias extract_issue     issue
-      alias milestone         issue
-      alias extract_milestone issue
+
+      def extract_milestone
+        return @milestone if defined? @milestone
+        extract_issue
+        if @issue
+          i = throb { api.get "/repos/#{repo}/issues/#{@issue}" }.body
+          if i["milestone"]
+            @milestone = i["milestone"]["number"]
+          else
+            @milestone = nil
+          end
+        end
+        @milestone
+      end
+
+      alias milestone extract_milestone
 
       def infer_issue_from_branch_prefix
         @issue = `git symbolic-ref --short HEAD 2>/dev/null`[/^\d+/];

lib/ghi/commands/milestone.rb

@@ -82,7 +82,6 @@ def options
       def execute
         self.action = 'index'
         require_repo
-        extract_milestone
 
         begin
           options.parse! args