Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add --local option to kup #2996

Merged
merged 1 commit into from
Oct 14, 2022
Merged

Add --local option to kup #2996

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
39 changes: 22 additions & 17 deletions kup/src/kup/__main__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
INSTALLED = "🟢 \033[92minstalled\033[0m"
AVAILABLE = "🔵 \033[94mavailable\033[0m"
UPDATE = "🟠 \033[93mnewer version available\033[0m"
LOCAL = "\033[3mnot managed by kup\033[0m"
LOCAL = "\033[3mlocal checkout\033[0m"

NIX_SUBSTITUTERS = [
'--option',
Expand Down Expand Up @@ -203,39 +203,42 @@ def list_package(package_name: str) -> None:
print(table.table)


def update_or_install_package(package: Union[AvailablePackage, ConcretePackage], version: Optional[str]) -> None:
def update_or_install_package(
package: Union[AvailablePackage, ConcretePackage], version: Optional[str], local_path: Optional[str]
) -> None:
version = '/' + version if version else ''
path = f'github:runtimeverification/{package.repo}{version}' if not local_path else local_path
if type(package) is ConcretePackage:
if package.immutable or version:
if package.immutable or version or local_path:
nix(['profile', 'remove', str(package.index)])
nix(['profile', 'install', f'github:runtimeverification/{package.repo}{version}#{package.package}'])
nix(['profile', 'install', f'{path}#{package.package}'])
else:
nix(['profile', 'upgrade', str(package.index)])
else:
nix(['profile', 'install', f'github:runtimeverification/{package.repo}{version}#{package.package}'])
nix(['profile', 'install', f'{path}#{package.package}'])


def install_package(package_name: str, package_version: Optional[str]) -> None:
def install_package(package_name: str, package_version: Optional[str], local_path: Optional[str]) -> None:
reload_packages()
if package_name not in available_packages.keys():
print(
f'❗ \033[91mThe package \'\033[94m{package_name}\033[91m\' does not exist.\033[0m\nUse \'\033[92mkup list\033[0m\' to see all the available packages.'
)
return
if package_name in installed_packages and not package_version:
if package_name in installed_packages and not (package_version or local_path):
print(
f'❗ The package \'\033[94m{package_name}\033[0m\' is already installed.\nUse \'\033[92mkup update {package_name}\033[0m\' to update to the latest version.'
)
return
if package_name in installed_packages:
package = packages[package_name]
update_or_install_package(package, package_version)
update_or_install_package(package, package_version, local_path)
else:
new_package = available_packages[package_name]
update_or_install_package(new_package, package_version)
update_or_install_package(new_package, package_version, local_path)


def update_package(package_name: str, package_version: Optional[str]) -> None:
def update_package(package_name: str, package_version: Optional[str], local_path: Optional[str]) -> None:
reload_packages()
if package_name not in available_packages.keys():
print(
Expand All @@ -248,11 +251,11 @@ def update_package(package_name: str, package_version: Optional[str]) -> None:
)
return
package = packages[package_name]
if package.status == INSTALLED and not package_version:
if package.status == INSTALLED and not (package_version or local_path):
print(f'The package \'\033[94m{package_name}\033[0m\' is up to date.')
return

update_or_install_package(package, package_version)
update_or_install_package(package, package_version, local_path)


def remove_package(package_name: str) -> None:
Expand Down Expand Up @@ -297,26 +300,29 @@ def main() -> None:
install = subparser.add_parser('install', help='Download and install the stated package')
install.add_argument('package', type=str)
install.add_argument('--version', type=str)
install.add_argument('--local', type=str)

uninstall = subparser.add_parser('remove', help='Remove the given package from the user\'s PATH')
uninstall.add_argument('package', type=str)

update = subparser.add_parser('update', help='Update the package to the latest version')
update.add_argument('package', type=str)
update.add_argument('--version', type=str)
update.add_argument('--local', type=str)

shell = subparser.add_parser('shell', help='Add the selected package to the current shell (temporary)')
shell.add_argument('package', type=str)
shell.add_argument('--version', type=str)
shell.add_argument('--local', type=str)

args = parser.parse_args()

if args.command == "list":
list_package(args.package)
elif args.command == "install":
install_package(args.package, args.version)
install_package(args.package, args.version, args.local)
elif args.command == "update":
update_package(args.package, args.version)
update_package(args.package, args.version, args.local)
elif args.command == "remove":
remove_package(args.package)
elif args.command == "shell":
Expand All @@ -328,9 +334,8 @@ def main() -> None:
return
temporary_package = available_packages[args.package]
version = '/' + args.version if args.version else ''
nix_detach(
['shell', f'github:runtimeverification/{temporary_package.repo}{version}#{temporary_package.package}']
)
path = f'github:runtimeverification/{temporary_package.repo}{version}' if not args.local else args.local
nix_detach(['shell', f'{path}#{temporary_package.package}'])


if __name__ == '__main__':
Expand Down