mirror of
https://github.com/LorenEteval/Furious.git
synced 2026-09-22 23:08:08 +03:00
92 lines
2.7 KiB
Python
92 lines
2.7 KiB
Python
# Copyright (C) 2024–present Loren Eteval & contributors <loren.eteval@proton.me>
|
||
#
|
||
# This file is part of Furious.
|
||
#
|
||
# This program is free software: you can redistribute it and/or modify
|
||
# it under the terms of the GNU General Public License as published by
|
||
# the Free Software Foundation, either version 3 of the License, or
|
||
# (at your option) any later version.
|
||
#
|
||
# This program is distributed in the hope that it will be useful,
|
||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
# GNU General Public License for more details.
|
||
#
|
||
# You should have received a copy of the GNU General Public License
|
||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
|
||
"""Persist user-defined TUN settings."""
|
||
|
||
from __future__ import annotations
|
||
|
||
from Furious.Frozenlib import *
|
||
from Furious.Interface import *
|
||
from Furious.Models.Encoding import *
|
||
|
||
import logging
|
||
|
||
__all__ = ['UserTUNSettings']
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
registerAppSettings('CustomTUNSettings')
|
||
|
||
|
||
class UserTUNSettings(Mixins.CleanupOnExit, StorageBackend):
|
||
"""Manage persisted TUN customization values."""
|
||
|
||
def __init__(self, *args, **kwargs):
|
||
"""Initialize the UserTUNSettings."""
|
||
super().__init__(*args, **kwargs)
|
||
|
||
self._restoreFailed = False
|
||
|
||
def restore():
|
||
"""Restore the user TUN settings."""
|
||
raw = AppSettings.get('CustomTUNSettings')
|
||
|
||
if raw is None:
|
||
return {}
|
||
|
||
try:
|
||
data = UJSONEncoder.decode(PyBase64Encoder.decode(raw))
|
||
|
||
if isinstance(data, dict):
|
||
return data
|
||
|
||
raise TypeError('TUN settings repository root must be an object')
|
||
except Exception:
|
||
# Any non-exit exceptions
|
||
|
||
self._restoreFailed = True
|
||
logger.exception('failed to restore persisted TUN settings')
|
||
|
||
return {}
|
||
|
||
self._data = restore()
|
||
|
||
def sync(self):
|
||
"""Persist the current user TUN settings data."""
|
||
AppSettings.set(
|
||
'CustomTUNSettings',
|
||
PyBase64Encoder.encode(
|
||
UJSONEncoder.encode(self._data).encode(),
|
||
),
|
||
)
|
||
self._restoreFailed = False
|
||
|
||
def data(self) -> dict[str, str]:
|
||
"""Return the live mutable collection managed by this repository."""
|
||
return self._data
|
||
|
||
def cleanup(self):
|
||
"""Release resources owned by the user TUN settings."""
|
||
if self._restoreFailed and not self._data:
|
||
logger.warning(
|
||
'preserving unreadable persisted TUN settings during cleanup'
|
||
)
|
||
|
||
return
|
||
|
||
self.sync()
|