From 8a42f31167507f22cf36bc509633ecdb93aa7008 Mon Sep 17 00:00:00 2001
From: cfig <yuyezhong@gmail.com>
Date: Wed, 21 Apr 2021 12:53:34 +0800
Subject: [PATCH] add prompt message for Issue #58

---
 bbootimg/src/main/kotlin/packable/BootImgParser.kt | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/bbootimg/src/main/kotlin/packable/BootImgParser.kt b/bbootimg/src/main/kotlin/packable/BootImgParser.kt
index 876ed09..8b040a2 100644
--- a/bbootimg/src/main/kotlin/packable/BootImgParser.kt
+++ b/bbootimg/src/main/kotlin/packable/BootImgParser.kt
@@ -1,13 +1,13 @@
 package cfig.packable
 
-import avb.AVBInfo
 import avb.blob.Footer
 import cfig.Avb
-import cfig.helper.Helper
 import cfig.bootimg.Common.Companion.probeHeaderVersion
 import cfig.bootimg.v2.BootV2
 import cfig.bootimg.v3.BootV3
+import cfig.helper.Helper
 import com.fasterxml.jackson.databind.ObjectMapper
+import de.vandermeer.asciitable.AsciiTable
 import org.slf4j.LoggerFactory
 import java.io.File
 import java.io.FileInputStream
@@ -51,6 +51,16 @@ class BootImgParser() : IPackable {
     override fun pack(fileName: String) {
         val cfgFile = workDir + fileName.removeSuffix(".img") + ".json"
         log.info("Loading config from $cfgFile")
+        if (!File(cfgFile).exists()) {
+            val tab = AsciiTable().let {
+                it.addRule()
+                it.addRow("'$cfgFile' doesn't exist, did you forget to 'unpack' ?")
+                it.addRule()
+                it
+            }
+            log.info("\n{}", tab.render())
+            return
+        }
         if (3 == probeHeaderVersion(fileName)) {
             ObjectMapper().readValue(File(cfgFile), BootV3::class.java)
                 .pack()