mirror of https://github.com/Aidaho12/haproxy-wi
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
206 lines
8.5 KiB
206 lines
8.5 KiB
from typing import Literal
|
|
from typing import Union
|
|
|
|
from flask.views import MethodView
|
|
from flask_pydantic import validate
|
|
from flask_jwt_extended import jwt_required
|
|
from flask import jsonify, g
|
|
from playhouse.shortcuts import model_to_dict
|
|
|
|
import app.modules.db.server as server_sql
|
|
import app.modules.db.service as service_sql
|
|
import app.modules.db.checker as checker_sql
|
|
import app.modules.roxywi.common as roxywi_common
|
|
from app.middleware import get_user_params, page_for_admin, check_group, check_services
|
|
from app.modules.roxywi.class_models import Checker, BaseResponse
|
|
from app.modules.db.db_model import CheckerSetting
|
|
from app.modules.common.common_classes import SupportClass
|
|
|
|
|
|
class CheckerView(MethodView):
|
|
methods = ["GET", "POST"]
|
|
decorators = [jwt_required(), get_user_params(), check_services, page_for_admin(level=3), check_group()]
|
|
|
|
@staticmethod
|
|
def get(service: Literal['haproxy', 'nginx', 'apache', 'keepalived'], server_id: Union[int, str]):
|
|
"""
|
|
Retrieves the status of specific checker services.
|
|
---
|
|
tags:
|
|
- Service Tools
|
|
parameters:
|
|
- in: path
|
|
name: service
|
|
type: 'integer'
|
|
required: true
|
|
description: The type of service (haproxy, nginx, apache, keepalived)
|
|
- in: path
|
|
name: server_id
|
|
type: 'integer'
|
|
required: true
|
|
description: The ID or IP of the server
|
|
responses:
|
|
200:
|
|
description: Successful operation
|
|
schema:
|
|
type: 'object'
|
|
properties:
|
|
server_id:
|
|
type: 'integer'
|
|
description: 'Server identifier'
|
|
backend_alert:
|
|
type: 'integer'
|
|
description: 'Alert status for backends. Only for HAProxy and Keepalived services'
|
|
maxconn_alert:
|
|
type: 'integer'
|
|
description: 'Alert status for max connections. Only for HAProxy service'
|
|
email:
|
|
type: 'integer'
|
|
description: 'Status flag for email notifications'
|
|
mm_id:
|
|
type: 'integer'
|
|
description: 'Identifier for Mattermost notifications'
|
|
pd_id:
|
|
type: 'integer'
|
|
description: 'Identifier for PagerDuty notifications'
|
|
service_alert:
|
|
type: 'integer'
|
|
description: 'Alert status for services'
|
|
slack_id:
|
|
type: 'integer'
|
|
description: 'Identifier for Slack notifications'
|
|
telegram_id:
|
|
type: 'integer'
|
|
description: 'Identifier for Telegram notifications'
|
|
checker:
|
|
type: 'integer'
|
|
description: 'Is Checker tools enabled for this service'
|
|
metrics:
|
|
type: 'integer'
|
|
description: 'Is Metrics tools enabled for this service'
|
|
auto_start:
|
|
type: 'integer'
|
|
description: 'Is Auto start tools enabled for this service'
|
|
default:
|
|
description: Unexpected error
|
|
"""
|
|
service_id = service_sql.select_service_id_by_slug(service)
|
|
|
|
try:
|
|
server_id = SupportClass().return_server_ip_or_id(server_id)
|
|
except Exception as e:
|
|
return roxywi_common.handler_exceptions_for_json_data(e, '')
|
|
|
|
try:
|
|
server = server_sql.get_server_by_id(server_id)
|
|
except Exception as e:
|
|
return roxywi_common.handler_exceptions_for_json_data(e, 'Cannot find server')
|
|
|
|
try:
|
|
roxywi_common.is_user_has_access_to_group(g.user_params['user_id'], server.group_id)
|
|
except Exception as e:
|
|
return roxywi_common.handler_exceptions_for_json_data(e, '')
|
|
|
|
try:
|
|
checker_settings = checker_sql.select_checker_settings_for_server(service_id, server_id)
|
|
except Exception as e:
|
|
return roxywi_common.handler_exceptions_for_json_data(e, 'Cannot get settings')
|
|
checker_setting = [model_to_dict(checker, recurse=False, exclude=[CheckerSetting.service_id, CheckerSetting.id]) for checker in checker_settings]
|
|
server = model_to_dict(server)
|
|
checker_setting[0]['checker'] = server[f'{service}_alert']
|
|
checker_setting[0]['metrics'] = server[f'{service}_metrics']
|
|
checker_setting[0]['auto_start'] = server[f'{service}_active']
|
|
return jsonify(checker_setting)
|
|
|
|
@validate(body=Checker)
|
|
def post(self, service: Literal['haproxy', 'nginx', 'apache', 'keepalived'], server_id: Union[int, str], body: Checker):
|
|
"""
|
|
Update the Tools settings of specific service.
|
|
---
|
|
tags:
|
|
- Service Tools
|
|
parameters:
|
|
- in: path
|
|
name: service
|
|
type: string
|
|
required: true
|
|
description: The type of service (haproxy, nginx, apache, keepalived)
|
|
- in: path
|
|
name: server_id
|
|
type: string
|
|
required: true
|
|
description: The ID or IP of the server
|
|
- in: body
|
|
required: true
|
|
name: body
|
|
description: The settings of the checker service
|
|
schema:
|
|
type: object
|
|
properties:
|
|
auto_start:
|
|
type: 'integer'
|
|
description: 'Sets if Auto start tools is enabled for this service'
|
|
backend_alert:
|
|
type: 'integer'
|
|
description: 'Sets the Alert status for backends. Only for HAProxy and Keepalived services'
|
|
checker:
|
|
type: 'integer'
|
|
description: 'Sets if Checker tools is enabled for this service'
|
|
email:
|
|
type: 'integer'
|
|
description: 'Sets the status flag for email notifications'
|
|
maxconn_alert:
|
|
type: 'integer'
|
|
description: 'Sets the Alert status for max connections. Only for HAProxy service'
|
|
metrics:
|
|
type: 'integer'
|
|
description: 'Sets if Metrics tools is enabled for this service'
|
|
mm_id:
|
|
type: 'integer'
|
|
description: 'Sets the Identifier for Mattermost notifications'
|
|
pd_id:
|
|
type: 'integer'
|
|
description: 'Sets the Identifier for PagerDuty notifications'
|
|
service_alert:
|
|
type: 'integer'
|
|
description: 'Sets the Alert status for services'
|
|
slack_id:
|
|
type: 'integer'
|
|
description: 'Sets the Identifier for Slack notifications'
|
|
telegram_id:
|
|
type: 'integer'
|
|
description: 'Sets the Identifier for Telegram notifications'
|
|
responses:
|
|
201:
|
|
description: Successful operation, settings updated
|
|
default:
|
|
description: Unexpected error
|
|
"""
|
|
service_id = service_sql.select_service_id_by_slug(service)
|
|
|
|
try:
|
|
server_id = SupportClass().return_server_ip_or_id(server_id)
|
|
except Exception as e:
|
|
return roxywi_common.handler_exceptions_for_json_data(e, '')
|
|
|
|
try:
|
|
server = server_sql.get_server_by_id(server_id)
|
|
except Exception as e:
|
|
return roxywi_common.handler_exceptions_for_json_data(e, 'Cannot find server')
|
|
|
|
try:
|
|
roxywi_common.is_user_has_access_to_group(g.user_params['user_id'], server.group_id)
|
|
except Exception as e:
|
|
return roxywi_common.handler_exceptions_for_json_data(e, '')
|
|
|
|
kwargs = body.model_dump(mode='json', exclude={'metrics', 'auto_start', 'checker'})
|
|
try:
|
|
checker_sql.update_checker_setting_for_server(service_id, server_id, **kwargs)
|
|
except Exception as e:
|
|
return roxywi_common.handler_exceptions_for_json_data(e, 'Cannot update Checker settings')
|
|
try:
|
|
service_sql.update_hapwi_server(server_id, body.checker, body.metrics, body.auto_start, service)
|
|
except Exception as e:
|
|
return roxywi_common.handler_exceptions_for_json_data(e, 'Cannot update Service settings')
|
|
return BaseResponse().model_dump(), 201
|