48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
import subprocess
|
|
|
|
import decman
|
|
from decman import File, Module
|
|
from decman.plugins.aur import packages as aur_packages
|
|
from decman.plugins.pacman import packages as pacman_packages
|
|
|
|
|
|
class ZshModule(Module):
|
|
def __init__(self, user: str):
|
|
super().__init__("zsh")
|
|
self.user = user
|
|
|
|
def files(self):
|
|
return {
|
|
f"/home/{self.user}/.zshrc": File(
|
|
source_file="./home/.zshrc",
|
|
owner=self.user,
|
|
),
|
|
}
|
|
|
|
@pacman_packages
|
|
def pacman_packages(self) -> set[str]:
|
|
return {
|
|
"fzf",
|
|
"zsh",
|
|
"zsh-autosuggestions",
|
|
"zsh-completions",
|
|
"zsh-syntax-highlighting",
|
|
}
|
|
|
|
@aur_packages
|
|
def aur_packages(self) -> set[str]:
|
|
return {
|
|
"fzf-tab-git",
|
|
}
|
|
|
|
def after_update(self, store):
|
|
result = subprocess.run(
|
|
["getent", "passwd", self.user], capture_output=True, text=True
|
|
)
|
|
if result.returncode != 0:
|
|
return
|
|
# passwd 格式: name:x:uid:gid:gecos:home:shell
|
|
shell = result.stdout.strip().split(":")[-1]
|
|
if shell != "/usr/bin/zsh":
|
|
decman.prg(["chsh", "-s", "/usr/bin/zsh", self.user])
|