# Copyright (C) 2009 The Android Open Source Project
# Copyright (c) 2011, The Linux Foundation. All rights reserved.
# Copyright (C) 2017-2020 The LineageOS 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 hashlib
import common
import re

def FullOTA_Assertions(info):
  AddModemAssertion(info)
  return

def FullOTA_InstallBegin(info):
  UnlockVendorPartition(info)
  AddVendorAssertion(info)
  return

def FullOTA_InstallEnd(info):
  RunCustomScript(info, "deunify.sh", "")
  RunCustomScript(info, "devinfo.sh", "")
  return

def IncrementalOTA_Assertions(info):
  AddModemAssertion(info)
  return

def IncrementalOTA_InstallBegin(info):
  UnlockVendorPartition(info)
  AddVendorAssertion(info)
  return

def IncrementalOTA_InstallEnd(info):
  RunCustomScript(info, "deunify.sh", "")
  RunCustomScript(info, "devinfo.sh", "")
  return

def AddVendorAssertion(info):
  info.script.AppendExtra('assert(run_program("/tmp/partprobe.sh", "/dev/block/sde") == "0");');
  info.script.AppendExtra('ifelse(is_mounted("/mnt/vendor"), unmount("/mnt/vendor"));');
  cmd = 'assert(leeco.file_exists("/dev/block/bootdevice/by-name/vendor") == "1" || \
abort("Error: Vendor partition doesn\'t exist! Please reboot to recovery and flash again!"););'
  info.script.AppendExtra(cmd)
  return

def AddModemAssertion(info):
  android_info = info.input_zip.read("OTA/android-info.txt")
  m = re.search(r'require\s+version-modem\s*=\s*(.+)', android_info)
  if m:
    version = m.group(1).rstrip()
    if len(version) and '*' not in version:
      info.script.AppendExtra(('assert(leeco.verify_modem("%s") == "1");' % (version)))
  return

def RunCustomScript(info, name, arg):
  info.script.AppendExtra(('run_program("/tmp/install/bin/%s", "%s");' % (name, arg)))
  return

def UnlockVendorPartition(info):
  info.script.AppendExtra('package_extract_file("install/bin/toybox", "/tmp/toybox");');
  info.script.AppendExtra('package_extract_file("install/bin/sgdisk", "/tmp/sgdisk");');
  info.script.AppendExtra('package_extract_file("install/bin/unlock-vendor.sh", "/tmp/unlock-vendor.sh");');
  info.script.AppendExtra('package_extract_file("install/bin/partprobe.sh", "/tmp/partprobe.sh");');
  info.script.AppendExtra('set_metadata("/tmp/toybox", "uid", 0, "gid", 0, "mode", 0755);');
  info.script.AppendExtra('set_metadata("/tmp/sgdisk", "uid", 0, "gid", 0, "mode", 0755);');
  info.script.AppendExtra('set_metadata("/tmp/unlock-vendor.sh", "uid", 0, "gid", 0, "mode", 0755);');
  info.script.AppendExtra('set_metadata("/tmp/partprobe.sh", "uid", 0, "gid", 0, "mode", 0755);');
  info.script.AppendExtra('ui_print("Checking for vendor partition...");');
  info.script.AppendExtra('if run_program("/tmp/unlock-vendor.sh") != 0 then');
  info.script.AppendExtra('abort("Unlocking vendor partition failed.");');
  info.script.AppendExtra('endif;');