module Bitbucket
  module Representation
    class PullRequest < Representation::Base
      def author
        raw.fetch('author', {}).fetch('username', nil)
      end

      def description
        raw['description']
      end

      def iid
        raw['id']
      end

      def state
        if raw['state'] == 'MERGED'
          'merged'
        elsif raw['state'] == 'DECLINED'
          'closed'
        else
          'opened'
        end
      end

      def created_at
        raw['created_on']
      end

      def updated_at
        raw['updated_on']
      end

      def title
        raw['title']
      end

      def source_branch_name
        source_branch.fetch('branch', {}).fetch('name', nil)
      end

      def source_branch_sha
        source_branch.fetch('commit', {}).fetch('hash', nil)
      end

      def target_branch_name
        target_branch.fetch('branch', {}).fetch('name', nil)
      end

      def target_branch_sha
        target_branch.fetch('commit', {}).fetch('hash', nil)
      end

      private

      def source_branch
        raw['source']
      end

      def target_branch
        raw['destination']
      end
    end
  end
end