mirror of
https://github.com/oddluck/limnoria-plugins.git
synced 2025-04-25 12:31:07 -05:00
221 lines
7.6 KiB
Python
221 lines
7.6 KiB
Python
###
|
|
# Copyright (c) SpiderDave
|
|
# Copyright (c) 2020, oddluck <oddluck@riseup.net>
|
|
# All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are met:
|
|
#
|
|
# * Redistributions of source code must retain the above copyright notice,
|
|
# this list of conditions, and the following disclaimer.
|
|
# * Redistributions in binary form must reproduce the above copyright notice,
|
|
# this list of conditions, and the following disclaimer in the
|
|
# documentation and/or other materials provided with the distribution.
|
|
# * Neither the name of the author of this software nor the name of
|
|
# contributors to this software may be used to endorse or promote products
|
|
# derived from this software without specific prior written consent.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
# POSSIBILITY OF SUCH DAMAGE.
|
|
###
|
|
|
|
import time
|
|
import supybot.conf as conf
|
|
import supybot.utils as utils
|
|
from supybot.commands import *
|
|
import supybot.ircmsgs as ircmsgs
|
|
import supybot.ircutils as ircutils
|
|
import supybot.callbacks as callbacks
|
|
import os, errno
|
|
import supybot.registry as registry
|
|
import supybot.ircdb as ircdb
|
|
import tarfile
|
|
import gzip
|
|
import socket
|
|
|
|
try:
|
|
import geoip2.database
|
|
except:
|
|
raise ImportError("The Geo plugin requires geoip2 be installed. Load aborted.")
|
|
|
|
|
|
class Geo(callbacks.Plugin):
|
|
threaded = True
|
|
"""
|
|
Geolocation using GeoLiteCity
|
|
"""
|
|
|
|
def make_sure_path_exists(self, path):
|
|
try:
|
|
os.makedirs(path)
|
|
except OSError as exception:
|
|
if exception.errno != errno.EEXIST:
|
|
raise
|
|
|
|
def geo(self, irc, msg, args, stuff):
|
|
"""[<ip> | <host> | <nick>]
|
|
Geolocation of an ip, hostname, or nick. Nick must be in channel.
|
|
"""
|
|
channel = msg.args[0]
|
|
try:
|
|
reader = geoip2.database.Reader(
|
|
"%s%sgeo%sGeoLite2-City.mmdb"
|
|
% (conf.supybot.directories.data(), os.sep, os.sep)
|
|
)
|
|
except:
|
|
irc.reply(
|
|
"Error: GeoLite2-City database not found, attempting to update..."
|
|
)
|
|
try:
|
|
self.do_update()
|
|
irc.reply("Update finished.")
|
|
except:
|
|
irc.reply("Update failed.")
|
|
return
|
|
if not irc.isChannel(channel):
|
|
private = True
|
|
else:
|
|
private = False
|
|
if not private and stuff.lower() in (
|
|
nick.lower() for nick in list(irc.state.channels[channel].users)
|
|
):
|
|
try:
|
|
stuff = irc.state.nickToHostmask(stuff).split("@")[1]
|
|
ip = socket.gethostbyname(stuff)
|
|
except:
|
|
irc.reply("Invalid hostname {0}".format(stuff))
|
|
return
|
|
elif not utils.net.isIP(stuff):
|
|
try:
|
|
ip = socket.gethostbyname(stuff)
|
|
except:
|
|
irc.reply("Invalid nick/hostname {0}".format(stuff))
|
|
return
|
|
elif utils.net.isIP(stuff):
|
|
ip = stuff
|
|
else:
|
|
irc.reply("invalid nick or hostname/ip {0}".format(stuff))
|
|
try:
|
|
res = reader.city(ip)
|
|
if res:
|
|
irc.reply(
|
|
"%s, %s, %s"
|
|
% (
|
|
res.city.name,
|
|
res.subdivisions.most_specific.name,
|
|
res.country.name,
|
|
)
|
|
)
|
|
else:
|
|
irc.reply("No results found")
|
|
except:
|
|
irc.reply("No results found")
|
|
|
|
geo = wrap(geo, ["text"])
|
|
|
|
def update(self, irc, msg, args):
|
|
"""
|
|
Update geoip database
|
|
"""
|
|
try:
|
|
irc.reply("Updating data file...")
|
|
self.do_update()
|
|
irc.reply("Update finished.")
|
|
except:
|
|
irc.reply("Update failed.")
|
|
return
|
|
|
|
update = wrap(update)
|
|
|
|
def do_update(self):
|
|
"""update the geo files"""
|
|
now = int(time.time())
|
|
try:
|
|
lastupdate = self.registryValue("datalastupdated")
|
|
self.log.info("Last update: %s seconds ago." % lastupdate)
|
|
except registry.NonExistentRegistryEntry:
|
|
self.log.info(
|
|
"supybot.plugins.%s.datalastupdate not set. Creating..." % self.name
|
|
)
|
|
conf.registerGlobalValue(
|
|
conf.supybot.plugins.get(self.name),
|
|
"datalastupdated",
|
|
registry.PositiveInteger(
|
|
1,
|
|
"""An integer representing the time since epoch the .dat file was last updated.""",
|
|
),
|
|
)
|
|
self.log.info("...success!")
|
|
lastupdate = 1
|
|
self.log.info("Last update: Unknown/Never")
|
|
|
|
# if (now-lastupdate)>604800: # updated weekly
|
|
if 1 == 1:
|
|
self.setRegistryValue("datalastupdated", now)
|
|
self.log.info("Starting update of Geo data files...")
|
|
self.getfile()
|
|
return
|
|
|
|
def getfile(self):
|
|
"""grabs the data file"""
|
|
license = self.registryValue("licenseKey")
|
|
u = "https://download.maxmind.com/app/geoip_download?edition_id=GeoLite2-City&license_key={}&suffix=tar.gz".format(
|
|
license
|
|
)
|
|
f = "%s%sgeo%sGeoLite2-City.tar.gz" % (
|
|
conf.supybot.directories.data(),
|
|
os.sep,
|
|
os.sep,
|
|
)
|
|
f2 = "%s%sgeo%sGeoLite2-City.tar" % (
|
|
conf.supybot.directories.data(),
|
|
os.sep,
|
|
os.sep,
|
|
)
|
|
self.make_sure_path_exists(
|
|
r"%s%sgeo" % (conf.supybot.directories.data(), os.sep)
|
|
)
|
|
path = "%s%sgeo" % (conf.supybot.directories.data(), os.sep)
|
|
self.log.info("Starting download: %s" % f)
|
|
h = utils.web.getUrl(u)
|
|
if h:
|
|
tempfile = open(f, "w+b")
|
|
if tempfile:
|
|
tempfile.write(h)
|
|
tempfile.close()
|
|
self.log.info("Completed: %s" % f)
|
|
self.log.info("Unzipping: %s" % f)
|
|
f_in = gzip.open(f, "rb")
|
|
f_out = open(f2, "wb")
|
|
f_out.writelines(f_in)
|
|
f_out.close()
|
|
f_in.close()
|
|
self.log.info("Finished Unzipping: %s" % f)
|
|
self.log.info("Untarring: %s" % f2)
|
|
tar = tarfile.open(f2)
|
|
tar.getmembers()
|
|
for member in tar.getmembers():
|
|
if "GeoLite2-City.mmdb" in member.name:
|
|
member.name = "GeoLite2-City.mmdb"
|
|
self.log.info(member.name)
|
|
tar.extract(member, path=path)
|
|
self.log.info("Finished Untarring: %s" % f2)
|
|
tar.close()
|
|
os.remove(f)
|
|
os.remove(f2)
|
|
else:
|
|
self.log.info("Could not download: %s" % f)
|
|
return
|
|
|
|
|
|
Class = Geo
|