| Akron | 80027d1 | 2016-01-20 17:36:36 +0100 | [diff] [blame] | 1 | package Kalamar::Plugin::KalamarUser; |
| 2 | use Mojo::Base 'Mojolicious::Plugin'; |
| Akron | 864c293 | 2018-11-16 17:18:55 +0100 | [diff] [blame] | 3 | use Mojo::Util qw/deprecated/; |
| Akron | 3239663 | 2018-10-11 17:08:37 +0200 | [diff] [blame] | 4 | use Mojo::Promise; |
| Akron | 80027d1 | 2016-01-20 17:36:36 +0100 | [diff] [blame] | 5 | use Mojo::ByteStream 'b'; |
| 6 | |
| 7 | has 'api'; |
| Akron | dadacf1 | 2016-01-22 19:24:59 +0100 | [diff] [blame] | 8 | has 'ua'; |
| Akron | 80027d1 | 2016-01-20 17:36:36 +0100 | [diff] [blame] | 9 | |
| 10 | sub register { |
| 11 | my ($plugin, $mojo, $param) = @_; |
| 12 | |
| 13 | # Load parameter from config file |
| 14 | if (my $config_param = $mojo->config('Kalamar')) { |
| 15 | $param = { %$param, %$config_param }; |
| 16 | }; |
| 17 | |
| Akron | 2cb9a3d | 2016-02-09 23:59:46 +0100 | [diff] [blame] | 18 | # Load 'notifications' plugin |
| 19 | unless (exists $mojo->renderer->helpers->{notify}) { |
| 20 | $mojo->plugin(Notifications => { |
| 21 | HTML => 1 |
| 22 | }); |
| 23 | }; |
| 24 | |
| Akron | 80027d1 | 2016-01-20 17:36:36 +0100 | [diff] [blame] | 25 | # Set API! |
| 26 | $plugin->api($param->{api}) or return; |
| Akron | c95c9e7 | 2017-03-30 21:53:51 +0200 | [diff] [blame] | 27 | $plugin->ua(Mojo::UserAgent->new( |
| Akron | a3c353c | 2019-02-14 23:50:00 +0100 | [diff] [blame] | 28 | connect_timeout => 90, |
| Akron | acef8dc | 2018-08-29 18:16:08 +0200 | [diff] [blame] | 29 | inactivity_timeout => 120, |
| 30 | max_redirects => 3 |
| Akron | c95c9e7 | 2017-03-30 21:53:51 +0200 | [diff] [blame] | 31 | )); |
| Akron | 80027d1 | 2016-01-20 17:36:36 +0100 | [diff] [blame] | 32 | |
| Akron | be9d5b3 | 2017-04-05 20:48:24 +0200 | [diff] [blame] | 33 | # Set app to server |
| 34 | $plugin->ua->server->app($mojo); |
| 35 | |
| Akron | 8bbbecf | 2019-07-01 18:57:30 +0200 | [diff] [blame] | 36 | # Get a user agent object for Kalamar |
| 37 | $mojo->helper( |
| 38 | 'kalamar_ua' => sub { |
| 39 | return $plugin->ua; |
| 40 | } |
| 41 | ); |
| 42 | |
| Akron | 864c293 | 2018-11-16 17:18:55 +0100 | [diff] [blame] | 43 | # Get user handle |
| 44 | $mojo->helper( |
| 45 | 'user_handle' => sub { |
| 46 | my $c = shift; |
| 47 | |
| 48 | # Get from stash |
| 49 | my $user = $c->stash('user'); |
| 50 | return $user if $user; |
| 51 | |
| 52 | # Get from session |
| 53 | $user = $c->session('user'); |
| 54 | |
| 55 | # Set in stash |
| 56 | if ($user) { |
| 57 | $c->stash(user => $user); |
| 58 | return $user; |
| 59 | }; |
| 60 | |
| 61 | return 'not_logged_in'; |
| 62 | } |
| 63 | ); |
| 64 | |
| 65 | # This is a new general korap_request helper, |
| 66 | # that can trigger some hooks for, e.g., authentication |
| 67 | # or analysis. It returns a promise. |
| 68 | $mojo->helper( |
| 69 | 'korap_request' => sub { |
| 70 | my $c = shift; |
| 71 | my $method = shift; |
| 72 | my $path = shift; |
| 73 | |
| 74 | # Get plugin user agent |
| 75 | my $ua = $plugin->ua; |
| 76 | |
| 77 | my $url = Mojo::URL->new($path); |
| 78 | my $tx = $ua->build_tx(uc($method), $url, @_); |
| 79 | |
| 80 | # Set X-Forwarded for |
| 81 | $tx->req->headers->header( |
| 82 | 'X-Forwarded-For' => $c->client_ip |
| 83 | ); |
| 84 | |
| Akron | 864c293 | 2018-11-16 17:18:55 +0100 | [diff] [blame] | 85 | # Emit Hook to alter request |
| 86 | $c->app->plugins->emit_hook( |
| 87 | before_korap_request => ($c, $tx) |
| 88 | ); |
| 89 | |
| 90 | return $ua->start_p($tx); |
| 91 | } |
| 92 | ); |
| Akron | 80027d1 | 2016-01-20 17:36:36 +0100 | [diff] [blame] | 93 | }; |
| 94 | |
| Akron | 80027d1 | 2016-01-20 17:36:36 +0100 | [diff] [blame] | 95 | 1; |
| 96 | |
| 97 | |
| 98 | __END__ |
| Akron | dadacf1 | 2016-01-22 19:24:59 +0100 | [diff] [blame] | 99 | |