module Badges
  class BuildService < Badges::BaseService
    # returns the created badge
    def execute(source)
      if source.is_a?(Group)
        GroupBadge.new(params.merge(group: source))
      else
        ProjectBadge.new(params.merge(project: source))
      end
    end
  end
end