From d06c13f93632218ee2a030f272143d70cb4cac97 Mon Sep 17 00:00:00 2001
From: lyken <lyken@m-labs.hk>
Date: Thu, 27 Jun 2024 16:25:45 +0800
Subject: [PATCH] core: fix crash on invalid subscripting

---
 nac3core/src/typecheck/type_inferencer/mod.rs | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/nac3core/src/typecheck/type_inferencer/mod.rs b/nac3core/src/typecheck/type_inferencer/mod.rs
index af2fd8de..4a991654 100644
--- a/nac3core/src/typecheck/type_inferencer/mod.rs
+++ b/nac3core/src/typecheck/type_inferencer/mod.rs
@@ -1856,6 +1856,18 @@ impl<'a> Inferencer<'a> {
         slice: &ast::Expr<Option<Type>>,
         ctx: ExprContext,
     ) -> InferenceResult {
+        let report_unscriptable_error = |unifier: &mut Unifier| {
+            // User is attempting to index into a value of an unsupported type.
+
+            let value_ty = value.custom.unwrap();
+            let value_ty_str = unifier.stringify(value_ty);
+
+            return report_error(
+                format!("'{value_ty_str}' object is not subscriptable").as_str(),
+                slice.location, // using the slice's location (rather than value's) because it is more clear
+            );
+        };
+
         let ty = self.unifier.get_dummy_var().ty;
         match &slice.node {
             ExprKind::Slice { lower, upper, step } => {
@@ -1871,7 +1883,9 @@ impl<'a> Inferencer<'a> {
                         make_ndarray_ty(self.unifier, self.primitives, Some(ty), Some(ndims))
                     }
 
-                    _ => unreachable!(),
+                    _ => {
+                        return report_unscriptable_error(self.unifier);
+                    }
                 };
                 self.constrain(value.custom.unwrap(), list_like_ty, &value.location)?;
                 Ok(list_like_ty)
@@ -1961,7 +1975,7 @@ impl<'a> Inferencer<'a> {
                         self.constrain(slice.custom.unwrap(), valid_index_ty, &slice.location)?;
                         self.infer_subscript_ndarray(value, slice, ty, ndims)
                     }
-                    _ => unreachable!(),
+                    _ => report_unscriptable_error(self.unifier),
                 }
             }
         }