{ config, lib, ... }:
let
  cfg = config.programs.zsh;
in
{
  options.programs.zsh =
    with lib;
    with types;
    {
      p10k.extraRightPromptElements = mkOption {
        type = listOf str;
        default = [ ];
        description = "Extra elements to display on the right side of the prompt.";
      };
    };

  config.programs.zsh =
    with lib;
    with cfg;
    {
      initExtra = ''
        export P10K_EXTRA_RIGHT_PROMPT_ELEMENTS=(${strings.concatStringsSep " " p10k.extraRightPromptElements})
      '';
    };
}