#!/usr/bin/python
#
# Cumulus: Efficient Filesystem Backup to the Cloud
# Copyright (C) 2012 The Cumulus Developers
# See the AUTHORS file for a list of contributors.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

# Print a dump containing file metadata and content digests for a specified
# file system tree.  Used to compare file system trees when running tests.

import hashlib
import os
import stat
import sys

include_mode = True
include_mtime = True

def stat_file(path):
    st = os.lstat(path)
    metadata = [path]
    if include_mode:
        metadata.append("mode=%o" % st.st_mode)
    if include_mtime:
        metadata.append("mtime=%d" % st.st_mtime)
    if stat.S_ISREG(st.st_mode):
        digest = hashlib.sha256()
        BUF_SIZE = 1 << 16
        f = open(path, "r")
        while True:
            buf = f.read(BUF_SIZE)
            if not buf: break
            digest.update(buf)
        metadata.append("sha256=%s" % digest.hexdigest())
    return metadata

def dump_tree(root):
    files = []
    for (dirpath, dirnames, filenames) in os.walk(root):
        for d in dirnames:
            files.append(stat_file(os.path.join(dirpath, d)))
        for f in filenames:
            files.append(stat_file(os.path.join(dirpath, f)))
    files.sort()
    return files

if __name__ == "__main__":
    if len(sys.argv) > 1:
        root = sys.argv[1]
    else:
        root = "."
    os.chdir(root)
    for file in dump_tree("."):
        print " ".join(map(str, file))