mirror of
https://github.com/trustedsec/hate_crack.git
synced 2026-07-28 22:51:14 -07:00
Justin BollingerandClaude Opus 4.6
fa07d37a84
The bash shim uses `uv run --directory <install_dir>` which changes the process CWD to the install directory. _ensure_hashfile_in_cwd() and the Hashview download path used os.getcwd() to determine the target directory for output files (.out, .nt, etc.), causing them to land in the install directory instead of where the user ran the command. Add orig_cwd() helper that reads HATE_CRACK_ORIG_CWD (set by the shim) and use it in _ensure_hashfile_in_cwd(), the Hashview download path, and the potfile fallback path. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
import logging
|
|
import os
|
|
import sys
|
|
from typing import Optional
|
|
|
|
|
|
def orig_cwd() -> str:
|
|
"""Return the caller's original working directory.
|
|
|
|
When invoked via the bash shim (``uv run --directory``), the process CWD
|
|
is the install directory, not where the user ran the command.
|
|
``HATE_CRACK_ORIG_CWD`` preserves the real CWD so that output files
|
|
(e.g. ``.out``, ``.nt``) are created next to the hashfile the user specified.
|
|
"""
|
|
return os.environ.get("HATE_CRACK_ORIG_CWD") or os.getcwd()
|
|
|
|
|
|
def resolve_path(value: Optional[str]) -> Optional[str]:
|
|
"""Expand user and return an absolute path, or None.
|
|
|
|
When invoked via the bash shim (``uv run --directory``), the working
|
|
directory is changed to the repo root before Python starts.
|
|
``HATE_CRACK_ORIG_CWD`` preserves the caller's real working directory
|
|
so that relative paths on the command line resolve correctly.
|
|
"""
|
|
if not value:
|
|
return None
|
|
value = os.path.expanduser(value)
|
|
if os.path.isabs(value):
|
|
return value
|
|
orig_cwd = os.environ.get("HATE_CRACK_ORIG_CWD")
|
|
if orig_cwd:
|
|
return os.path.normpath(os.path.join(orig_cwd, value))
|
|
return os.path.abspath(value)
|
|
|
|
|
|
def add_common_args(parser) -> None:
|
|
pass # All config items are now set via config file only
|
|
|
|
|
|
def setup_logging(logger: logging.Logger, hate_path: str, debug_mode: bool) -> None:
|
|
if not debug_mode:
|
|
return
|
|
logger.setLevel(logging.DEBUG)
|
|
# Debug logs go to stderr by default (no log file side effects).
|
|
has_stream = any(
|
|
isinstance(h, logging.StreamHandler) and not isinstance(h, logging.FileHandler)
|
|
for h in logger.handlers
|
|
)
|
|
if not has_stream:
|
|
stream_handler = logging.StreamHandler(sys.stderr)
|
|
stream_handler.setFormatter(
|
|
logging.Formatter("%(asctime)s %(levelname)s %(message)s")
|
|
)
|
|
logger.addHandler(stream_handler)
|
|
# Show HTTP requests made by the requests/urllib3 library.
|
|
debug_handler = logging.StreamHandler(sys.stderr)
|
|
debug_handler.setFormatter(logging.Formatter("%(asctime)s %(levelname)s %(name)s %(message)s"))
|
|
urllib3_logger = logging.getLogger("urllib3")
|
|
urllib3_logger.setLevel(logging.DEBUG)
|
|
urllib3_logger.addHandler(debug_handler)
|