1 files changed, 46 insertions, 0 deletions
diff --git a/releasetools/galaxys2_edify_generator.py b/releasetools/galaxys2_edify_generator.py
new file mode 100755
index 0000000..95bdf48
--- /dev/null
+++ b/releasetools/galaxys2_edify_generator.py
@@ -0,0 +1,46 @@
+#
+# Copyright (C) 2008 The Android Open Source Project
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import os, sys
+
+LOCAL_DIR = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), '..'))
+RELEASETOOLS_DIR = os.path.abspath(os.path.join(LOCAL_DIR, '../../../build/tools/releasetools'))
+
+import edify_generator
+
+class EdifyGenerator(edify_generator.EdifyGenerator):
+    def UnpackPackageFile(self, src, dst):
+      """Unpack a given file from the OTA package into the given
+      destination file."""
+      self.script.append('package_extract_file("%s", "%s");' % (src, dst))
+
+    def EMMCWriteRawImage(self, partition, image):
+      """Write the given package file into the given partition."""
+
+      args = {'partition': partition, 'image': image}
+
+      self.script.append(
+            ('assert(package_extract_file("%(image)s", "/tmp/%(image)s"),\n'
+             '       write_raw_image("/tmp/%(image)s", "%(partition)s"),\n'
+             '       delete("/tmp/%(image)s"));') % args)
+
+    def Unmount(self, mount_point):
+      """Unmount the partition with the given mount_point."""
+      fstab = self.info.get("fstab", None)
+      if fstab:
+        p = fstab[mount_point]
+        self.script.append('unmount("%s");' %
+								(p.mount_point))
+        self.mounts.add(p.mount_point)