dotfiles/function-kill.zsh at main · taking/dotfiles · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

kill-process() {
  local selected pids source_cmd

  if command -v procs >/dev/null 2>&1; then
    source_cmd='procs --no-header'
  else
    source_cmd='ps -ef | sed 1d'
  fi

  selected=$(
    eval "$source_cmd" \
      | fzf \
          --multi \
          --height 75% \
          --reverse \
          --prompt='kill> ' \
          --header='TAB: multi-select / ENTER: choose / ESC: cancel' \
          --preview '
            line="{}"
            pid=$(printf "%s\n" "$line" | awk "{for(i=1;i<=NF;i++) if(\$i ~ /^[0-9]+$/) {print \$i; exit}}")
            if [ -n "$pid" ]; then
              echo "PID: $pid"
              echo
              ps -p "$pid" -o pid=,ppid=,%cpu=,%mem=,etime=,user=,command=
              echo
              lsof -Pan -p "$pid" -i 2>/dev/null | head -30
            else
              echo "$line"
            fi
          ' \
          --preview-window=right:65%:wrap
  ) || return

  [[ -z "$selected" ]] && {
    echo "Cancelled."
    return 1
  }

  pids=$(
    awk '
      {
        for (i=1; i<=NF; i++) {
          if ($i ~ /^[0-9]+$/) {
            print $i
            break
          }
        }
      }
    ' <<< "$selected" | sort -u
  )

  [[ -z "$pids" ]] && {
    echo "No PID found."
    return 1
  }

  echo "Selected PID(s): $(echo "$pids" | tr "\n" " ")"
  read "confirm?Kill selected PID(s)? (y/N): "

  if [[ "$confirm" =~ ^[Yy]$ ]]; then
    echo "$pids" | xargs kill 2>/dev/null || echo "$pids" | xargs kill -9
    echo "Killed."
  else
    echo "Cancelled."
  fi
}

kill-port() {
  local selected pids

  selected=$(
    lsof -nP -iTCP -sTCP:LISTEN 2>/dev/null \
      | sed 1d \
      | fzf \
          --multi \
          --height 75% \
          --reverse \
          --prompt='port> ' \
          --header='TAB: multi-select / ENTER: choose / ESC: cancel' \
          --preview '
            line="{}"
            pid=$(printf "%s\n" "$line" | awk "{print \$2}")
            if [ -n "$pid" ]; then
              echo "PID: $pid"
              echo
              ps -p "$pid" -o pid=,ppid=,%cpu=,%mem=,etime=,user=,command=
              echo
              lsof -Pan -p "$pid" -i 2>/dev/null | head -40
            else
              echo "$line"
            fi
          ' \
          --preview-window=right:65%:wrap
  ) || return

  [[ -z "$selected" ]] && {
    echo "Cancelled."
    return 1
  }

  pids=$(awk '{print $2}' <<< "$selected" | sort -u)

  [[ -z "$pids" ]] && {
    echo "No PID found."
    return 1
  }

  echo "Selected PID(s): $(echo "$pids" | tr '\n' ' ')"
  read "confirm?Kill selected port process(es)? (y/N): "

  if [[ "$confirm" =~ ^[Yy]$ ]]; then
    echo "$pids" | xargs kill 2>/dev/null || echo "$pids" | xargs kill -9
    echo "Killed."
  else
    echo "Cancelled."
  fi
}