|
| 1 | +# Copyright 2024 Akretion (http://www.akretion.com). |
| 2 | +# @author Florian Mounier <florian.mounier@akretion.com> |
| 3 | +# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl). |
| 4 | +from urllib.parse import urlparse |
| 5 | + |
| 6 | +import requests |
| 7 | + |
| 8 | +from odoo import _, api, fields, models |
| 9 | +from odoo.exceptions import UserError |
| 10 | + |
| 11 | + |
| 12 | +class CrossConnectServer(models.Model): |
| 13 | + _name = "cross.connect.server" |
| 14 | + _description = "Cross Connect Server" |
| 15 | + _inherit = "server.env.mixin" |
| 16 | + |
| 17 | + name = fields.Char( |
| 18 | + required=True, compute="_compute_name", readonly=False, store=True |
| 19 | + ) |
| 20 | + server_url = fields.Char(required=True) |
| 21 | + api_key = fields.Char( |
| 22 | + required=True, |
| 23 | + ) |
| 24 | + group_ids = fields.One2many( |
| 25 | + "res.groups", |
| 26 | + inverse_name="cross_connect_server_id", |
| 27 | + string="Cross Connect Server Groups", |
| 28 | + readonly=True, |
| 29 | + ) |
| 30 | + menu_id = fields.Many2one( |
| 31 | + "ir.ui.menu", |
| 32 | + string="Menu", |
| 33 | + help="Menu to display the Cross Connect Server in the menu", |
| 34 | + compute="_compute_menu_id", |
| 35 | + store=True, |
| 36 | + ) |
| 37 | + web_icon_data = fields.Binary( |
| 38 | + compute="_compute_web_icon_data", inverse="_inverse_web_icon_data" |
| 39 | + ) |
| 40 | + |
| 41 | + @api.depends("server_url") |
| 42 | + def _compute_name(self): |
| 43 | + for record in self: |
| 44 | + if not record.name or record.name == "/": |
| 45 | + try: |
| 46 | + record.name = urlparse(record.server_url).hostname |
| 47 | + except Exception: |
| 48 | + record.name = "/" |
| 49 | + |
| 50 | + @api.depends("name", "group_ids") |
| 51 | + def _compute_menu_id(self): |
| 52 | + for record in self: |
| 53 | + if not record.group_ids: |
| 54 | + if record.menu_id: |
| 55 | + if record.menu_id.action: |
| 56 | + record.menu_id.action.unlink() |
| 57 | + record.menu_id.unlink() |
| 58 | + record.menu_id = False |
| 59 | + continue |
| 60 | + |
| 61 | + menu_groups = self.env.ref("base.group_system") | record.group_ids |
| 62 | + |
| 63 | + if not record.menu_id: |
| 64 | + action = self.env["ir.actions.act_url"].create( |
| 65 | + { |
| 66 | + "name": record.name, |
| 67 | + "url": f"/cross_connect_server/{record.id}", |
| 68 | + "target": "self", |
| 69 | + } |
| 70 | + ) |
| 71 | + |
| 72 | + record.menu_id = self.env["ir.ui.menu"].create( |
| 73 | + { |
| 74 | + "name": record.name, |
| 75 | + "action": f"ir.actions.act_url,{action.id}", # noqa |
| 76 | + "web_icon": "cross_connect_client,static/description/web_icon_data.png", |
| 77 | + "groups_id": [(6, 0, menu_groups.ids)], |
| 78 | + "sequence": 100, |
| 79 | + } |
| 80 | + ) |
| 81 | + else: |
| 82 | + record.menu_id.name = record.name |
| 83 | + record.menu_id.groups_id = [(6, 0, menu_groups.ids)] |
| 84 | + |
| 85 | + @api.depends("menu_id") |
| 86 | + def _compute_web_icon_data(self): |
| 87 | + for record in self: |
| 88 | + record.web_icon_data = record.menu_id.web_icon_data |
| 89 | + |
| 90 | + def _inverse_web_icon_data(self): |
| 91 | + for record in self: |
| 92 | + record.menu_id.web_icon_data = record.web_icon_data |
| 93 | + |
| 94 | + def _absolute_url_for(self, path): |
| 95 | + return f"{self.server_url.rstrip('/')}/cross_connect/{path.lstrip('/')}" |
| 96 | + |
| 97 | + def _request(self, method, url, headers=None, data=None): |
| 98 | + headers = headers or {} |
| 99 | + headers["api-key"] = self.api_key |
| 100 | + response = requests.request( |
| 101 | + method, |
| 102 | + self._absolute_url_for(url), |
| 103 | + headers=headers, |
| 104 | + json=data, |
| 105 | + timeout=10, |
| 106 | + ) |
| 107 | + response.raise_for_status() |
| 108 | + return response.json() |
| 109 | + |
| 110 | + def _get_cross_connect_url(self): |
| 111 | + self.ensure_one() |
| 112 | + groups = self.env.user.groups_id & self.group_ids |
| 113 | + if not groups: |
| 114 | + raise UserError(_("You are not allowed to access this server")) |
| 115 | + |
| 116 | + response = self._request( |
| 117 | + "POST", |
| 118 | + "/access", |
| 119 | + data={ |
| 120 | + "id": self.env.user.id, |
| 121 | + "name": self.env.user.name, |
| 122 | + "login": self.env.user.login, |
| 123 | + "lang": self.env.user.lang, |
| 124 | + "groups": [group.cross_connect_server_group_id for group in groups], |
| 125 | + }, |
| 126 | + ) |
| 127 | + client_id = response.get("client_id") |
| 128 | + token = response.get("token") |
| 129 | + if not token: |
| 130 | + raise UserError(_("Missing token")) |
| 131 | + |
| 132 | + return self._absolute_url_for(f"login/{client_id}/{token}") |
| 133 | + |
| 134 | + def _sync_groups(self): |
| 135 | + self.ensure_one() |
| 136 | + response = self._request("GET", "/sync") |
| 137 | + remote_groups = response.get("groups", []) |
| 138 | + # Removing groups that are not on the remote server |
| 139 | + remote_groups_ids = {remote_group["id"] for remote_group in remote_groups} |
| 140 | + self.group_ids.filtered( |
| 141 | + lambda group: group.cross_connect_server_group_id not in remote_groups_ids |
| 142 | + ).unlink() |
| 143 | + |
| 144 | + # Create or Update existing groups |
| 145 | + for remote_group in remote_groups: |
| 146 | + existing_group = self.group_ids.filtered( |
| 147 | + lambda group: group.cross_connect_server_group_id == remote_group["id"] |
| 148 | + ) |
| 149 | + if existing_group: |
| 150 | + existing_group.sudo().write( |
| 151 | + { |
| 152 | + "name": f"{self.name}: {remote_group['name']}", |
| 153 | + "comment": remote_group["comment"], |
| 154 | + } |
| 155 | + ) |
| 156 | + else: |
| 157 | + self.env["res.groups"].sudo().create( |
| 158 | + { |
| 159 | + "cross_connect_server_id": self.id, |
| 160 | + "cross_connect_server_group_id": remote_group["id"], |
| 161 | + "name": f"{self.name}: {remote_group['name']}", |
| 162 | + "comment": remote_group["comment"], |
| 163 | + } |
| 164 | + ) |
| 165 | + |
| 166 | + def action_sync(self): |
| 167 | + for record in self: |
| 168 | + record._sync_groups() |
| 169 | + |
| 170 | + def action_disable(self): |
| 171 | + for record in self: |
| 172 | + record.group_ids.unlink() |
| 173 | + |
| 174 | + @property |
| 175 | + def _server_env_fields(self): |
| 176 | + return {"api_key": {}} |
0 commit comments