98c6b347b4
* Add files via upload * Add files via upload * Logging functionality (Resolves #146) * Created customized logger and setup file * Start replacing prints * Custom StreamHandler to allow '\r' as line terminator and updated more prints * Remove setup.py * Logger functionality to write red lines and records without format * Possibility to set logging level when logging without format and usage of debug level instead of verboseOutput * Replace utils logger function calls * Fixes * Import missing info color * Move xsstrike.py imports to properly initialize loggers and add logger method to debug data using json * Minor fix
40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
import os
|
|
import re
|
|
from requests import get
|
|
|
|
from core.config import changes
|
|
from core.colors import run, que, good, info, end, green
|
|
from core.log import setup_logger
|
|
|
|
logger = setup_logger(__name__)
|
|
|
|
|
|
def updater():
|
|
logger.run('Checking for updates')
|
|
latestCommit = get(
|
|
'https://raw.githubusercontent.com/s0md3v/XSStrike/master/core/config.py').text
|
|
|
|
if changes not in latestCommit: # just a hack to see if a new version is available
|
|
changelog = re.search(r"changes = '''(.*?)'''", latestCommit)
|
|
changelog = changelog.group(1).split(
|
|
';') # splitting the changes to form a list
|
|
logger.good('A new version of XSStrike is available.')
|
|
changes_str = 'Changes:\n'
|
|
for change in changelog: # prepare changes to print
|
|
changes_str += '%s>%s %s\n' % (green, end, change)
|
|
logger.info(changes_str)
|
|
currentPath = os.getcwd().split('/') # if you know it, you know it
|
|
folder = currentPath[-1] # current directory name
|
|
path = '/'.join(currentPath) # current directory path
|
|
choice = input('%s Would you like to update? [Y/n] ' % que).lower()
|
|
|
|
if choice != 'n':
|
|
logger.run('Updating XSStrike')
|
|
os.system(
|
|
'git clone --quiet https://github.com/s0md3v/XSStrike %s' % (folder))
|
|
os.system('cp -r %s/%s/* %s && rm -r %s/%s/ 2>/dev/null' %
|
|
(path, folder, path, path, folder))
|
|
logger.good('Update successful!')
|
|
else:
|
|
logger.good('XSStrike is up to date!')
|