mud-git 1.0.0__py3-none-any.whl

mud/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.0.0"

mud/__init__.py

@@ -0,0 +1,19 @@
+#!/usr/bin/env python3
+
+import sys
+
+from mud import utils, settings
+from mud.app import App
+
+def run():
+	try:
+		utils.settings = settings.Settings(utils.SETTINGS_FILE_NAME)
+
+		app = App()
+		app.run()
+	except KeyboardInterrupt:
+		utils.print_error('Stopped by user.', 0)
+
+
+if __name__ == '__main__':
+	run()

mud/app.py

@@ -0,0 +1,275 @@
+import os
+import sys
+import asyncio
+import argparse
+import subprocess
+
+from argparse import ArgumentParser
+
+from mud import config
+
+from mud import utils
+from mud.utils import glyphs
+
+from mud.styles import *
+from mud.commands import *
+from mud.runner import Runner
+
+
+class App:
+	def __init__(self):
+		self.cmd_runner = None
+		self.command = None
+		self.config = None
+		self.parser = self._create_parser()
+
+	@staticmethod
+	def _create_parser() -> ArgumentParser:
+		parser = argparse.ArgumentParser(description=f'{BOLD}mud{RESET} allows you to run commands in multiple repositories.')
+		subparsers = parser.add_subparsers(dest='command')
+
+		subparsers.add_parser(LOG[0], aliases=LOG[1:], help='Displays log of latest commit messages for all repositories in a table view.')
+		subparsers.add_parser(INFO[0], aliases=INFO[1:], help='Displays branch divergence and working directory changes')
+		subparsers.add_parser(INIT[0], aliases=INIT[1:], help=f'Initializes the {BOLD}.mudconfig{RESET} and adds all repositories in this directory to {BOLD}.mudconfig{RESET}.')
+		subparsers.add_parser(TAGS[0], aliases=TAGS[1:], help='Displays git tags in repositories.')
+		subparsers.add_parser(LABELS[0], aliases=LABELS[1:], help='Displays mud labels across repositories.')
+		subparsers.add_parser(STATUS[0], aliases=STATUS[1:], help='Displays working directory changes.')
+		subparsers.add_parser(BRANCHES[0], aliases=BRANCHES[1:], help='Displays all branches in repositories.')
+		subparsers.add_parser(REMOTE_BRANCHES[0], aliases=REMOTE_BRANCHES[1:], help='Displays all remote branches in repositories.')
+		subparsers.add_parser(CONFIGURE[0], aliases=CONFIGURE[1:], help='Runs the interactive configuration wizard.')
+
+		add_parser = subparsers.add_parser(ADD[0], aliases=ADD[1:], help='Adds repository or labels an existing repository.')
+		add_parser.add_argument('label', help='The label to add (optional).', nargs='?', default='', type=str)
+		add_parser.add_argument('path', help='Repository to add (optional).', nargs='?', type=str)
+
+		remove_parser = subparsers.add_parser(REMOVE[0], aliases=REMOVE[1:], help='Removes repository or removes the label from an existing repository.')
+		remove_parser.add_argument('label', help='Label to remove from repository (optional).', nargs='?', default='', type=str)
+		remove_parser.add_argument('path', help='Repository to remove (optional).', nargs='?', type=str)
+
+		parser.add_argument(*COMMAND_ATTR, metavar='COMMAND', nargs='?', default='', type=str, help=f'Explicit command argument. Use this when you want to run a command that has a special characters.')
+		parser.add_argument(*TABLE_ATTR, metavar='TABLE', nargs='?', default='', type=str, help=f'Switches table view, runs in table view it is disabled in {BOLD}.mudsettings{RESET}.')
+		parser.add_argument(*LABEL_PREFIX, metavar='LABEL', nargs='?', default='', type=str, help='Includes repositories with provided label.')
+		parser.add_argument(*NOT_LABEL_PREFIX, metavar='NOT_LABEL', nargs='?', default='', type=str, help=f'Excludes repositories with provided label..')
+		parser.add_argument(*BRANCH_PREFIX, metavar='BRANCH', nargs='?', default='', type=str, help='Includes repositories on a provided branch.')
+		parser.add_argument(*NOT_BRANCH_PREFIX, metavar='NOT_BRANCH', nargs='?', default='', type=str, help='Excludes repositories on a provided branch.')
+		parser.add_argument(*MODIFIED_ATTR, action='store_true', help='Filters modified repositories.')
+		parser.add_argument(*DIVERGED_ATTR, action='store_true', help='Filters repositories with diverged branches.')
+		parser.add_argument(*ASYNC_ATTR, action='store_true', help='Switches asynchronous run feature.')
+		parser.add_argument(SET_GLOBAL[0], help=f'Sets {BOLD}.mudconfig{RESET} in the current repository as your fallback {BOLD}.mudconfig{RESET}.', action='store_true')
+		parser.add_argument(VERSION[0], help='Displays the current version of mud.', action='store_true')
+		parser.add_argument('catch_all', nargs='*', help='Type any commands to execute among repositories.')
+		return parser
+
+	def run(self) -> None:
+		# Displays default help message
+		if len(sys.argv) == 1 or sys.argv[1] in HELP:
+			utils.version()
+			self.parser.print_help()
+			return
+		# Sets global repository in .mudsettings
+		if sys.argv[1] in SET_GLOBAL:
+			config_path = os.path.join(os.getcwd(), utils.CONFIG_FILE_NAME)
+			if os.path.exists(config_path):
+				utils.settings.config.set('mud', 'config_path', config_path)
+				utils.settings.save()
+				print(f'Current {BOLD}.mudconfig{RESET} set as a global configuration.')
+			return
+		# Prints version
+		elif sys.argv[1] in VERSION:
+			utils.version()
+			return
+		# Runs configuration wizard
+		elif sys.argv[1] in CONFIGURE:
+			utils.configure()
+			return
+
+		current_directory = os.getcwd()
+		self.config = config.Config()
+
+		# Discovers repositories in current directory
+		if sys.argv[1] in INIT:
+			self.init(self.parser.parse_args())
+			return
+
+		self.config.find()
+		self._filter_with_arguments()
+
+		self.cmd_runner = Runner(self.config)
+		# Handling commands
+		if len(sys.argv) > 1 and sys.argv[1] in [cmd for group in COMMANDS for cmd in group]:
+			args = self.parser.parse_args()
+			if args.command in INIT:
+				os.chdir(current_directory)
+				self.init(args)
+			elif args.command in ADD:
+				self.add(args)
+			elif args.command in REMOVE:
+				self.remove(args)
+			else:
+				if len(self.repos) == 0:
+					utils.print_error('No repositories are matching this filter.', 1)
+					return
+				if args.command in INFO:
+					self.cmd_runner.info(self.repos)
+				elif args.command in LOG:
+					self.cmd_runner.log(self.repos)
+				elif args.command in REMOTE_BRANCHES:
+					self.cmd_runner.remote_branches(self.repos)
+				elif args.command in BRANCHES:
+					self.cmd_runner.branches(self.repos)
+				elif args.command in LABELS:
+					self.cmd_runner.labels(self.repos)
+				elif args.command in TAGS:
+					self.cmd_runner.tags(self.repos)
+				elif args.command in STATUS:
+					self.cmd_runner.status(self.repos)
+		# Handling subcommands
+		else:
+			del sys.argv[0]
+			if self.command is None:
+				if len(sys.argv) == 0:
+					self.parser.print_help()
+					return
+				self.command = ' '.join(sys.argv)
+				self._parse_aliases()
+			try:
+				if self.run_async:
+					if self.table:
+						asyncio.run(self.cmd_runner.run_async_table_view(self.repos.keys(), self.command))
+					else:
+						asyncio.run(self.cmd_runner.run_async(self.repos.keys(), self.command))
+				else:
+					self.cmd_runner.run_ordered(self.repos.keys(), self.command)
+			except Exception as exception:
+				utils.print_error(f'Invalid command. {exception}', 2)
+
+	def init(self, args) -> None:
+		table = utils.get_table(['Path', 'Status'])
+		self.config.data = {}
+		index = 0
+		directories = [d for d in os.listdir('.') if os.path.isdir(d) and os.path.isdir(os.path.join(d, '.git'))]
+		for directory in directories:
+			if directory in self.config.paths():
+				continue
+			self.config.add_label(directory, getattr(args, 'label', ''))
+			index += 1
+			table.add_row([f'{DIM}{directory}{RESET}', f'{GREEN}{glyphs("added")}{RESET}'])
+		if index == 0:
+			utils.print_error('No git repositories were found in this directory.', 3)
+			return
+		self.config.save(utils.CONFIG_FILE_NAME)
+		utils.print_table(table)
+
+	def add(self, args) -> None:
+		self.config.add_label(args.path, args.label)
+		self.config.save(utils.CONFIG_FILE_NAME)
+
+	def remove(self, args) -> None:
+		if args.path:
+			self.config.remove_label(args.path, args.label)
+		elif args.label:
+			self.config.remove_path(args.label)
+		else:
+			utils.print_error(f'Invalid input. Please provide a value to remove.', 4)
+		self.config.save(utils.CONFIG_FILE_NAME)
+
+	# Filter out repositories if user provided filters
+	def _filter_with_arguments(self) -> None:
+		self.repos = self.config.data
+		self.table = utils.settings.config['mud'].getboolean('run_table', fallback=True)
+		self.run_async = utils.settings.config['mud'].getboolean('run_async', fallback=True)
+
+		for path, labels in self.config.filter_label('ignore', self.config.data).items():
+			del self.repos[path]
+		include_labels = []
+		exclude_labels = []
+		include_branches = []
+		exclude_branches = []
+		modified = False
+		diverged = False
+		index = 1
+		while index < len(sys.argv):
+			arg = sys.argv[index]
+			if not arg.startswith('-'):
+				break
+			if any(arg.startswith(prefix) for prefix in LABEL_PREFIX):
+				include_labels.append(arg.split('=', 1)[1])
+			elif any(arg.startswith(prefix) for prefix in NOT_LABEL_PREFIX):
+				exclude_labels.append(arg.split('=', 1)[1])
+			elif any(arg.startswith(prefix) for prefix in BRANCH_PREFIX):
+				include_branches.append(arg.split('=', 1)[1])
+			elif any(arg.startswith(prefix) for prefix in NOT_BRANCH_PREFIX):
+				exclude_branches.append(arg.split('=', 1)[1])
+			elif arg in MODIFIED_ATTR:
+				modified = True
+			elif arg in DIVERGED_ATTR:
+				diverged = True
+			elif arg in TABLE_ATTR:
+				self.table = not self.table
+			elif arg in ASYNC_ATTR:
+				self.run_async = not self.run_async
+			elif any(arg.startswith(prefix) for prefix in COMMAND_ATTR):
+				self.command = arg.split('=', 1)[1]
+			else:
+				index += 1
+				continue
+			del sys.argv[index]
+		directory = os.getcwd()
+		to_delete = []
+
+		for repo, labels in self.repos.items():
+			abs_path = os.path.join(directory, repo)
+
+			if not os.path.isdir(abs_path):
+				utils.print_error(f'Invalid path {BOLD}{repo}{RESET}.', 12, False)
+				to_delete.append(repo)
+				continue
+			elif not os.path.isdir(os.path.join(abs_path, '.git')):
+				utils.print_error(f'{BOLD}.git{RESET} directory not found at target "{repo}".', 13, False)
+				to_delete.append(repo)
+				continue
+
+			os.chdir(abs_path)
+			delete = False
+
+			if any(include_labels) and not any(item in include_labels for item in labels):
+				delete = True
+			if any(exclude_labels) and any(item in exclude_labels for item in labels):
+				delete = True
+
+			if not delete:
+				try:
+					branch = subprocess.check_output('git rev-parse --abbrev-ref HEAD', shell=True, text=True).splitlines()[0]
+				except subprocess.CalledProcessError:
+					branch = 'NA'
+				if any(include_branches) and branch not in include_branches:
+					delete = True
+				if any(exclude_branches) and branch in exclude_branches:
+					delete = True
+
+			if not delete and modified:
+				status_output = subprocess.check_output('git status --porcelain', shell=True, stderr=subprocess.DEVNULL)
+				if not status_output:
+					delete = True
+
+			if not delete and diverged:
+				branch_status = subprocess.check_output('git status --branch --porcelain', shell=True, text=True).splitlines()
+				if not any('ahead' in line or 'behind' in line for line in branch_status if line.startswith('##')):
+					delete = True
+
+			if delete:
+				to_delete.append(repo)
+
+		for repo in to_delete:
+			del self.repos[repo]
+
+		os.chdir(directory)
+
+	def _parse_aliases(self) -> None:
+		if utils.settings.alias_settings is None:
+			return
+		for alias, command in dict(utils.settings.alias_settings).items():
+			args = self.command.split(' ')
+			if args[0] == alias:
+				del args[0]
+				self.command = ' '.join(command.split(' ') + args)

mud/commands.py

@@ -0,0 +1,28 @@
+# Commands
+ADD = ['add', 'a']
+REMOVE = ['remove', 'rm']
+LOG = ['log', 'l']
+INFO = ['info', 'i']
+INIT = ['init']
+TAGS = ['tags', 'tag', 't']
+LABELS = ['labels', 'lb']
+STATUS = ['status', 'st']
+BRANCHES = ['branch', 'branches', 'br']
+REMOTE_BRANCHES = ['remote-branch', 'remote-branches', 'rbr']
+HELP = ['help', '--help', '-h']
+VERSION = ['--version', '-v', 'version']
+CONFIGURE = ['configure', 'config']
+SET_GLOBAL = ['--set-global']
+
+COMMANDS = [ADD, REMOVE, LOG, INFO, INIT, TAGS, LABELS, STATUS, BRANCHES, REMOTE_BRANCHES, HELP, VERSION, CONFIGURE, SET_GLOBAL]
+
+# Filters
+ASYNC_ATTR = '-a', '--async'
+TABLE_ATTR = '-t', '--table'
+COMMAND_ATTR = '-c', '--command'
+MODIFIED_ATTR = '-m', '--modified'
+DIVERGED_ATTR = '-d', '--diverged'
+LABEL_PREFIX = '-l=', '--label='
+NOT_LABEL_PREFIX = '-nl=', '--not-label='
+BRANCH_PREFIX = '-b=', '--branch='
+NOT_BRANCH_PREFIX = '-nb=', '--not-branch='

mud/config.py

@@ -0,0 +1,99 @@
+import os
+import re
+import csv
+
+from typing import List, Dict
+
+from mud import utils
+from mud.styles import *
+
+
+class Config:
+	def __init__(self):
+		self.data = {}
+
+	def save(self, file_path: str) -> None:
+		def _filter_labels(label: str):
+			return bool(re.match(r'^\w+$', label))
+
+		with open(file_path, 'w', newline='') as tsvfile:
+			writer = csv.writer(tsvfile, delimiter='\t')
+
+			for path, labels in self.data.items():
+				valid_labels = [label for label in labels if _filter_labels(label)]
+				formatted_labels = ','.join(valid_labels) if valid_labels else ''
+				writer.writerow([path, formatted_labels])
+
+	def find(self) -> None:
+		if os.path.exists(utils.CONFIG_FILE_NAME):
+			self.load(utils.CONFIG_FILE_NAME)
+			return
+
+		directory = os.getcwd()
+		current_path = directory
+		while os.path.dirname(current_path) != current_path:
+			os.chdir(current_path)
+			if os.path.exists(utils.CONFIG_FILE_NAME):
+				self.load(utils.CONFIG_FILE_NAME)
+				return utils.CONFIG_FILE_NAME
+			current_path = os.path.dirname(current_path)
+
+		if utils.settings.mud_settings['config_path'] != '' and os.path.exists(
+				utils.settings.mud_settings['config_path']):
+			directory = os.path.dirname(utils.settings.mud_settings['config_path'])
+			os.chdir(directory)
+			os.environ['PWD'] = directory
+			self.load(utils.CONFIG_FILE_NAME)
+			return
+
+		utils.print_error(f'{BOLD}.mudconfig{RESET} file was not found. Type `mud init` to create configuration file.', 11)
+		return
+
+	def load(self, file_path: str) -> None:
+		self.data = {}
+		with open(file_path, 'r') as tsvfile:
+			reader = csv.reader(tsvfile, delimiter='\t')
+			for row in reader:
+				path = row[0]
+
+				if path.startswith('~'):
+					path = os.path.expanduser(path)
+
+				labels = [label.strip() for label in row[1].split(',') if len(row) > 1 and label.strip()] if len(row) > 1 else []
+				self.data[path] = labels
+
+	def paths(self) -> List[str]:
+		return list(self.data.keys())
+
+	def filter_label(self, label: str, repos: Dict[str, List[str]] = None) -> Dict[str, List[str]]:
+		if repos is None:
+			repos = self.data
+		if label == '':
+			return repos
+		result = {}
+		for path, labels in repos.items():
+			if label in labels:
+				result[path] = labels
+		return result
+
+	def add_label(self, path: str, label: str) -> None:
+		if path is None:
+			path = label
+			label = None
+		if not os.path.isdir(path):
+			utils.print_error(f'Invalid path {BOLD}{path}{RESET}. Remember that path should be relative.', 14)
+			return
+		if path not in self.data:
+			self.data[path] = []
+		if label is not None and label not in self.data[path]:
+			self.data[path].append(label)
+
+	def remove_path(self, path: str) -> None:
+		if path in self.data:
+			del self.data[path]
+
+	def remove_label(self, path: str, label: str) -> None:
+		if path in self.data and label in self.data[path]:
+			self.data[path].remove(label)
+			if not self.data[path]:
+				del self.data[path]