1 files changed, 109 insertions, 1 deletions
diff --git a/scripts/scriptslib.py b/scripts/scriptslib.py
index 68470c28..1f79fb95 100644
--- a/scripts/scriptslib.py
+++ b/scripts/scriptslib.py
@@ -17,7 +17,8 @@
 
 import yaml
 import subprocess
-
+import os
+import sys
 
 aliases = {
   'baserock:': 'git://%(trove)s/baserock/',
@@ -43,3 +44,110 @@ def definitions_root():
 def load_yaml_file(yaml_file):
     with open(yaml_file, 'r') as f:
         return yaml.safe_load(f)
+
+
+class BaserockMeta(object):
+    '''An object representing Baserock metadata contained in a Baserock
+       system image, for available metadata formats'''
+
+    def __init__(self):
+        self.metas = {}
+
+    def get_each(self):
+        '''Yield an iterable for the whole list of metas'''
+        for key in self.metas:
+            yield self.metas[key]
+
+    def get_name(self, name):
+        '''Yield an iterable of metadata matched by name, e.g. `bash`'''
+        for key in self.metas:
+            if self.metas[key]['source-name'] == name:
+                yield self.metas[key]
+
+    def import_meta(self, meta_text):
+        importers = (self.import_meta_ybd,
+                     self.import_meta_morph)
+
+        for i in importers:
+            try:
+                i(meta_text)
+                return
+            except (KeyError, Exception) as err:
+                pass
+
+        # Shouldn't get here
+        sys.stderr.write('Metadata format not recognised.\n'
+                         'Error:\n')
+        raise err
+
+    def import_meta_morph(self, meta_text):
+        self._add_meta(yaml.load(meta_text))
+
+    def import_meta_ybd(self, meta_text):
+        source = yaml.load(meta_text)
+
+        null = '0' * 32
+
+        if 'repo' not in source:
+            kind = 'stratum'
+            contents = 'components'
+            source['repo'] = 'upstream:definitions'
+            source['ref'] = null # No ref info
+        else:
+            kind = 'chunk'
+            contents = 'files'
+
+        repo = parse_repo_alias(source['repo'])
+        source_name = '-'.join(
+                      source['products'][0]['artifact'].split('-')[:-1])
+
+        # Needed until YBD provides cache-key in metadata
+        if not 'cache-key' in source:
+            source['cache-key'] = null
+
+        for product in source['products']:
+
+            self._add_meta({
+                'kind':           kind,
+                'source-name':    source_name,
+                'artifact-name':  product['artifact'],
+                'contents':       product[contents],
+                'repo':           repo,
+                'repo-alias':     source['repo'],
+                'sha1':           source['ref'],
+                'original_ref':   source['ref'],
+                'cache-key':      source['cache-key']
+                })
+
+    def _add_meta(self, meta_dict):
+        '''Validate and add a meta'''
+
+        ignore = ('configuration',
+                  'system-artifact-name')
+
+        for i in ignore:
+            if i in meta_dict:
+                return
+
+        required_fields = ('repo', 'sha1', 'contents')
+        for f in required_fields:
+            if not f in meta_dict:
+                raise Exception('Metadata format not recognised, no '
+                    'value for \'%s\'. Data: \'%s\''% (f, str(meta_dict)))
+
+        self.metas[meta_dict['artifact-name']] = meta_dict
+
+
+def meta_load_from_dir(meta_dir_path):
+    '''Read Baserock metadata from a directory'''
+
+    files = [f for f in os.listdir(meta_dir_path)
+             if os.path.isfile(os.path.join(meta_dir_path, f))]
+
+    meta = BaserockMeta()
+    for f in files:
+        if f.endswith('.meta'):
+            meta.import_meta(
+                open(os.path.join(meta_dir_path, f), 'r').read())
+
+    return meta