dotfiles/win/Modules/Pass/Pass.psm1
2021-06-02 16:06:42 +08:00

134 lines
3.2 KiB
PowerShell

$PASSWORD_STORE_DIR = Get-Item "~\.password-store"
function GeneratePassword {
param(
[Int] $Size = 10,
[Char[]] $Charsets = "ULNS",
[Char[]] $Exclude
)
$Chars = @(); $TokenSet = @()
If (!$TokenSets) {
$Global:TokenSets = @{
U = [Char[]]'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
L = [Char[]]'abcdefghijklmnopqrstuvwxyz'
N = [Char[]]'0123456789'
S = [Char[]]'!"#$%&''()*+,-./:;<=>?@[\]^_`{|}~'
}
}
$CharSets | ForEach {
$Tokens = $TokenSets."$_" | ForEach {If ($Exclude -cNotContains $_) {$_}}
If ($Tokens) {
$TokensSet += $Tokens
If ($_ -cle [Char]"Z") {$Chars += $Tokens | Get-Random}
}
}
While ($Chars.Count -lt $Size) {$Chars += $TokensSet | Get-Random}
return ($Chars | Sort-Object {Get-Random}) -Join ""
}
function EnsurePath {
param(
[String] $Path
)
$dir = Split-Path $Path
if (!$dir) {
$dir = "."
}
if (!(Test-Path -PathType Container $dir)) {
New-Item -ItemType Directory -Path $dir
}
if (!$Path.EndsWith(".gpg")) {
$Path = $Path + ".gpg"
}
Join-Path $PASSWORD_STORE_DIR.FullName $Path
}
function GetUid {
if ((gpg --list-secret-keys | findstr uid) -match '<(.*?)>') {
$matches[1]
} else {
throw "unable to find default uid"
}
}
function Edit-Pass {
[Cmdletbinding()]
param(
[Parameter(Mandatory=$true)] [String] $Path
)
$Path = EnsurePath($Path)
echo $Path
$tmpfile = (New-TemporaryFile).FullName
gpg --decrypt $Path > $tmpfile
nvim $tmpfile
if ($?) {
gpg -r (GetUid) -o "$tmpfile.gpg" --encrypt $tmpfile
Move-Item -Path "$tmpfile.gpg" -Destination "$Path" -Force
Remove-Item $tmpfile -Force
}
}
function New-Pass {
[Cmdletbinding()]
param(
[Parameter(Mandatory=$true)] [String] $Path
)
$Path = EnsurePath $Path
$pass = GeneratePassword
if (Test-Path -PathType Leaf $Path) {
$text = gpg --decrypt $Path
$text[0] = $pass
} else {
$text = @($pass)
}
Remove-Item $Path -Force
$text | gpg -r (GetUid) -o $Path --encrypt -
Set-Clipboard $pass
Write-Host $pass
Write-Host "new password is saved and copied to Clipboard"
}
function Get-Pass {
[Cmdletbinding()]
param(
[Parameter(Mandatory=$true)] [String] $Path,
[Bool] $Clipboard=$true
)
$Path = EnsurePath $Path
if ($Clipboard) {
$pass = gpg --decrypt $Path | Select -First 1
if ($pass) {
Set-Clipboard $pass
Write-Host "password is copied to Clipboard successfully"
#TODO clear clipboard after centain period
} else {
throw "password is empty"
}
} else {
Write-Host $pass
}
}
function Find-Pass {
$selected = Get-ChildItem $PASSWORD_STORE_DIR -Recurse -Filter *.gpg | %{
$_.FullName.SubString(
$PASSWORD_STORE_DIR.FullName.Length+1,
$_.FullName.Length-$PASSWORD_STORE_DIR.FullName.Length-5
)
} | Invoke-Fzf
if ($selected) {
Get-Pass $selected
}
}
Export-ModuleMember -Function Edit-Pass,New-Pass,Get-Pass,Find-Pass