69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
import logging, json, os, re, sys, time
|
|
import gevent
|
|
from Plugin import PluginManager
|
|
from Config import config
|
|
from Debug import Debug
|
|
from .domainLookup import lookupDomain
|
|
|
|
allow_reload = False # No reload supported
|
|
|
|
log = logging.getLogger("Zeroname-localPlugin")
|
|
|
|
|
|
@PluginManager.registerTo("SiteManager")
|
|
class SiteManagerPlugin(object):
|
|
def load(self):
|
|
super(SiteManagerPlugin, self).load()
|
|
|
|
# Checks if its a valid address
|
|
def isAddress(self, address):
|
|
if self.isDomain(address):
|
|
return True
|
|
else:
|
|
return super(SiteManagerPlugin, self).isAddress(address)
|
|
|
|
|
|
# Return: True if the address is domain
|
|
def isDomain(self, address):
|
|
return re.match(r"(.*?)([A-Za-z0-9_-]+\.[A-Za-z0-9]+)$", address)
|
|
|
|
|
|
# Resolve domain
|
|
# Return: The address or None
|
|
def resolveDomain(self, domain):
|
|
return lookupDomain(domain)
|
|
|
|
|
|
# Return or create site and start download site files
|
|
# Return: Site or None if dns resolve failed
|
|
def need(self, address, all_file=True):
|
|
if self.isDomain(address): # Its looks like a domain
|
|
address_resolved = self.resolveDomain(address)
|
|
if address_resolved:
|
|
address = address_resolved
|
|
else:
|
|
return None
|
|
|
|
return super(SiteManagerPlugin, self).need(address, all_file)
|
|
|
|
|
|
# Return: Site object or None if not found
|
|
def get(self, address):
|
|
if self.sites == None: # Not loaded yet
|
|
self.load()
|
|
if self.isDomain(address): # Its looks like a domain
|
|
address_resolved = self.resolveDomain(address)
|
|
if address_resolved: # Domain found
|
|
site = self.sites.get(address_resolved)
|
|
if site:
|
|
site_domain = site.settings.get("domain")
|
|
if site_domain != address:
|
|
site.settings["domain"] = address
|
|
else: # Domain not found
|
|
site = self.sites.get(address)
|
|
|
|
else: # Access by site address
|
|
site = self.sites.get(address)
|
|
return site
|
|
|