#!/usr/bin/env bash
# This file is part of GUFI, which is part of MarFS, which is released
# under the BSD license.
#
#
# Copyright (c) 2017, Los Alamos National Security (LANS), LLC
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without modification,
# are permitted provided that the following conditions are met:
#
# 1. Redistributions of source code must retain the above copyright notice, this
# list of conditions and the following disclaimer.
#
# 2. Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions and the following disclaimer in the documentation and/or
# other materials provided with the distribution.
#
# 3. Neither the name of the copyright holder nor the names of its contributors
# may be used to endorse or promote products derived from this software without
# specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
# IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
# INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
# BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
# OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
# ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
#
# From Los Alamos National Security, LLC:
# LA-CC-15-039
#
# Copyright (c) 2017, Los Alamos National Security, LLC All rights reserved.
# Copyright 2017. Los Alamos National Security, LLC. This software was produced
# under U.S. Government contract DE-AC52-06NA25396 for Los Alamos National
# Laboratory (LANL), which is operated by Los Alamos National Security, LLC for
# the U.S. Department of Energy. The U.S. Government has rights to use,
# reproduce, and distribute this software.  NEITHER THE GOVERNMENT NOR LOS
# ALAMOS NATIONAL SECURITY, LLC MAKES ANY WARRANTY, EXPRESS OR IMPLIED, OR
# ASSUMES ANY LIABILITY FOR THE USE OF THIS SOFTWARE.  If software is
# modified to produce derivative works, such modified software should be
# clearly marked, so as not to confuse it with the version available from
# LANL.
#
# THIS SOFTWARE IS PROVIDED BY LOS ALAMOS NATIONAL SECURITY, LLC AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
# THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL LOS ALAMOS NATIONAL SECURITY, LLC OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
# OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
# IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
# OF SUCH DAMAGE.


set -e

if [[ "$#" -lt 1 ]]; then
    echo "Syntax: $0 GUFI_scripts_prefix"
    exit 1
fi

# if [[ "$UID" -ne 0 ]]; then
#     echo "This script requires sudo"
#     exit 1
# fi

GUFI_PREFIX=$(realpath "$1")

echo
echo "Top 10 Users taking up the most space"
echo 3 > /proc/sys/vm/drop_caches
sudo "${GUFI_PREFIX}/gufi_stats" --order most --num_results 10 total-filesize

echo
echo "Total Number of Files"
echo 3 > /proc/sys/vm/drop_caches
sudo "${GUFI_PREFIX}/gufi_stats" -c total-filecount

echo
echo "Total Space Taken"
echo 3 > /proc/sys/vm/drop_caches
sudo "${GUFI_PREFIX}/gufi_stats" -c total-filesize

echo
echo "Top 10 Users with the most files"
echo 3 > /proc/sys/vm/drop_caches
MOST_FILES="$(sudo "${GUFI_PREFIX}/gufi_stats" --order most --num_results 10 total-filecount)"
echo "${MOST_FILES}"

MOST_FILES="$(echo "${MOST_FILES}" | head -n 1 | awk -F' ' '{printf $1}')"

echo
echo "Number of files owned by user ${MOST_FILES} (gufi_find)"
echo 3 > /proc/sys/vm/drop_caches
sudo -u \#"${MOST_FILES}" "${GUFI_PREFIX}/gufi_find" -type f | wc -l

echo
echo "Number of files owned by user ${MOST_FILES} (gufi_stats)"
echo 3 > /proc/sys/vm/drop_caches
sudo -u \#"${MOST_FILES}" "${GUFI_PREFIX}/gufi_stats" -c total-filecount

echo
echo "Number of files owned by user ${MOST_FILES} larger than 1KB (gufi_find)"
echo 3 > /proc/sys/vm/drop_caches
sudo -u \#"${MOST_FILES}" "${GUFI_PREFIX}/gufi_find" -type f -size=+1024c | wc -l
