from __future__ import annotations from virtualenv.activation.via_template import ViaTemplateActivator class NushellActivator(ViaTemplateActivator): def templates(self): yield "activate.nu" @staticmethod def quote(string): """ Nushell supports raw strings like: r###'this is a string'###. This method finds the maximum continuous sharps in the string and then quote it with an extra sharp. """ max_sharps = 0 current_sharps = 0 for char in string: if char == "#": current_sharps += 1 max_sharps = max(current_sharps, max_sharps) else: current_sharps = 0 wrapping = "#" * (max_sharps + 1) return f"r{wrapping}'{string}'{wrapping}" def replacements(self, creator, dest_folder): # noqa: ARG002 return { "__VIRTUAL_PROMPT__": "" if self.flag_prompt is None else self.flag_prompt, "__VIRTUAL_ENV__": str(creator.dest), "__VIRTUAL_NAME__": creator.env_name, "__BIN_NAME__": str(creator.bin_dir.relative_to(creator.dest)), } __all__ = [ "NushellActivator", ]