226 lines
7.2 KiB
Bash
226 lines
7.2 KiB
Bash
#!/bin/zsh
|
|
|
|
# Common global variables that specify what to cache where.
|
|
FIG_CACHE_BASE_DIR=/tmp/fig-cache
|
|
FIG_STATE_INDICATOR_FILES=(".hg/blackbox.log" ".citc/manifest.rawproto")
|
|
FIG_STATUS_FILE="status.txt"
|
|
FIG_STATE_HASH_FILE="state.hash"
|
|
# Lines from .blackbox.log matching the following pattern will be ignored for
|
|
# hash computation and thus will prevent the status being updated. Add commands
|
|
# in here that do not change the Fig state.
|
|
BLACKBOX_IGNORE=">.*(xl|ll|figstatus|status|diff|root|exportedcl|preloading|debugfetchconfig)"
|
|
|
|
# Returns the root directory of the current Fig client. Should only be called
|
|
# from within a Fig client.
|
|
function get_fig_client_root() {
|
|
local root_dir="${PWD%/google3*}"
|
|
echo "$root_dir"
|
|
}
|
|
|
|
# Returns the name of the current Fig client. Should only be called from within
|
|
# a Fig client.
|
|
function get_fig_client_name() {
|
|
local root_dir="$( get_fig_client_root )"
|
|
local client_name="${root_dir##/google/*/}"
|
|
echo "$client_name"
|
|
}
|
|
|
|
# Returns the directory where we cache the status for the current Fig client.
|
|
function get_fig_client_cache_dir() {
|
|
[ ! -d "$FIG_CACHE_BASE_DIR" ] && mkdir "$FIG_CACHE_BASE_DIR"
|
|
client_name=$( get_fig_client_name )
|
|
client_cache_dir="$FIG_CACHE_BASE_DIR/$client_name"
|
|
# Make sure the cache dir exists.
|
|
[ ! -d "$client_cache_dir" ] && mkdir "$client_cache_dir"
|
|
echo "$client_cache_dir"
|
|
}
|
|
|
|
# Returns the hash of the file state of the current Fig client.
|
|
function get_fig_client_state_hash() {
|
|
local state_hash=""
|
|
local -a state_files
|
|
for file in ${FIG_STATE_INDICATOR_FILES}; do
|
|
file="$( get_fig_client_root )/${file}"
|
|
if [ -f "$file" ]; then
|
|
if [[ "$file" =~ ".*blackbox.log" ]]; then
|
|
state_hash="${state_hash}\n$( egrep -v "${BLACKBOX_IGNORE}" "$file" | \
|
|
md5sum )"
|
|
else
|
|
state_hash="${state_hash}\n$( md5sum "$file" )"
|
|
fi
|
|
fi
|
|
done
|
|
echo "$state_hash"
|
|
}
|
|
|
|
# Returns 0 if the status cache for the current Fig client is outdated and
|
|
# needs to be updated and 1 otherwise. Should only be called from within a Fig
|
|
# client.
|
|
function should_update_fig_status_cache() {
|
|
local state_hash="$( get_fig_client_state_hash )"
|
|
local cached_state_hash="emptycache"
|
|
local cached_state_hash_file
|
|
cached_state_hash_file="$( get_fig_client_cache_dir )/$FIG_STATE_HASH_FILE"
|
|
if [ -f "$cached_state_hash_file" ]; then
|
|
cached_state_hash=$(<"$cached_state_hash_file")
|
|
fi
|
|
if [ "$cached_state_hash" != "$state_hash" ]; then
|
|
# Update cache since the state hash has changed.
|
|
return 0
|
|
fi
|
|
|
|
# Do not update cache.
|
|
return 1
|
|
}
|
|
|
|
# Updates the status cache for the current Fig client. Should only be called
|
|
# from within a Fig client.
|
|
function update_fig_status_cache() {
|
|
local status_file="$( get_fig_client_cache_dir )/$FIG_STATUS_FILE"
|
|
hg figstatus >! "$status_file"
|
|
|
|
local state_hash="$( get_fig_client_state_hash )"
|
|
local cached_state_hash_file
|
|
cached_state_hash_file="$( get_fig_client_cache_dir )/$FIG_STATE_HASH_FILE"
|
|
echo "$state_hash" >! "$cached_state_hash_file"
|
|
}
|
|
|
|
# Returns a multi-line output of the status of the current Fig client by either
|
|
# looking the status up in the cache file if the cache is current or loading it
|
|
# directly from Fig and updating the cache. Should only be called from within a
|
|
# Fig client.
|
|
function fig_status() {
|
|
if should_update_fig_status_cache; then
|
|
update_fig_status_cache
|
|
fi
|
|
local status_file="$( get_fig_client_cache_dir )/$FIG_STATUS_FILE"
|
|
cat "$status_file"
|
|
}
|
|
|
|
# Returns whether the current directory is (likely) a Fig-on-CITC client.
|
|
function is_fig_client() {
|
|
if [[ "$PWD" =~ /google/src/cloud/$USER/* ]] && \
|
|
[[ -d ${PWD%%/google3*}/.hg ]]; then
|
|
return 0
|
|
fi
|
|
return 1
|
|
}
|
|
|
|
# Returns a pseudo-template string that can be used by get_fig_prompt to create
|
|
# a ZSH Fig prompt.
|
|
#
|
|
# The prompt returned by get_fig_prompt can be easily customized by overriding
|
|
# this function.
|
|
#
|
|
# Template Arguments:
|
|
# FIG_PROMPT_MODIFIED: Replaced with $modified
|
|
# FIG_PROMPT_ADDED: Replaced with $added
|
|
# FIG_PROMPT_DELETED: Replaced with $deleted
|
|
# FIG_PROMPT_UNKNOWN: Replaced with $unknown
|
|
# FIG_PROMPT_UNEXPORTED: Replaced with $unexported
|
|
# FIG_PROMPT_OBSOLETE: Replaced with $obsolete
|
|
# FIG_PROMPT_CL: Replaced with $cl
|
|
# FIG_PROMPT_DESCRIPTION: Replaced with $description
|
|
# FIG_PROMPT_CHANGENAME: Replaced with $changename
|
|
# FIG_PROMPT_HAS_SHELVE: Replaced with $has_shelve
|
|
function get_fig_prompt_template() {
|
|
echo -n '%b:%B%F{white}['
|
|
echo -n '%F{yellow}FIG_PROMPT_MODIFIED%F{green}FIG_PROMPT_ADDED'
|
|
echo -n '%F{red}FIG_PROMPT_DELETED%F{cyan}FIG_PROMPT_UNKNOWN'
|
|
echo -n '%F{magenta}FIG_PROMPT_HAS_SHELVE%F{white}FIG_PROMPT_CHANGENAME'
|
|
echo -n '%F{green}FIG_PROMPT_UNEXPORTED%F{red}FIG_PROMPT_OBSOLETE'
|
|
echo -n '%F{white}]%b'
|
|
}
|
|
|
|
# Returns a string that can be used in a ZSH prompt to display the status of
|
|
# the current Fig client.
|
|
function get_fig_prompt() {
|
|
if ! is_fig_client; then
|
|
return 0
|
|
fi
|
|
figstatus=("${(@f)$( fig_status )}")
|
|
modified=$figstatus[1]
|
|
added=$figstatus[2]
|
|
deleted=$figstatus[3]
|
|
unknown=$figstatus[4]
|
|
unexported=$figstatus[5]
|
|
obsolete=$figstatus[6]
|
|
cl=$figstatus[7]
|
|
description=$figstatus[8]
|
|
branch=$figstatus[9]
|
|
if [ -z "$branch" ]; then
|
|
branch="$cl"
|
|
fi
|
|
changename=$figstatus[13]
|
|
if [ -z "$changename" ]; then
|
|
changename="$branch"
|
|
fi
|
|
has_shelve=""
|
|
# POSIX-compatible way to check whether shelved-directory is non-empty.
|
|
shelve_dir="$( get_fig_client_root )/.hg/shelved/"
|
|
if [ -d "$shelve_dir" ] && /bin/ls -1qA "$shelve_dir" | grep -q .; then
|
|
has_shelve="!"
|
|
fi
|
|
|
|
local tpl="$(get_fig_prompt_template)"
|
|
tpl="${tpl//FIG_PROMPT_MODIFIED/$modified}"
|
|
tpl="${tpl//FIG_PROMPT_ADDED/$added}"
|
|
tpl="${tpl//FIG_PROMPT_DELETED/$deleted}"
|
|
tpl="${tpl//FIG_PROMPT_UNKNOWN/$unknown}"
|
|
tpl="${tpl//FIG_PROMPT_UNEXPORTED/$unexported}"
|
|
tpl="${tpl//FIG_PROMPT_OBSOLETE/$obsolete}"
|
|
tpl="${tpl//FIG_PROMPT_CL/$cl}"
|
|
tpl="${tpl//FIG_PROMPT_DESCRIPTION/$description}"
|
|
tpl="${tpl//FIG_PROMPT_CHANGENAME/$changename}"
|
|
tpl="${tpl//FIG_PROMPT_HAS_SHELVE/$has_shelve}"
|
|
echo "$tpl"
|
|
}
|
|
|
|
typeset -Ag FIG_STATUS_INDICES
|
|
FIG_STATUS_INDICES[modified]=1
|
|
FIG_STATUS_INDICES[added]=2
|
|
FIG_STATUS_INDICES[deleted]=3
|
|
FIG_STATUS_INDICES[unknown]=4
|
|
FIG_STATUS_INDICES[unexported]=5
|
|
FIG_STATUS_INDICES[obsolete]=6
|
|
FIG_STATUS_INDICES[cl]=7
|
|
FIG_STATUS_INDICES[description]=8
|
|
FIG_STATUS_INDICES[branch]=9
|
|
FIG_STATUS_INDICES[changename]=13
|
|
|
|
# Returns the value of the specified property in the current Fig repository.
|
|
# While not as fast as get_fig_prompt, allows for personal configuration of the
|
|
# prompt string in a user's zshrc.
|
|
#
|
|
# TODO(cclausen): This would be far more convenient (and likely quicker) if we
|
|
# could pass around an associative array instead of reparsing the output for
|
|
# each property being used.
|
|
#
|
|
# Example usage:
|
|
# function my_hg_prompt {
|
|
# echo "$(fig_prop "modified") $(fig_prop "cl") $(fig_prop "description")"
|
|
# }
|
|
# PROMPT='%n@%m[$(my_hg_prompt)] %#'
|
|
#
|
|
# Prompt output:
|
|
# user@host[? 181818181 Excerpt from CL description] %
|
|
#
|
|
# Valid properties (from figstatus):
|
|
# modified
|
|
# added
|
|
# deleted
|
|
# unknown
|
|
# unexported
|
|
# obsolete
|
|
# cl
|
|
# description
|
|
# branch
|
|
function fig_prop() {
|
|
if ! is_fig_client; then
|
|
return 0
|
|
fi
|
|
|
|
figstatus=("${(@f)$( fig_status )}")
|
|
echo "${figstatus[$FIG_STATUS_INDICES[$1]]}"
|
|
}
|