from django.shortcuts import render from django.views import View from rest_framework.parsers import FormParser from rest_framework.views import APIView from core.lib.manage import threshold from core.views.manage.permissions import SuperUserRequiredMixin class ThresholdIRCStats(SuperUserRequiredMixin, View): stats_template = "manage/threshold/irc/overview/stats.html" def get(self, request): stats = threshold.get_irc_stats() context = {"stats": stats} return render(request, self.stats_template, context) class ThresholdIRCOverviewAlerts(SuperUserRequiredMixin, View): stats_template = "manage/threshold/irc/overview/alerts.html" def get(self, request): alerts = threshold.get_irc_alerts(request.user) context = {"alerts": alerts} return render(request, self.stats_template, context) class ThresholdIRCNetworks(SuperUserRequiredMixin, View): template_name = "manage/threshold/irc/overview/networks.html" def get(self, request): networks = threshold.get_irc_networks() context = {"networks": networks} return render(request, self.template_name, context) class ThresholdIRCNetworkDel(SuperUserRequiredMixin, View): template_name = "manage/threshold/irc/overview/networks.html" def delete(self, request, net): deleted = threshold.del_network(net) message = f"Deleted network {net}" message_class = "success" if not deleted["success"]: message = deleted["reason"] message_class = "danger" networks = threshold.get_irc_networks() context = { "networks": networks, "message": message, "class": message_class, } return render(request, self.template_name, context) class ThresholdIRCNetworkInfo(SuperUserRequiredMixin, View): template_name = "manage/threshold/irc/network/info.html" def get(self, request, net): network = threshold.get_irc_network(net) context = {"network": network} return render(request, self.template_name, context) class ThresholdIRCNetworkInfoEdit(SuperUserRequiredMixin, APIView): template_name = "manage/threshold/irc/network/edit-network.html" parser_classes = [FormParser] def get(self, request, net): """ Return the form to edit a network. """ network = threshold.get_irc_network(net) editable = ["auth", "host", "last", "port", "security"] context = { "net": net, "network": {k: v for k, v in network.items() if k in editable}, } return render(request, self.template_name, context) def post(self, request, net): """ Actually edit the network. Returns the info pane with a message about the success. """ template_name = "manage/threshold/irc/network/info.html" edited = threshold.edit_irc_network(net, request.data) if edited["success"]: message = "Successfully edited!" message_class = "success" else: if "reason" in edited: message = f"Error editing network: {edited['reason']}" else: message = "Error editing network" message_class = "danger" network = threshold.get_irc_network(net) context = {"network": network, "message": message, "class": message_class} return render(request, template_name, context) class ThresholdIRCNetworkRelays(SuperUserRequiredMixin, View): template_name = "manage/threshold/irc/network/relays.html" def get(self, request, net): relays = threshold.get_irc_relays(net) context = {"relays": relays["relays"]} return render(request, self.template_name, context) class ThresholdIRCNetworkRelayDel(SuperUserRequiredMixin, APIView): template_name = "manage/threshold/irc/network/relays.html" def delete(self, request, net, num): """ Delete a relay """ deleted = threshold.del_relay(net, num) if deleted["success"]: message = f"Deleted relay {num}" message_class = "success" else: message = deleted["reason"] message_class = "danger" relays = threshold.get_irc_relays(net) context = { "net": net, "message": message, "class": message_class, "relays": relays["relays"], } return render(request, self.template_name, context) class ThresholdIRCNetworkRelayStatus(SuperUserRequiredMixin, APIView): template_name = "manage/threshold/irc/network/relays.html" def get(self, request, net, num, status): """ Change the status of a relay. """ edited = threshold.change_network_status(net, num, status) if edited["success"]: message = "Successfully completed!" message_class = "success" else: if "reason" in edited: message = f"Error updating status: {edited['reason']}" else: message = "Error updating status" message_class = "danger" relays = threshold.get_irc_relays(net) context = { "relays": relays["relays"], "message": message, "class": message_class, } return render(request, self.template_name, context) class ThresholdIRCNetworkChannels(SuperUserRequiredMixin, APIView): """ List the channels a network is on. Allow parting and joining new ones. Re-render the same template if we join/part to make it simple. """ template_name = "manage/threshold/irc/network/channels.html" parser_classes = [FormParser] def get(self, request, net): """ Get list of channels for network. :param net: network name """ channels = threshold.get_irc_channels(net) context = {"net": net, "channels": channels["channels"]} return render(request, self.template_name, context) def delete(self, request, net, channel): """ Part a channel. :param net: network name :param channel: channel name """ parted = threshold.part_channel(net, channel) if parted["success"]: message = f"Requested part on relays: {', '.join(parted['relays'])}" message_class = "success" else: message = parted["reason"] message_class = "danger" channels = threshold.get_irc_channels(net) context = { "net": net, "channels": channels["channels"], "message": message, "class": message_class, } return render(request, self.template_name, context) def put(self, request, net): """ Join a channel. :param net: network name """ if "channel" not in request.data: message = "No channel specified" message_class = "danger" else: channel = request.data["channel"] joined = threshold.join_channel(net, channel) if joined["success"]: message = f"Requested join on relay: {joined['relays']}" message_class = "success" else: message = joined["reason"] message_class = "danger" channels = threshold.get_irc_channels(net) context = { "net": net, "channels": channels["channels"], "message": message, "class": message_class, } return render(request, self.template_name, context) class ThresholdIRCAliases(SuperUserRequiredMixin, APIView): template_name = "manage/threshold/irc/overview/aliases.html" def get(self, request): """ Get aliases. """ aliases = threshold.get_aliases() context = { "aliases": aliases["aliases"], } return render(request, self.template_name, context) class ThresholdIRCActions(SuperUserRequiredMixin, APIView): template_name = "manage/threshold/irc/overview/actions.html" def get(self, request): """ Get actions pane. """ return render(request, self.template_name) class ThresholdIRCActionsAddNetwork(SuperUserRequiredMixin, APIView): template_name = "manage/threshold/irc/overview/modals/add-network.html" parser_classes = [FormParser] def get(self, request): """ Get actions pane. """ return render(request, self.template_name) def put(self, request): """ Create a network. """ template_name = "manage/threshold/irc/overview/networks.html" created = threshold.create_network(request.data) message = "Network created successfully." message_class = "success" if not created["success"]: message = created["reason"] message_class = "danger" networks = threshold.get_irc_networks() context = { "networks": networks, "message": message, "class": message_class, } return render(request, template_name, context) class ThresholdIRCAliasesEdit(SuperUserRequiredMixin, APIView): template_name = "manage/threshold/irc/overview/modals/edit-aliases.html" parser_classes = [FormParser] def get(self, request): """ Get aliases edit form. """ aliases = threshold.get_aliases() context = { "aliases": aliases["aliases"], } return render(request, self.template_name, context) def post(self, request): """ Set the aliases """ message = "Successfully updated aliases" message_class = "success" new_aliases = {} for item in request.data: if "|" not in item: continue spl = item.split("|") if len(spl) != 2: message = f"Invalid item: {item}" message_class = "danger" break num, field = spl value = request.data[item] if field == "emails": value = request.data[item].split("\n") for email in value: # Check field isn't just empty if len(email) != 0: # Simple check for valid email if "@" not in email: message = f"Invalid email: {email}" message_class = "danger" break if num not in new_aliases: new_aliases[num] = {field: value} else: new_aliases[num][field] = value if not message_class == "danger": updated = threshold.update_aliases(new_aliases) if not updated: message = updated["reason"] message_class = "danger" template_name = "manage/threshold/irc/overview/aliases.html" aliases = threshold.get_aliases() context = { "aliases": aliases["aliases"], "message": message, "class": message_class, } return render(request, template_name, context) class ThresholdIRCNetworkActions(SuperUserRequiredMixin, View): template_name = "manage/threshold/irc/network/actions.html" def get(self, request, net): """ Get actions page. """ context = {"net": net} return render(request, self.template_name, context) class ThresholdIRCNetworkActionsAuto(SuperUserRequiredMixin, View): template_name = "manage/threshold/irc/network/actions.html" def post(self, request, net): """ Run network automation. """ ran = threshold.run_auto(net) if ran["success"]: message = f"Ran auto on network {net}" message_class = "success" if "message" in ran: message = ran["message"] else: message = ran["reason"] message_class = "danger" context = { "net": net, "message": message, "class": message_class, } return render(request, self.template_name, context) class ThresholdIRCNetworkActionsList(SuperUserRequiredMixin, View): template_name = "manage/threshold/irc/network/actions.html" def post(self, request, net): """ Run LIST on network. """ ran = threshold.run_list(net) if ran["success"]: message = f"Ran list on network {net}" message_class = "success" if "message" in ran: message = ran["message"] else: message = ran["reason"] message_class = "danger" context = { "net": net, "message": message, "class": message_class, } return render(request, self.template_name, context) class ThresholdIRCNetworkActionsRelay(SuperUserRequiredMixin, APIView): template_name = "manage/threshold/irc/network/actions.html" parser_classes = [FormParser] def put(self, request, net): """ Create a relay """ if "num" not in request.data: num = 0 else: num = request.data["num"] if not num.isdigit(): message = "Num is not an integer" message_class = "danger" else: num = int(num) created = threshold.add_relay(net, num) if created["success"]: id = created["id"] alias = created["alias"] message = f"Created new relay {id} with alias {alias}" message_class = "success" else: message = created["reason"] message_class = "danger" context = { "net": net, "message": message, "class": message_class, } return render(request, self.template_name, context) class ThresholdIRCSendMessage(SuperUserRequiredMixin, APIView): parser_classes = [FormParser] template_name = "partials/context-input.html" def put(self, request, net, num, channel, nick=None): """ Send a message """ if "msg" not in request.data: message = "No message to send" message_class = "danger" return render( request, self.template_name, {"message": message, "class": message_class}, ) print("IRC", nick) if nick: messaged = threshold.send_irc_message( net, num, channel, request.data["msg"], nick=nick ) else: messaged = threshold.send_irc_message( net, num, channel, request.data["msg"] ) if not messaged: message = "Failed to send message" message_class = "danger" elif messaged["success"]: message = "Send message" message_class = "success" else: message = messaged["reason"] message_class = "danger" print("ERROR", message) context = { "net": net, "message": message, "class": message_class, } return render(request, self.template_name, context)