#!/usr/bin/env bash

# Copyright (c) 2018 rehashedsalt/vintagesalt
# 
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
# 
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
# 
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.

## Clean up some variables that I use in device-specific scripts
# General
unset TERMINAL
# Polybar
unset PB_MONITOR
unset PB_WM_RESTACK
unset PB_BAR_PRIMARY_MODULES_LEFT PB_BAR_PRIMARY_MODULES_CENTER PB_BAR_PRIMARY_MODULES_RIGHT
unset PB_BAR_PRIMARY_2_MODULES_LEFT PB_BAR_PRIMARY_2_MODULES_CENTER PB_BAR_PRIMARY_2_MODULES_RIGHT
unset PB_BAR_SECONDARY_MODULES_LEFT PB_BAR_SECONDARY_MODULES_CENTER PB_BAR_SECONDARY_MODULES_RIGHT
unset PB_BAR_SECONDARY_2_MODULES_LEFT PB_BAR_SECONDARY_2_MODULES_CENTER PB_BAR_SECONDARY_2_MODULES_RIGHT
unset PB_MODULE_BAR_WIDTH
unset PB_MODULE_ETH_INTERFACE
unset PB_MODULE_WLAN_INTERFACE
# bspwm monitorset.sh
unset BSPWM_DESKTOPS_TOTAL
unset BSPWM_DESKTOPS_PRIMARY
# xob
unset XOB_PIPE_BRIGHTNESS
unset XOB_PIPE_VOLUME
unset XOB_PIPE_MISC

## Patch path, if not already done
if [[ "$PATH" != "*$HOME/.local/bin*" ]]; then
	printf "[INFO] Patching path for $HOME/.local/bin\n"
	export PATH="$HOME/.local/bin:$PATH"
fi

## Load device-specific scripts
if [[ -x "$HOME/.config/dtfscripts/$(hostname).rc" ]]; then
	printf "[INFO] Sourcing ~/.config/dtfscripts/$(hostname).rc\n"
	source $HOME/.config/dtfscripts/$(hostname).rc
fi

printf "[INFO] Performing simple configuration\n"
## SIMPLE CONFIGURATION
xset -b
xset -dpms
xset s off
# This depends on a script found at rehashedsalt/bin
xrdbupdate

## SERVICES AND DAEMONS
# Kill everything
printf "[INFO] Stopping existing services\n"
services="plasmashell xembedsniproxy krunner xfdesktop polybar compton conky sxhkd dunst xob"
for service in $services; do
	printf "  [INFO] Sending signal to all $service\n"
	killall $service
done
# Wait for them to die
for service in $services; do
	i=0
	while pgrep $service > /dev/null 2>&1; do
		if [[ "$i" = "10" ]]; then
			printf "  [INFO] Waited too long, killing $service with prejudice\n"
			pkill -9 $service
			break
		else
			printf "  [INFO] Waiting on $service\n"
			sleep 0.5
			i=$(expr $i + 1)
		fi
	done
	unset i
done

# Start them back up
printf "[INFO] Starting services\n"
# Start with key functionality
# SXHKD
if pgrep bspwm > /dev/null 2>&1; then
	printf "  [INFO] Found bspwm, starting sxhkd\n"
	sxhkd &
else
	printf "  [WARN] bspwm was started, but sxhkd wasn't! Session may be unusable!\n"
fi

# Compton
if which compton > /dev/null 2>&1; then
	# Don't start the compositor in a VM
	if [[ $(hostname) != "vm"* ]]; then
		printf "  [INFO] Starting compton\n"
		compton &
	else
		printf "  [INFO] In a VM, not starting compositor\n"
	fi
else
	printf "  [INFO] Could not find a compositor\n"
fi

# Polybar
if which polybar > /dev/null 2>&1; then
	printf "  [INFO] Starting polybar\n"
	$HOME/.config/polybar/start.sh &
else
	printf "  [INFO] Could not find a bar\n"
fi

# Dunst
if which dunst > /dev/null 2>&1; then
	printf "  [INFO] Starting dunst\n"
	dunst&
else
	printf "  [INFO] Could not find a notification daemon\n"
fi

# XOB
if which xob > /dev/null 2>&1; then
	printf "  [INFO] Starting xob instances\n"
	pipes="XOB_PIPE_BRIGHTNESS XOB_PIPE_VOLUME XOB_PIPE_MISC"
	# Clean up earlier temp folders
	for file in /tmp/$USER-xob-*; do
		printf "    [INFO] Found old xob directory \"$file\"\n"
		for pipe in $pipes; do
			if [ -p "$file/$pipe" ]; then
				printf "      [INFO] Contained old pipe \"$pipe\". Removing\n"
				rm "$file/$pipe"
			fi
		done
		rmdir "$file" && printf "    [INFO] Successfully cleaned up\n" || printf "    [WARN] Failed to clean up\n"
	done
	# Set up new pipes
	export XOB_PIPE_DIR="/tmp/$USER-xob-$(date +%s)-$(uuidgen)"
	printf "    [INFO] Using pipe directory \"$XOB_PIPE_DIR\"\n"
	mkdir -m 700 "$XOB_PIPE_DIR"
	for pipe in $pipes; do
		if [ -p "$XOB_PIPE_DIR/$pipe" ]; then
			printf "    [INFO] Removing existing pipe \"$pipe\"\n"
			rm "$XOB_PIPE_DIR/$pipe"
		fi
		mkfifo "$XOB_PIPE_DIR/$pipe"
		if [ -p "$XOB_PIPE_DIR/$pipe" ]; then
			printf "    [INFO] Successfully created pipe \"$pipe\". Binding an xob instance to it\n"
			tail -f "$XOB_PIPE_DIR/$pipe" | xob -s "$pipe" &
		else
			printf "    [WARN] Failed to create pipe \"$pipe\"\n"
		fi
	done
else
	printf "  [INFO] Could not find a bar overlay application\n"
fi

# Conky
if which conky > /dev/null 2>&1; then
	printf "  [INFO] Starting conky\n"
	# Use all the config files in its config directory
	for file in $HOME/.config/conky/*.conf; do
		printf "    [INFO] Found config file $file\n"
		conky -c $file&
	done
else
	printf "  [INFO] Could not find a desktop statistics application\n"
fi

# Wallpaper management
printf "[INFO] Setting wallpaper\n"
if which xfdesktop > /dev/null 2>&1; then
	printf "  [INFO] Starting xfdesktop\n"
	xfdesktop --disable-wm-check&
elif which feh > /dev/null 2>&1; then
	printf "  [INFO] Setting wallpaper with feh\n"
	printf "  [WARN] This solution may cause problems on multi-monitor systems\n"
	feh --randomize --bg-fill ~/Pictures/Wallpapers/.active &
elif which xsetroot > /dev/null 2>&1; then
	printf "  [INFO] Setting root window color\n"
	xsetroot -solid "#282828"
else
	printf "  [WARN] Could not find a wallpaper manager\n"
fi

printf "[INFO] Finished initialization\n"