package org.apache.maven.plugin.github;
   
   /*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance
   * with the License.  You may obtain a copy of the License at
   *
   *   http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing,
   * software distributed under the License is distributed on an
   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   * KIND, either express or implied.  See the License for the
   * specific language governing permissions and limitations
   * under the License.
   */
  
  import org.apache.maven.plugin.issues.Issue;
  import org.apache.maven.project.MavenProject;
  import org.eclipse.egit.github.core.Label;
  import org.eclipse.egit.github.core.client.GitHubClient;
  import org.eclipse.egit.github.core.service.IssueService;
  
  import java.io.IOException;
  import java.net.MalformedURLException;
  import java.net.URL;
  import java.util.ArrayList;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  /**
   * @since 2.8
   */
  public class GitHubDownloader
  {
  
      /**
       * The github client.
       */
      private GitHubClient client;
  
      /**
       * A boolean to indicate if we should include open issues as well
       */
      private boolean includeOpenIssues;
  
      /**
       * A boolean to indicate if we should only include issues with milestones
       */
      private boolean onlyMilestoneIssues;
  
      /**
       * The owner/organization of the github repo.
       */
      private String githubOwner;
  
      /**
       * The name of the github repo.
       */
      private String githubRepo;
  
      /**
       * The url to the github repo's issue management
       */
      private String githubIssueURL;
  
      public GitHubDownloader( MavenProject project, String githubScheme, int githubPort, boolean includeOpenIssues,
                               boolean onlyMilestoneIssues )
          throws MalformedURLException
      {
          this.includeOpenIssues = includeOpenIssues;
          this.onlyMilestoneIssues = onlyMilestoneIssues;
  
          URL githubURL = new URL( project.getIssueManagement().getUrl() );
  
          // The githubclient prefers to connect to 'github.com' using the api domain, unlike github enterprise
          // which can connect fine using its domain, so for github.com use empty constructor
          if ( githubURL.getHost().equalsIgnoreCase( "github.com" ) )
          {
              this.client = new GitHubClient();
          }
          else
          {
              this.client = new GitHubClient( githubURL.getHost(), githubPort, githubScheme );
          }
  
          this.githubIssueURL = project.getIssueManagement().getUrl();
          if ( !this.githubIssueURL.endsWith( "/" ) )
          {
              this.githubIssueURL = this.githubIssueURL + "/";
          }
  
          String urlPath = githubURL.getPath();
          if ( urlPath.startsWith( "/" ) )
         {
             urlPath = urlPath.substring( 1 );
         }
 
         if ( urlPath.endsWith( "/" ) )
         {
             urlPath = urlPath.substring( 0, urlPath.length() - 2 );
         }
 
         String[] urlPathParts = urlPath.split( "/" );
 
         if ( urlPathParts.length != 3 )
         {
             throw new MalformedURLException(
                 "GitHub issue management URL must look like, [GITHUB_DOMAIN]/[OWNER]/[REPO]/issues" );
         }
 
         this.githubOwner = urlPathParts[0];
         this.githubRepo = urlPathParts[1];
     }
 
     private Issue createIssue( org.eclipse.egit.github.core.Issue githubIssue )
     {
         Issue issue = new Issue();
 
         issue.setId( String.valueOf( githubIssue.getNumber() ) );
 
         issue.setLink( this.githubIssueURL + githubIssue.getNumber() );
 
         issue.setCreated( githubIssue.getCreatedAt() );
 
         issue.setUpdated( githubIssue.getUpdatedAt() );
 
         if ( githubIssue.getAssignee() != null )
         {
             if ( githubIssue.getAssignee().getName() != null )
             {
                 issue.setAssignee( githubIssue.getAssignee().getName() );
             }
             else
             {
                 issue.setAssignee( githubIssue.getAssignee().getLogin() );
             }
         }
 
         issue.setTitle( githubIssue.getTitle() );
 
         issue.setSummary( githubIssue.getTitle() );
 
         if ( githubIssue.getMilestone() != null )
         {
             issue.addFixVersion( githubIssue.getMilestone().getTitle() );
         }
 
         issue.setReporter( githubIssue.getUser().getLogin() );
 
         if ( githubIssue.getClosedAt() != null )
         {
             issue.setStatus( "closed" );
         }
         else
         {
             issue.setStatus( "open" );
         }
 
         List<Label> labels = githubIssue.getLabels();
         if ( labels != null && !labels.isEmpty() )
         {
             issue.setType( labels.get( 0 ).getName() );
         }
 
         return issue;
     }
 
     public List<Issue> getIssueList()
         throws IOException
     {
         List<Issue> issueList = new ArrayList<Issue>();
 
         IssueService service = new IssueService( client );
         Map<String, String> issueFilter = new HashMap<String, String>();
 
         if ( includeOpenIssues )
         {
             // Adding open issues
 
             for ( org.eclipse.egit.github.core.Issue issue : service.getIssues( githubOwner, githubRepo, issueFilter ) )
             {
                 if ( !onlyMilestoneIssues || issue.getMilestone() != null )
                 {
                     issueList.add( createIssue( issue ) );
                 }
             }
         }
 
         // Adding closed issues
 
         issueFilter.put( "state", "closed" );
 
         for ( org.eclipse.egit.github.core.Issue issue : service.getIssues( githubOwner, githubRepo, issueFilter ) )
         {
             if ( !onlyMilestoneIssues || issue.getMilestone() != null )
             {
                 issueList.add( createIssue( issue ) );
             }
         }
 
         return issueList;
     }
 
 }