forked from glenux/mfm
Initial import
This commit is contained in:
commit
97212b20ad
5 changed files with 229 additions and 0 deletions
2
.gitignore
vendored
Normal file
2
.gitignore
vendored
Normal file
|
@ -0,0 +1,2 @@
|
||||||
|
bin
|
||||||
|
lib
|
5
Makefile
Normal file
5
Makefile
Normal file
|
@ -0,0 +1,5 @@
|
||||||
|
|
||||||
|
all: build
|
||||||
|
|
||||||
|
build:
|
||||||
|
shards build
|
106
_attic/gx-vault.bash
Executable file
106
_attic/gx-vault.bash
Executable file
|
@ -0,0 +1,106 @@
|
||||||
|
#!/bin/sh
|
||||||
|
# vim: set ft=sh :
|
||||||
|
|
||||||
|
set -e
|
||||||
|
set -u
|
||||||
|
|
||||||
|
CONFIG_FILE="$HOME/.config/gx-vault.yml"
|
||||||
|
MNTDIR="$HOME/mnt"
|
||||||
|
|
||||||
|
gxvault_gocryptfs_is_mounted() {
|
||||||
|
src="$1"
|
||||||
|
|
||||||
|
if LANG=C mount |grep -q "^$src on " ; then
|
||||||
|
# >&2 echo "W: $src is already mounted"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
|
||||||
|
gxvault_gocryptfs_mount() {
|
||||||
|
name="$1"
|
||||||
|
src="$2"
|
||||||
|
dst="$3"
|
||||||
|
mkdir -p "$dst"
|
||||||
|
|
||||||
|
if [ ! -e "$src" ]; then
|
||||||
|
printf '\033[31m-> Missing input directory. Skipping.\033[0m\n'
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
if LANG=C mount |grep -q "$src on $dst" ; then
|
||||||
|
printf '\033[33m-> Already mounted. Skipping.\033[0m\n'
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
gocryptfs -idle 15m "$src" "$dst"
|
||||||
|
printf '\033[32mVault %s is now available on %s\033[0m\n' "$name" "$(echo "$dst" |sed -e "s|^$HOME|~|")"
|
||||||
|
}
|
||||||
|
|
||||||
|
# Ensure that yq is installed with the right version
|
||||||
|
gxvault_ensure_dependency_yq() {
|
||||||
|
if ! hash yq >/dev/null 2>&1 ; then
|
||||||
|
>&2 echo "ERROR: unable to find yq (yaml util)"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
YQ_VERSION="$(yq --version |sed 's/.*version //' |cut -d '.' -f1 |sed -e 's/^v//' )"
|
||||||
|
if [ "$YQ_VERSION" -lt 4 ]; then
|
||||||
|
>&2 echo "ERROR: installed version of yq is too old (found $YQ_VERSION instead of 4+)"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# Ensure that all dependencies are installed
|
||||||
|
gxvault_ensure_dependency_yq
|
||||||
|
|
||||||
|
|
||||||
|
# Get list of sources
|
||||||
|
SRC_LIST="$(mktemp)"
|
||||||
|
SRC_COUNT="$(yq eval '.vaults[].name' "$CONFIG_FILE" |wc -l )"
|
||||||
|
|
||||||
|
{
|
||||||
|
for INDEX1 in $(seq 1 "$SRC_COUNT") ; do
|
||||||
|
INDEX0=$((INDEX1 - 1))
|
||||||
|
CUR_NAME="$(yq eval ".vaults[$INDEX0].name" "$CONFIG_FILE")"
|
||||||
|
CUR_DIR="$(yq eval ".vaults[$INDEX0].encrypted_path" "$CONFIG_FILE")"
|
||||||
|
if gxvault_gocryptfs_is_mounted "$CUR_DIR" ; then
|
||||||
|
printf "%s\v [\e[32mopen\e[0m]\n" "$CUR_NAME"
|
||||||
|
else
|
||||||
|
echo "$CUR_NAME"
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
} | sort > "$SRC_LIST"
|
||||||
|
|
||||||
|
# cat "$SRC_LIST"
|
||||||
|
SRC_NAME="$(fzf --ansi < "$SRC_LIST")"
|
||||||
|
# echo "fzf: $SRC_NAME"
|
||||||
|
SRC_NAME="$(echo "$SRC_NAME" |sed -e "s/\v.*//")"
|
||||||
|
# echo "sed: $SRC_NAME"
|
||||||
|
rm -f "$SRC_LIST"
|
||||||
|
|
||||||
|
if [ -z "$SRC_NAME" ]; then
|
||||||
|
echo "All vaults already mounted or no vaults defined"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
SRC_DIR="$(yq eval ".vaults[] | select(.name == \"$SRC_NAME\").encrypted_path" "$CONFIG_FILE")"
|
||||||
|
DST_DIR="$MNTDIR/$SRC_NAME.Open"
|
||||||
|
if [ -z "$SRC_DIR" ]; then
|
||||||
|
>&2 echo "ERROR: Unable to detect encrypted_path for $SRC_NAME"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if ! gxvault_gocryptfs_is_mounted "$SRC_DIR" ; then
|
||||||
|
echo "Opening vault $SRC_NAME..."
|
||||||
|
# echo " src_name=$SRC_NAME"
|
||||||
|
# echo " src_dir=$SRC_DIR"
|
||||||
|
# echo " dst_dir=$DST_DIR"
|
||||||
|
gxvault_gocryptfs_mount \
|
||||||
|
"$SRC_NAME" \
|
||||||
|
"$SRC_DIR" \
|
||||||
|
"$DST_DIR"
|
||||||
|
else
|
||||||
|
echo "Closing vault $SRC_NAME..."
|
||||||
|
fusermount -u "$DST_DIR"
|
||||||
|
printf '\033[32mVault %s is now closed\033[0m\n' "$SRC_NAME"
|
||||||
|
fi
|
||||||
|
|
23
shard.yml
Normal file
23
shard.yml
Normal file
|
@ -0,0 +1,23 @@
|
||||||
|
name: gx-vault
|
||||||
|
version: 0.1.0
|
||||||
|
|
||||||
|
targets:
|
||||||
|
gx-vault:
|
||||||
|
main: src/main.cr
|
||||||
|
|
||||||
|
# authors:
|
||||||
|
# - name <email@example.com>
|
||||||
|
|
||||||
|
# description: |
|
||||||
|
# Short description of gx-vault
|
||||||
|
|
||||||
|
# dependencies:
|
||||||
|
# pg:
|
||||||
|
# github: will/crystal-pg
|
||||||
|
# version: "~> 0.5"
|
||||||
|
|
||||||
|
# development_dependencies:
|
||||||
|
# webmock:
|
||||||
|
# github: manastech/webmock.cr
|
||||||
|
|
||||||
|
# license: MIT
|
93
src/main.cr
Normal file
93
src/main.cr
Normal file
|
@ -0,0 +1,93 @@
|
||||||
|
require "yaml"
|
||||||
|
require "colorize"
|
||||||
|
require "json"
|
||||||
|
|
||||||
|
# Vault class
|
||||||
|
class Vault
|
||||||
|
getter name : String
|
||||||
|
getter encrypted_path : String
|
||||||
|
getter mount_dir : String
|
||||||
|
|
||||||
|
def initialize(@name, @encrypted_path, mount_name : String)
|
||||||
|
home_dir = ENV["HOME"] || raise "Home directory not found"
|
||||||
|
@mount_dir = File.join(home_dir, "mnt/#{mount_name}")
|
||||||
|
end
|
||||||
|
|
||||||
|
def mounted? : Bool
|
||||||
|
`mount`.includes?("#{encrypted_path} on #{mount_dir}")
|
||||||
|
end
|
||||||
|
|
||||||
|
def mount
|
||||||
|
Dir.mkdir_p(mount_dir) unless Dir.exists?(mount_dir)
|
||||||
|
|
||||||
|
if mounted?
|
||||||
|
puts "Already mounted. Skipping.".colorize(:yellow)
|
||||||
|
else
|
||||||
|
input = STDIN
|
||||||
|
output = STDOUT
|
||||||
|
error = STDERR
|
||||||
|
process = Process.new("gocryptfs", ["-idle", "15m", encrypted_path, mount_dir], input: input, output: output, error: error)
|
||||||
|
unless process.wait.success?
|
||||||
|
puts "Error mounting the vault".colorize(:red)
|
||||||
|
return
|
||||||
|
end
|
||||||
|
puts "Vault #{name} is now available on #{mount_dir}".colorize(:green)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
def unmount
|
||||||
|
`fusermount -u #{mount_dir}`
|
||||||
|
puts "Vault #{name} is now closed.".colorize(:green)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
# Config class
|
||||||
|
class Config
|
||||||
|
getter vaults : Array(Vault)
|
||||||
|
|
||||||
|
def initialize(config_path : String)
|
||||||
|
@vaults = [] of Vault
|
||||||
|
load_vaults(config_path)
|
||||||
|
end
|
||||||
|
|
||||||
|
private def load_vaults(config_path : String)
|
||||||
|
yaml_data = YAML.parse(File.read(config_path))
|
||||||
|
vaults_data = yaml_data["vaults"].as_a
|
||||||
|
|
||||||
|
vaults_data.each do |vault_data|
|
||||||
|
name = vault_data["name"].as_s
|
||||||
|
encrypted_path = vault_data["encrypted_path"].as_s
|
||||||
|
@vaults << Vault.new(name, encrypted_path, "#{name}.Open")
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
# Main execution starts here
|
||||||
|
home_dir = ENV["HOME"] || raise "Home directory not found"
|
||||||
|
config = Config.new(File.join(home_dir, ".config/gx-vault.yml"))
|
||||||
|
|
||||||
|
# Correcting the fzf interaction part
|
||||||
|
input = IO::Memory.new
|
||||||
|
config.vaults.each do |vault|
|
||||||
|
name_display = vault.mounted? ? "#{vault.name} [#{ "open".colorize(:green) }]" : vault.name
|
||||||
|
input.puts name_display
|
||||||
|
end
|
||||||
|
input.rewind
|
||||||
|
|
||||||
|
output = IO::Memory.new
|
||||||
|
error = STDERR
|
||||||
|
process = Process.new("fzf", ["--ansi"], input: input, output: output, error: error)
|
||||||
|
|
||||||
|
unless process.wait.success?
|
||||||
|
puts "Error executing fzf: #{error.to_s.strip}".colorize(:red)
|
||||||
|
exit(1)
|
||||||
|
end
|
||||||
|
|
||||||
|
selected_vault_name = output.to_s.strip.split.first?
|
||||||
|
selected_vault = config.vaults.find { |v| v.name == selected_vault_name }
|
||||||
|
|
||||||
|
if selected_vault
|
||||||
|
selected_vault.mounted? ? selected_vault.unmount : selected_vault.mount
|
||||||
|
else
|
||||||
|
puts "Vault not found.".colorize(:red)
|
||||||
|
end
|
||||||
|
|
Loading…
Reference in a new issue