home/.zshc/func.zsh

function clip {
  if [[ $(uname) == "Darwin" ]];then
    command pbcopy $@

  elif [[ "${XDG_SESSION_TYPE}" == "x11" ]]; then
    if (( $+commands[xsel] )); then
      command xsel -ib $@
    elif (( $+commands[xclip] )); then
      command xclip $@
    else
      echo "Detected X11, but xsel or xclip are not installed."
    fi

  elif [[ -n "${WAYLAND_DISPLAY}" ]]; then
    if (( $+commands[wl-copy] )); then
      command wl-copy $@
    else
      echo "Detected Wayland, but wl-copy is not installed."
    fi

  elif [[ -n "${SSH_CONNECTION}" ]]; then
    if (( $+commands[lemonade] )); then
      command lemonade copy $@
    else
      echo "Detected an SSH connection, but lemonade is not installed."
    fi

  else
    echo "Cannot determine desktop session type, are you running a display server that is not X11 or Wayland?"
  fi
}

function zipline {
  # see spectacle-screenshot and flameshot-screenshot for example usage

  readonly file=${1:?Please provide the path to the file you want to upload.}
  readonly name=${2}
  readonly desktopentry=${3}
  readonly token=${ZIPLINE_TOKEN:?Environment variable ZIPLINE_TOKEN is not set. Please set it to your Zipline API token. I recommend placing it in a file at ~/.config/environment.d/zipline.conf}
  readonly url=${ZIPLINE_URL:?Environment variable ZIPLINE_URL is not set. Please set it to the URL of your Zipline instance. I recommend placing it in a file at ~/.config/environment.d/zipline.conf}

  temp_file=$(mktemp /tmp/zipline-upload.XXXXXXXX.json)
  trap 'rm -f "$temp_file"' EXIT

  if (( ! $+commands[jq] )); then
    echo "jq is not installed."
    return 127
  fi

  if (( ! $+commands[curl] )); then
    echo "curl is not installed."
    return 127
  fi

  if [[ ! -f "${file}" ]]; then
    echo "File does not exist."
    return 66
  fi

  if [[ -n "${name}" && -n "${desktopentry}" ]]; then
    if ! does-desktop-entry-exist "$desktopentry"; then
      echo "Desktop entry does not exist."
      return 66
    fi
      if (( ! $+commands[notify-send] )); then
      echo "notify-send is not installed."
      return 127
    fi
    readonly use_send_notify=true
  else
    readonly use_send_notify=false
  fi

  if curl -sS -fH "authorization: $token" "${url%/}/api/upload" -F file="@$file" -H "Content-Type: multipart/form-data" >"$temp_file" 2>&1; then
    readonly link=$(jq -r '.files[0]' <"$temp_file" | tr -d '\n')
    echo -n "$link" | clip
    echo "Upload successful. Link copied to clipboard: $link"
    if $use_send_notify; then
      notify-send -a "$name" -u low -c transfer.complete -i "$file" -h "string:desktop-entry:$desktopentry" "Upload Successful" "Link copied to clipboard:\n$link"
    fi
  else
    echo "Upload failed. Error message: $(<"$temp_file")"
    if $use_send_notify; then
      notify-send -a "$name" -u critical -c transfer.error -h "string:desktop-entry $desktopentry" "Upload Failed" "$(<"$temp_file")"
    fi
    return 1
  fi
}

function spectacle-screenshot {
  # takes two arguments:
  # 1 - whether to save a recording or a screenshot (boolean) - defaults to screenshot (false) - RECORDING DOES NOT WORK ON X11
  # 2 - the path to save the screenshot/recording to - if not provided, a temp file will be used
  if (( ! $+commands[spectacle] )); then
    echo "spectacle is not installed."
    return 127
  fi

  if [[ -z "${1}" ]]; then
    readonly save_recording=false
  else
    readonly save_recording=$1
  fi

  if [[ -f "${2}" ]]; then
    echo "File already exists. Please provide a different file path, or use the zipline function to upload the file."
    echo "Example: zipline \"$file\""
    return 1
  fi

  if [[ -z "${2}" ]]; then
    if $save_recording; then
      readonly file=$(mktemp /tmp/spectacle-recording.XXXXXXXX.mp4)
    else
      readonly file=$(mktemp /tmp/spectacle-screenshot.XXXXXXXX.png)
    fi
  else
    readonly file=$2
  fi

  if $save_recording; then
    command spectacle --nonotify --background --record=region --copy-image --output "$file"
  else
    command spectacle --nonotify --background --region --copy-image --output "$file"
  fi
  [ ! -s "$file" ] && return 1

  trap 'rm -f "$file"' EXIT
  zipline "$file" Spectacle org.kde.spectacle.desktop
}

function flameshot-screenshot {
  # takes one argument, the path to save the screenshot to - if not provided, a temp file will be used
  if (( ! $+commands[flameshot] )); then
    echo "flameshot is not installed."
    return 127
  fi

  if [[ -f "${1}" ]]; then
    echo "File already exists. Please provide a different file path, or use the zipline function to upload the file."
    echo "Example: zipline \"${1}\""
    return 1
  fi

  if [[ -z "${1}" ]]; then
    readonly file=$(mktemp /tmp/flameshot-screenshot.XXXXXXXX.png)
  else
    readonly file=$1
  fi

  command flameshot gui --clipboard --raw > $file
  echo "Screenshot saved to $file"
  [ ! -s "$file" ] && return 1

  trap 'rm -f "$file"' EXIT
  zipline "$file" Flameshot org.flameshot.Flameshot.desktop
}

function does-desktop-entry-exist {
  readonly desktopentry=${1:?Please provide the full filename of the desktop entry.}
  local entrypaths=(
    "/usr/share/applications"
    "/usr/local/share/applications"
    "$HOME/.local/share/applications"
  )
  for entrypath in "${entrypaths[@]}"; do
    if [[ -f "$entrypath/$desktopentry" ]]; then
      return 0
    fi
  done
  echo "Desktop entry $desktopentry does not exist."
  return 1
}

function htop btop {
  if (( $+commands[btop] )); then
    command btop $@
  else
    command htop $@
  fi
}

function fastfetch ff neofetch nf {
  if (( $+commands[fastfetch] )); then
    command fastfetch $@
  elif (( $+commands[neofetch] )); then
    echo "Warning: fastfetch is not installed, falling back to neofetch."
    command neofetch $@
  else
    echo "Fatal: fastfetch and neofetch are not installed, cannot fetch system information."
  fi
}