Skip to content

Commit

Permalink
Merge branch 'master' of github.com:yamikaitou/sc-cogs
Browse files Browse the repository at this point in the history
  • Loading branch information
@yamikaitou
yamikaitou committed May 9, 2018
2 parents cb71376 + a26f99c commit 5acbe63
Show file tree
Hide file tree
Showing 2 changed files with 245 additions and 17 deletions.
2 changes: 2 additions & 0 deletions .gitignore
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
cogs
.idea
260 changes: 243 additions & 17 deletions customdns/customdns.py
View file
Original file line number Diff line number Diff line change
@@ -1,45 +1,54 @@
import discord
from discord.ext import commands
from discord import Embed
from __main__ import send_cmd_help
import asyncio
import aiodns
from cogs.utils.dataIO import dataIO
from cogs.utils import checks
from google import cloud
from google.cloud import dns
from google.oauth2 import service_account
import os
import datetime


class CustomDNS:
"""My custom cog that does stuff!"""

def __init__(self, bot):
self.bot = bot
self.settings = dataIO.load_json(os.path.join("data", "dns", "settings.json"))
self.gmod = dataIO.load_json(os.path.join("data", "dns", "gmod.json"))
self.csgo = dataIO.load_json(os.path.join("data", "dns", "csgo.json"))
self.tf2 = dataIO.load_json(os.path.join("data", "dns", "tf2.json"))
self.unk = dataIO.load_json(os.path.join("data", "dns", "unk.json"))
self.creds = service_account.Credentials.from_service_account_file(os.path.join("data", "dns", "gcloud.json"))
loop = asyncio.get_event_loop()
self.resolver = aiodns.DNSResolver(loop=loop)

@commands.group(pass_context=True, no_pm=True)
@checks.is_owner()
async def dnsconfig(self, ctx):
"""Config GCloud DNS settings"""
if ctx.invoked_subcommand is None:
await self.bot.say("Project: {}\nDomain: {}\nZone: {}".format(self.settings['project'][0], self.settings['domain'][0], self.settings['zone'][0]))

await self.bot.say(
"Project: {}\nDomain: {}\nZone: {}".format(self.settings['project'][0], self.settings['domain'][0],
self.settings['zone'][0]))

@dnsconfig.command(pass_context=True, no_pm=True)
@checks.is_owner()
async def project(self, ctx, *value):
"""Set the Project-ID"""
self.settings["project"] = value
dataIO.save_json(os.path.join("data", "dns", "settings.json"), self.settings)

@dnsconfig.command(pass_context=True, no_pm=True)
@checks.is_owner()
async def domain(self, ctx, *value):
"""Set the Domain """
self.settings["domain"] = value
dataIO.save_json(os.path.join("data", "dns", "settings.json"), self.settings)

@dnsconfig.command(pass_context=True, no_pm=True)
@checks.is_owner()
async def zone(self, ctx, *value):
Expand All @@ -48,30 +57,227 @@ async def zone(self, ctx, *value):
dataIO.save_json(os.path.join("data", "dns", "settings.json"), self.settings)

@commands.group(pass_context=True, no_pm=True, name="dns")
@checks.serverowner_or_permissions(administrator=True)
async def _dns(self, ctx):
"""Manage the DNS entries"""
if checks.role_or_permissions(ctx, lambda r: r.name.lower() in ('monkey','server manager','owner')) is False:
return False

if ctx.invoked_subcommand is None:
server = ctx.message.server
await send_cmd_help(ctx)




@_dns.command(pass_context=True, no_pm=True)
@checks.serverowner_or_permissions(administrator=True)
async def nfo(self, ctx, ident, sub):
"""Create a DNS entry from an NFO name"""

async def nfo(self, ctx, sub, ident):
"""Create a DNS entry from an NFO name
sub = DNS prefix, the part before .scgc.xyz
ident = NFO Prefix/Identifier, the part before .game.nfoservers.com"""
if checks.role_or_permissions(ctx, lambda r: r.name.lower() in ('monkey','server manager','owner')) is False:
return False

ip = await self.resolver.query("{}.game.nfoservers.com".format(ident), 'A')
client = cloud.dns.Client(project=self.settings["project"][0], credentials=self.creds)
client = dns.Client(project=self.settings["project"][0], credentials=self.creds)
zone = client.zone(self.settings['zone'][0], self.settings['domain'][0])
record_set = zone.resource_record_set('{}.{}.'.format(sub, self.settings['domain'][0]), 'A', 60 * 60,
[ip[0].host])
changes = zone.changes()
changes.add_record_set(record_set)
changes.create()
while changes.status != 'done':
asyncio.sleep(60)
changes.reload()

self.unk[sub] = {'dns':sub, 'nfo':ident, 'port':0, 'name':'unk'}
dataIO.save_json(os.path.join("data", "dns", "unk.json"), self.unk)
self.settings['last'] = int(datetime.datetime.utcnow().timestamp())
dataIO.save_json(os.path.join("data", "dns", "settings.json"), self.settings)
await self.bot.say("DNS Entry Created. Don't forget to update the server list, !serverlist edit {}".format(sub))

@_dns.command(pass_context=True, no_pm=True)
async def a(self, ctx, sub, ip):
"""Create or Modify a DNS A Record
sub = DNS prefix, the part before .scgc.xyz
ip = IP the record should point to"""
if checks.role_or_permissions(ctx, lambda r: r.name.lower() in ('monkey','server manager','owner')) is False:
return False

client = dns.Client(project=self.settings["project"][0], credentials=self.creds)
zone = client.zone(self.settings['zone'][0], self.settings['domain'][0])
record_set = zone.resource_record_set('{}.{}.'.format(sub, self.settings['domain'][0]), 'A', 60*60*2, [ip[0].host])
record_set = zone.resource_record_set('{}.{}.'.format(sub, self.settings['domain'][0]), 'A', 60 * 60 * 2,
[ip])
changes = zone.changes()
changes.add_record_set(record_set)
changes.create()
while changes.status != 'done':
asyncio.sleep(60)
changes.reload()
await self.bot.say("DNS Entry Created")


self.unk[sub] = {'dns':sub, 'nfo':ip, 'port': 0, 'name': 'unk'}
dataIO.save_json(os.path.join("data", "dns", "unk.json"), self.unk)
self.settings['last'] = int(datetime.datetime.utcnow().timestamp())
dataIO.save_json(os.path.join("data", "dns", "settings.json"), self.settings)
await self.bot.say("DNS Entry Created. Don't forget to update the server list if needed, !serverlist edit {}".format(sub))

@commands.group(pass_context=True, no_pm=True, name="serverlist")
async def serverlist(self, ctx):
"""Shows a list of our servers"""
if ctx.invoked_subcommand is None:
embed = Embed(colour=discord.Colour(0x426156), timestamp=datetime.datetime.fromtimestamp(self.settings['last']))

embed.set_author(name="SuperCentral Server List", url="https://supercentral.co",
icon_url="https://supercentral.co/srvmgr/images/sclogo.jpg")
embed.set_footer(text="Last Updated")

if len(self.gmod) != 0:
val = ""
for server in self.gmod.items():
if server[1]['port'] not in (27015,0):
val += "{}.scgc.xyz - {}\n".format(server[1]['dns'], server[1]['name'])
else:
val += "{}.scgc.xyz:{} - {}\n".format(server[1]['dns'], server[1]['port'], server[1]['name'])

embed.add_field(name="Garry's Mod",
value=val,
inline=False)

if len(self.csgo) != 0:
val = ""
for server in self.csgo.items():
if server[1]['port'] not in (27015,0):
val += "{}.scgc.xyz - {}\n".format(server[1]['dns'], server[1]['name'])
else:
val += "{}.scgc.xyz:{} - {}\n".format(server[1]['dns'], server[1]['port'], server[1]['name'])

embed.add_field(name="Counter Strike Global Offensive",
value=val,
inline=False)

if len(self.tf2) != 0:
val = ""
for server in self.tf2.items():
if server[1]['port'] not in (27015,0):
val += "{}.scgc.xyz - {}\n".format(server[1]['dns'], server[1]['name'])
else:
val += "{}.scgc.xyz:{} - {}\n".format(server[1]['dns'], server[1]['port'], server[1]['name'])

embed.add_field(name="Team Fortress 2",
value=val,
inline=False)

if len(self.unk) != 0:
val = ""
for server in self.unk.items():
if server[1]['port'] == 0:
val += "{}.scgc.xyz - {}\n".format(server[1]['dns'], server[1]['name'])
else:
val += "{}.scgc.xyz:{} - {}\n".format(server[1]['dns'], server[1]['port'], server[1]['name'])

embed.add_field(name="Undefined",
value=val,
inline=False)

embed.add_field(name="Voice Servers",
value="discord.scgc.xyz - Discord (the one you are in)\nvoice.scgc.xyz - Teamspeak3",
inline=False)

await self.bot.say(embed=embed)

@serverlist.command(pass_context=True, no_pm=True)
async def edit(self, ctx, sub, game, port, name):
"""Modify entries on the Server List
sub = DNS prefix, the part before .scgc.xyz
game = gmod, csgo or tf2
port = Server Port (will be displayed if something other than 27015
name = Name to display on the !serverlist. Enclose in quotes if needed"""
if checks.role_or_permissions(ctx, lambda r: r.name.lower() in ('monkey','server manager','owner')) is False:
return False

if game not in ('gmod','csgo','tf2'):
await self.bot.say("Invalid game, please use gmod, csgo or tf2")

if sub in self.unk:
val = self.unk.pop(sub)
val['port'] = port
val['name'] = name
if game == 'gmod':
self.gmod[sub] = val
dataIO.save_json(os.path.join("data", "dns", "gmod.json"), self.gmod)
elif game == 'csgo':
self.csgo[sub] = val
dataIO.save_json(os.path.join("data", "dns", "csgo.json"), self.csgo)
elif game == 'tf2':
self.tf2[sub] = val
dataIO.save_json(os.path.join("data", "dns", "tf2.json"), self.tf2)

dataIO.save_json(os.path.join("data", "dns", "unk.json"), self.unk)
self.settings['last'] = int(datetime.datetime.utcnow().timestamp())
dataIO.save_json(os.path.join("data", "dns", "settings.json"), self.settings)
await self.bot.say("Edit successful")
elif sub in self.gmod:
if sub == 'gmod':
self.gmod[sub]['port'] = port
self.gmod[sub]['name'] = name
dataIO.save_json(os.path.join("data", "dns", "gmod.json"), self.gmod)
else:
val = self.gmod.pop(sub)
val['port'] = port
val['name'] = name
if game == 'csgo':
self.csgo[sub] = val
dataIO.save_json(os.path.join("data", "dns", "csgo.json"), self.csgo)
elif game == 'tf2':
self.tf2[sub] = val
dataIO.save_json(os.path.join("data", "dns", "tf2.json"), self.tf2)

dataIO.save_json(os.path.join("data", "dns", "gmod.json"), self.unk)
self.settings['last'] = int(datetime.datetime.utcnow().timestamp())
dataIO.save_json(os.path.join("data", "dns", "settings.json"), self.settings)
await self.bot.say("Edit successful")
elif sub in self.csgo:
if sub == 'csgo':
self.csgo[sub]['port'] = port
self.csgo[sub]['name'] = name
dataIO.save_json(os.path.join("data", "dns", "csgo.json"), self.csgo)
else:
val = self.csgo.pop(sub)
val['port'] = port
val['name'] = name
if game == 'gmod':
self.gmod[sub] = val
dataIO.save_json(os.path.join("data", "dns", "gmod.json"), self.gmod)
elif game == 'tf2':
self.tf2[sub] = val
dataIO.save_json(os.path.join("data", "dns", "tf2.json"), self.tf2)

dataIO.save_json(os.path.join("data", "dns", "csgo.json"), self.unk)
self.settings['last'] = int(datetime.datetime.utcnow().timestamp())
dataIO.save_json(os.path.join("data", "dns", "settings.json"), self.settings)
await self.bot.say("Edit successful")
elif sub in self.tf2:
if sub == 'tf2':
self.tf2[sub]['port'] = port
self.tf2[sub]['name'] = name
dataIO.save_json(os.path.join("data", "dns", "tf2.json"), self.tf2)
else:
val = self.tf2.pop(sub)
val['port'] = port
val['name'] = name
if game == 'csgo':
self.csgo[sub] = val
dataIO.save_json(os.path.join("data", "dns", "csgo.json"), self.csgo)
elif game == 'gmod':
self.gmod[sub] = val
dataIO.save_json(os.path.join("data", "dns", "gmod.json"), self.gmod)

dataIO.save_json(os.path.join("data", "dns", "tf2.json"), self.unk)
self.settings['last'] = int(datetime.datetime.utcnow().timestamp())
dataIO.save_json(os.path.join("data", "dns", "settings.json"), self.settings)
await self.bot.say("Edit successful")
else:
await self.bot.say("No server list entry found for that DNS name")



def check_folders():
if not os.path.exists("data/dns"):
Expand All @@ -80,12 +286,32 @@ def check_folders():


def check_files():

f = "data/dns/settings.json"
if not dataIO.is_valid_json(f):
print("Creating default dns's settings.json...")
dataIO.save_json(f, {})

f = "data/dns/gmod.json"
if not dataIO.is_valid_json(f):
print("Creating default dns's gmod.json...")
dataIO.save_json(f, {})

f = "data/dns/csgo.json"
if not dataIO.is_valid_json(f):
print("Creating default dns's csgo.json...")
dataIO.save_json(f, {})

f = "data/dns/tf2.json"
if not dataIO.is_valid_json(f):
print("Creating default dns's tf2.json...")
dataIO.save_json(f, {})

f = "data/dns/unk.json"
if not dataIO.is_valid_json(f):
print("Creating default dns's unk.json...")
dataIO.save_json(f, {})


def setup(bot):
check_folders()
check_files()
Expand Down

0 comments on commit 5acbe63

Please sign in to comment.