Fix login paths

app/controllers/application_controller.rb

@@ -17,8 +17,14 @@ class ApplicationController < ActionController::Base
 
   def after_sign_in_path_for(_resource)
     rt = session[:user_return_to].to_s.presence
-    return session[:user_return_to].to_s if rt && rt != admin_login_path
+    login_paths = [admin_login_path, registrar_login_path]
-    admin_dashboard_path
+    return rt if rt && !login_paths.include?(rt)
+
+    if request.path.match('registrar')
+      registrar_root_path
+    elsif request.path.match('admin')
+      admin_root_path
+    end
   end
 
   def after_sign_out_path_for(_resource)

app/views/layouts/login.haml

@@ -21,7 +21,7 @@
         %h2.form-signin-heading.text-center Eesti Interneti SA
         %hr
         / TODO: Refactor this when ID card login is done
-        = button_to 'ID card (user1)', 'sessions',
+        = button_to 'ID card (user1)', admin_sessions_path,
           class: 'btn btn-lg btn-primary btn-block', name: 'user1'
-        = button_to 'ID card (user2)', 'sessions',
+        = button_to 'ID card (user2)', admin_sessions_path,
           class: 'btn btn-lg btn-primary btn-block', name: 'user2'

app/views/layouts/registrar.haml

@@ -20,7 +20,7 @@
             %span.icon-bar
             %span.icon-bar
             %span.icon-bar
-          = link_to registrar_invoices_path, class: 'navbar-brand' do
+          = link_to registrar_root_path, class: 'navbar-brand' do
             EIS Registrar
             - if unstable_env.present?
               .text-center

config/routes.rb

@@ -75,7 +75,7 @@ Rails.application.routes.draw do
       get 'logout' => '/devise/sessions#destroy'
     end
 
-    root 'domains#index'
+    root 'dashboards#show'
   end
 
   namespace(:registrar) do
@@ -96,16 +96,16 @@ Rails.application.routes.draw do
     #   root to: 'domains#index', as: :authenticated_root
     # end
 
-    root 'domains#index'
+    root 'invoices#index'
   end
 
   devise_for :users
 
-  # authenticated :user do
+  devise_scope :user do
-  #   root to: 'admin/domains#index', as: :authenticated_root
+    get 'login' => 'admin/sessions#login'
-  # end
+  end
 
-  root to: redirect('admin/login')
+  root to: redirect('login')
 
   # The priority is based upon order of creation: first created -> highest priority.
   # See how all your routes lay out with "rake routes".