maryam

#!/usr/bin/env python3
"""
OWASP Maryam!

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""

__VERSION__ = '2.0.0'

import sys
import platform
import argparse

# Check the OS
if platform.system() not in ('Linux', 'Darwin', 'OSX', "FreeBSD"):
	print('[!] This tool is designed for Linux, Darwin, OSX, and FreeBSD. Maybe not work!')

# Check the version
if sys.version_info.major != 3:
	print('[!] Maryam migrated to python3.')
	exit(0)

# Import framework
from core import initial

def cui(args, format_help):
	x = initial.initialize
	if args:
		option = args[0]
		if option in ['-v', '--version']:
			print(f"OWASP MARYAM V.{__VERSION__}")
			exit(0)
		command = ' '.join(args[1:])
		if not v or option not in ['-e', '--execute', '-s', '--start']:
			print(format_help)
			exit(0)
		x = x('execute')
		if option in ['-e', '--execute']:
			x.onecmd(command)
			exit(0)
		elif args[0] in ['-s', '--start']:
			x.onecmd(command)
	else:
		x = x('run')
	while True:
		try:
			x.cmdloop()
			break
		except KeyboardInterrupt:
			print('\n[!] Use exit command to exit ')
		except Exception as e:
			raise e


desc = 'maryam -h'
parse = argparse.ArgumentParser(description=desc)
parse.add_argument('-e', help='execute a command and exit', metavar='execute', dest='execute', action='store')
parse.add_argument('-s', help='run a command without exit', metavar='start', dest='start', action='store')
parse.add_argument('-v', help='show version and exit', action='version', version=f"OWASP MARYAM V.{__VERSION__}")
format_help = parse.format_help()
argv = sys.argv[1:]
for k, v in enumerate(argv):
	if ' ' in v:
		argv[k] = f'"{v}"'

cui(argv, format_help)