```ruby
require 'trinity/current'
require 'trinity/policy'

class DashboardController < ApplicationController
  layout 'dashboard'
  before_action :authenticate_user!, :team_pick!

  private
  def team_pick!
    begin
      # TODO: Remove those magic string
      if request.host.start_with?('team-') && (request.host.end_with?('noty.im') || request.host.end_with?('noty.dev'))
        team = TeamService.find_team_from_host request.host
        if TeamPolicy.can_manage?(team, current.user) || TeamPolicy.can_view?(team, current.user)
          session[:team] = team.id.to_s
        else
          return head :forbidden
        end
      end

      if session[:team] && (team = Team.find(session[:team]))
        if TeamPolicy.can_manage?(team, current.user) || TeamPolicy.can_view?(team, current.user)
          current.team = team
        else
          current.team = session[:team] = nil
        end
      else
        current.team = session[:team] = current.user.teams.first.id.to_s
      end

    rescue => e
      Bugsnag.notify e
      session[:team] = nil
      # This is terrible wrong, we need to log it
      # TODO Log it
    end
  end
end
```

Store team object on current, team id on session
user
```ruby
require 'trinity/current'
require 'trinity/policy'

class DashboardController < ApplicationController
  layout 'dashboard'
  before_action :authenticate_user!, :team_pick!

  private
  def team_pick!
    begin
      # TODO: Remove those magic string
      if request.host.start_with?('team-') && (request.host.end_with?('noty.im') || request.host.end_with?('noty.dev'))
        team = TeamService.find_team_from_host request.host
        if TeamPolicy.can_manage?(team, current.user) || TeamPolicy.can_view?(team, current.user)
          session[:team] = team.id.to_s
        else
          return head :forbidden
        end
      end

      if session[:team] && (team = Team.find(session[:team]))
        if TeamPolicy.can_manage?(team, current.user) || TeamPolicy.can_view?(team, current.user)
          current.team = team
        else
          current.team = session[:team] = nil
        end
      else
        current.team = current.user.default_team
        session[:team] = current.user.default_team.id.to_s
      end

    rescue => e
      Bugsnag.notify e
      session[:team] = nil
      # This is terrible wrong, we need to log it
      # TODO Log it
    end
  end
end
```
assistant