%PDF-
%PDF-
Mini Shell
Mini Shell
"""
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/>.
Copyright © 2019 Cloud Linux Software Inc.
This software is also available under ImunifyAV commercial license,
see <https://www.imunify360.com/legal/eula>
"""
import functools
import logging
from distutils.version import StrictVersion
from defence360agent.feature_management.constants import (
EXTENSION_DEFAULTS,
FEATURE_EXT_VARIABLES,
NATIVE_EXTENSION_NAME,
NATIVE_FEATURE_MANAGEMENT_PACKAGE_EXTENSION_FILES,
)
from defence360agent.feature_management.utils import reset_features
from defence360agent.subsys.panels.cpanel import cPanel
from defence360agent.subsys.panels.hosting_panel import HostingPanel
logger = logging.getLogger(__name__)
def supported(func):
"""Do not run a function on an unsupported panel"""
@functools.wraps(func)
async def wrapper(*args, **kwargs):
if await is_native_feature_management_supported():
return await func(*args, **kwargs)
logger.info("Native feature management is not supported")
return wrapper
async def is_native_feature_management_supported():
"""Whether we support native feature management on the panel."""
hp = HostingPanel()
if hp.NAME == cPanel.NAME:
try:
return StrictVersion(await hp.version()) >= StrictVersion("68.0")
except ValueError:
return False
return False
@supported
async def is_native_feature_management_enabled():
"""Whether the native feature management is enabled."""
hp = HostingPanel()
return (
hp.is_extension_installed(
pkgs=NATIVE_FEATURE_MANAGEMENT_PACKAGE_EXTENSION_FILES
)
and await hp.is_hook_installed()
)
@supported
async def enable_native_feature_management():
"""Enable native feature management."""
hp = HostingPanel()
# reset feature values for all existing users
await reset_features(
**{
feature: EXTENSION_DEFAULTS[pe_var]
for feature, pe_var in FEATURE_EXT_VARIABLES.items()
}
)
await hp.install_extension(
NATIVE_EXTENSION_NAME,
NATIVE_FEATURE_MANAGEMENT_PACKAGE_EXTENSION_FILES,
**EXTENSION_DEFAULTS,
)
logger.info("Imunify360 native feature management enabled.")
@supported
async def disable_native_feature_management():
"""Disable native feature management."""
if not await is_native_feature_management_enabled():
logger.info("No Imunify360 package extensions to disable.")
return True
await HostingPanel().uninstall_extension(
NATIVE_EXTENSION_NAME,
NATIVE_FEATURE_MANAGEMENT_PACKAGE_EXTENSION_FILES,
)
logger.info("Imunify360 native feature management disabled.")
return True # disabled successfully
Zerion Mini Shell 1.0