stagemanager: Move functionality into the scripts directory

Since scripts is now in PATH thanks to the layer functionality there is
no longer any need to have this recipe full of special cases, the scripts
can just be placed there.

Signed-off-by: Richard Purdie <rpurdie@linux.intel.com>

1 files changed, 156 insertions, 0 deletions

diff --git a/scripts/stage-manager b/scripts/stage-manager
new file mode 100755
index 0000000000..536d1afda0
--- /dev/null
+++ b/scripts/stage-manager
@@ -0,0 +1,156 @@
+#!/usr/bin/env python
+
+# Copyright (C) 2006-2007 Richard Purdie
+#
+# This program is free software; you can redistribute it and/or modify it under
+# the terms of the GNU General Public License version 2 as published by the Free
+# Software Foundation;
+#
+# 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.
+
+import optparse
+import os, sys, stat
+
+__version__ = "0.0.1"
+
+
+def write_cache(cachefile, cachedata):
+    fd = open(cachefile, 'w')
+    for f in cachedata:
+        s = f + '|' + str(cachedata[f]['ts']) + '|' + str(cachedata[f]['size'])
+        fd.write(s + '\n')
+    fd.close()
+
+def read_cache(cachefile):
+    cache = {}
+    f = open(cachefile, 'r')
+    lines = f.readlines()
+    f.close()
+    for l in lines:
+        data = l.split('|')
+        cache[data[0]] = {}
+        cache[data[0]]['ts'] = int(data[1])
+        cache[data[0]]['size'] = int(data[2])
+        cache[data[0]]['seen'] = False
+    return cache
+
+def mkdirhier(dir):
+    """Create a directory like 'mkdir -p', but does not complain if
+    directory already exists like os.makedirs
+    """
+    try:
+        os.makedirs(dir)
+    except OSError, e:
+        if e.errno != 17: raise e
+
+if __name__ == "__main__":
+    parser = optparse.OptionParser( version = "Metadata Stage Manager version %s" % ( __version__ ),
+    usage = """%prog [options]\n\nPerforms mamagement tasks on a metadata staging area.""" )
+
+    parser.add_option( "-p", "--parentdir", help = "the path to the metadata parent directory",
+               action = "store", dest = "parentdir", default = None)
+
+    parser.add_option( "-c", "--cachefile", help = "the cache file to use",
+               action = "store", dest = "cachefile", default = None)
+
+    parser.add_option( "-d", "--copydir", help = "copy changed files to this directory",
+               action = "store", dest = "copydir", default = None)
+
+    parser.add_option( "-u", "--update", help = "update the cache file",
+               action = "store_true", dest = "update", default = False)
+
+    (options, args) = parser.parse_args()
+
+    if options.parentdir is None:
+        print("Error, --parentdir option not supplied")
+        sys.exit(1)
+
+    if options.cachefile is None:
+        print("Error, --cachefile option not supplied")
+        sys.exit(1)
+
+    if not options.parentdir.endswith('/'):
+        options.parentdir = options.parentdir + '/'
+
+    cache = {}
+    if os.access(options.cachefile, os.F_OK):
+        cache = read_cache(options.cachefile)
+
+    found_difference = False
+
+    def updateCache(path, fstamp):
+        cache[path] = {}
+        cache[path]['ts'] = fstamp[stat.ST_MTIME]
+        cache[path]['size'] = fstamp[stat.ST_SIZE]
+        cache[path]['seen'] = True
+        found_difference = True
+
+    def copyfile(path):
+        if options.copydir:
+            copypath = os.path.join(options.copydir, path.replace(options.parentdir, '', 1))
+            mkdirhier(os.path.split(copypath)[0])
+            os.system("cp -dp " + path + " " + copypath)
+
+    def copydir(path, fstamp):
+        if options.copydir:
+            copypath = os.path.join(options.copydir, path.replace(options.parentdir, '', 1))
+            if os.path.exists(copypath):
+                os.system("rm -rf " + copypath)
+            if os.path.islink(path):
+                os.symlink(os.readlink(path), copypath)
+            else:
+                mkdirhier(copypath)
+                os.utime(copypath, (fstamp[stat.ST_ATIME], fstamp[stat.ST_MTIME]))
+
+    for root, dirs, files in os.walk(options.parentdir):
+        for f in files:
+            path = os.path.join(root, f)
+            if not os.access(path, os.R_OK):
+                continue
+            fstamp = os.lstat(path)
+            if path not in cache:
+                print "new file %s" % path
+                updateCache(path, fstamp)
+                copyfile(path)
+            else:
+                if cache[path]['ts'] != fstamp[stat.ST_MTIME] or cache[path]['size'] != fstamp[stat.ST_SIZE]:
+                    print "file %s changed" % path
+                    updateCache(path, fstamp)
+                    copyfile(path)
+                cache[path]['seen'] = True
+        for d in dirs:
+            path = os.path.join(root, d)
+            fstamp = os.lstat(path)
+            if path not in cache:
+                print "new dir %s" % path
+                updateCache(path, fstamp)
+                copydir(path, fstamp)
+            else:
+                if cache[path]['ts'] != fstamp[stat.ST_MTIME]:
+                    print "dir %s changed" % path
+                    updateCache(path, fstamp)
+                    copydir(path, fstamp)
+                cache[path]['seen'] = True
+
+    todel = []
+    for path in cache:
+        if not cache[path]['seen']:
+            print "%s removed" % path
+            found_difference = True
+            todel.append(path)
+
+    if options.update:
+        print "Updating"
+        for path in todel:
+            del cache[path]
+        mkdirhier(os.path.split(options.cachefile)[0])
+        write_cache(options.cachefile, cache)
+
+    if found_difference:
+        sys.exit(5)
+    sys.exit(0)
+
+
+