Refactor environment variable handling in configuration files

- Introduced a new helper function `_get_int_env_var` to retrieve environment variables as integers with default fallback and error handling.
- Updated `config.py` and `server_config.py` to use the new helper function for better clarity and maintainability.
- Simplified `_get_env_var` in `config_utils.py` to focus solely on retrieving string values without type casting.

Author: Cheelax

src/gitingest/config.py

@@ -5,12 +5,20 @@
 
 from gitingest.utils.config_utils import _get_env_var
 
-MAX_FILE_SIZE = _get_env_var("MAX_FILE_SIZE", 10 * 1024 * 1024, int)  # Max file size to process in bytes (10 MB)
-MAX_FILES = _get_env_var("MAX_FILES", 10_000, int)  # Max number of files to process
-MAX_TOTAL_SIZE_BYTES = _get_env_var("MAX_TOTAL_SIZE_BYTES", 500 * 1024 * 1024, int)  # Max output file size (500 MB)
-MAX_DIRECTORY_DEPTH = _get_env_var("MAX_DIRECTORY_DEPTH", 20, int)  # Max depth of directory traversal
+def _get_int_env_var(key: str, default: int) -> int:
+    """Get environment variable as integer with fallback to default."""
+    try:
+        return int(_get_env_var(key, str(default)))
+    except ValueError:
+        print(f"Warning: Invalid value for GITINGEST_{key}. Using default: {default}")
+        return default
 
-DEFAULT_TIMEOUT = _get_env_var("DEFAULT_TIMEOUT", 60, int)  # Default timeout for git operations in seconds
+MAX_FILE_SIZE = _get_int_env_var("MAX_FILE_SIZE", 10 * 1024 * 1024)  # Max file size to process in bytes (10 MB)
+MAX_FILES = _get_int_env_var("MAX_FILES", 10_000)  # Max number of files to process
+MAX_TOTAL_SIZE_BYTES = _get_int_env_var("MAX_TOTAL_SIZE_BYTES", 500 * 1024 * 1024)  # Max output file size (500 MB)
+MAX_DIRECTORY_DEPTH = _get_int_env_var("MAX_DIRECTORY_DEPTH", 20)  # Max depth of directory traversal
+
+DEFAULT_TIMEOUT = _get_int_env_var("DEFAULT_TIMEOUT", 60)  # Default timeout for git operations in seconds
 
 OUTPUT_FILE_NAME = _get_env_var("OUTPUT_FILE_NAME", "digest.txt")
 TMP_BASE_PATH = Path(_get_env_var("TMP_BASE_PATH", tempfile.gettempdir())) / "gitingest"

src/gitingest/utils/config_utils.py

@@ -3,25 +3,22 @@
 from __future__ import annotations
 
 import os
-from typing import Callable
 
 
-def _get_env_var(key: str, default: int | str, cast_func: Callable[[str], int | str] | None = None) -> int | str:
-    """Get environment variable with ``GITINGEST_`` prefix and optional type casting.
+def _get_env_var(key: str, default: str) -> str:
+    """Get environment variable with ``GITINGEST_`` prefix.
 
     Parameters
     ----------
     key : str
         The name of the environment variable.
-    default : int | str
+    default : str
         The default value to return if the environment variable is not set.
-    cast_func : Callable[[str], int  |  str] | None
-        The function to cast the environment variable to the desired type.
 
     Returns
     -------
-    int | str
-        The value of the environment variable, cast to the desired type if provided.
+    str
+        The value of the environment variable as a string.
 
     """
     env_key = f"GITINGEST_{key}"
@@ -30,11 +27,4 @@ def _get_env_var(key: str, default: int | str, cast_func: Callable[[str], int |
     if value is None:
         return default
 
-    if cast_func:
-        try:
-            return cast_func(value)
-        except (ValueError, TypeError):
-            print(f"Warning: Invalid value for {env_key}: {value}. Using default: {default}")
-            return default
-
     return value

src/server/server_config.py

@@ -6,12 +6,20 @@
 
 from gitingest.utils.config_utils import _get_env_var
 
-MAX_DISPLAY_SIZE: int = _get_env_var("MAX_DISPLAY_SIZE", 300_000, int)
-DELETE_REPO_AFTER: int = _get_env_var("DELETE_REPO_AFTER", 60 * 60, int)  # In seconds (1 hour)
+def _get_int_env_var(key: str, default: int) -> int:
+    """Get environment variable as integer with fallback to default."""
+    try:
+        return int(_get_env_var(key, str(default)))
+    except ValueError:
+        print(f"Warning: Invalid value for GITINGEST_{key}. Using default: {default}")
+        return default
+
+MAX_DISPLAY_SIZE: int = _get_int_env_var("MAX_DISPLAY_SIZE", 300_000)
+DELETE_REPO_AFTER: int = _get_int_env_var("DELETE_REPO_AFTER", 60 * 60)  # In seconds (1 hour)
 
 # Slider configuration (if updated, update the logSliderToSize function in src/static/js/utils.js)
-MAX_FILE_SIZE_KB: int = _get_env_var("MAX_FILE_SIZE_KB", 100 * 1024, int)  # 100 MB
-MAX_SLIDER_POSITION: int = _get_env_var("MAX_SLIDER_POSITION", 500, int)  # Maximum slider position
+MAX_FILE_SIZE_KB: int = _get_int_env_var("MAX_FILE_SIZE_KB", 100 * 1024)  # 100 MB
+MAX_SLIDER_POSITION: int = _get_int_env_var("MAX_SLIDER_POSITION", 500)  # Maximum slider position
 
 EXAMPLE_REPOS: list[dict[str, str]] = [
     {"name": "Gitingest", "url": "https://github.com/cyclotruc/gitingest"},