support [] index for bytes

Author: pikasTech

package/PikaStdLib/PikaStdLib_SysObj.c

@@ -161,6 +161,13 @@ Arg* PikaStdLib_SysObj___get__(PikaObj* self, Arg* key, Arg* obj) {
         char_buff[0] = str_pyload[index];
         return arg_setStr(NULL, "", char_buff);
     }
+    if (ARG_TYPE_BYTES == obj_type) {
+        int index = arg_getInt(key);
+        uint8_t* bytes_pyload = arg_getBytes(obj);
+        uint8_t byte_buff[] = " ";
+        byte_buff[0] = bytes_pyload[index];
+        return arg_setBytes(NULL, "", byte_buff, 1);
+    }
     if (ARG_TYPE_OBJECT == obj_type) {
         PikaObj* arg_obj = arg_getPtr(obj);
         obj_setArg(arg_obj, "__key", key);
@@ -198,6 +205,14 @@ void PikaStdLib_SysObj___set__(PikaObj* self,
         str_pyload[index] = str_val[0];
         obj_setStr(self, obj_str, str_pyload);
     }
+    if (ARG_TYPE_BYTES == obj_type) {
+        int index = arg_getInt(key);
+        uint8_t* bytes_val = arg_getBytes(val);
+        uint8_t* bytes_pyload = arg_getBytes(obj);
+        size_t bytes_len = arg_getBytesSize(obj);
+        bytes_pyload[index] = bytes_val[0];
+        obj_setBytes(self, obj_str, bytes_pyload, bytes_len);
+    }
     if (ARG_TYPE_OBJECT == obj_type) {
         PikaObj* arg_obj = arg_getPtr(obj);
         obj_setArg(arg_obj, "__key", key);

port/linux/package/pikascript/pikascript-lib/PikaStdLib/PikaStdLib_SysObj.c

@@ -161,6 +161,13 @@ Arg* PikaStdLib_SysObj___get__(PikaObj* self, Arg* key, Arg* obj) {
         char_buff[0] = str_pyload[index];
         return arg_setStr(NULL, "", char_buff);
     }
+    if (ARG_TYPE_BYTES == obj_type) {
+        int index = arg_getInt(key);
+        uint8_t* bytes_pyload = arg_getBytes(obj);
+        uint8_t byte_buff[] = " ";
+        byte_buff[0] = bytes_pyload[index];
+        return arg_setBytes(NULL, "", byte_buff, 1);
+    }
     if (ARG_TYPE_OBJECT == obj_type) {
         PikaObj* arg_obj = arg_getPtr(obj);
         obj_setArg(arg_obj, "__key", key);
@@ -198,6 +205,14 @@ void PikaStdLib_SysObj___set__(PikaObj* self,
         str_pyload[index] = str_val[0];
         obj_setStr(self, obj_str, str_pyload);
     }
+    if (ARG_TYPE_BYTES == obj_type) {
+        int index = arg_getInt(key);
+        uint8_t* bytes_val = arg_getBytes(val);
+        uint8_t* bytes_pyload = arg_getBytes(obj);
+        size_t bytes_len = arg_getBytesSize(obj);
+        bytes_pyload[index] = bytes_val[0];
+        obj_setBytes(self, obj_str, bytes_pyload, bytes_len);
+    }
     if (ARG_TYPE_OBJECT == obj_type) {
         PikaObj* arg_obj = arg_getPtr(obj);
         obj_setArg(arg_obj, "__key", key);

port/linux/test/pikaMain-test.cpp

@@ -833,6 +833,32 @@ TEST(pikaMain, str__index__) {
     EXPECT_EQ(pikaMemNow(), 0);
 }
 
+TEST(pikaMain, bytes__index__) {
+    /* init */
+    pikaMemInfo.heapUsedMax = 0;
+    PikaObj* pikaMain = newRootObj("pikaMain", New_PikaMain);
+    /* run */
+    __platform_printf("BEGIN\n");
+    obj_run(pikaMain,
+            "s = b'test'\n"
+            "res = s[2]\n"
+            "res2 = b'eqrt'[2]\n"
+            "s[2] = b'q'\n"
+            "\n");
+    /* collect */
+    uint8_t* res = obj_getBytes(pikaMain, "res");
+    uint8_t* res2 = obj_getBytes(pikaMain, "res2");
+    uint8_t* s = obj_getBytes(pikaMain, "s");
+    /* assert */
+    EXPECT_STREQ((char*)res, "s");
+    EXPECT_STREQ((char*)res2, "r");
+    EXPECT_EQ(s[2], 'q');
+    // EXPECT_STREQ(s, "teqt");
+    /* deinit */
+    obj_deinit(pikaMain);
+    EXPECT_EQ(pikaMemNow(), 0);
+}
+
 TEST(pikaMain, list_index) {
     /* init */
     pikaMemInfo.heapUsedMax = 0;