1 files changed, 56 insertions, 0 deletions
diff --git a/py/qstr.c b/py/qstr.c
new file mode 100644
index 0000000000..33d15c7e73
--- /dev/null
+++ b/py/qstr.c
@@ -0,0 +1,56 @@
+#include <assert.h>
+#include <string.h>
+
+#include "misc.h"
+
+static int qstrs_alloc;
+static int qstrs_len;
+static const char **qstrs;
+
+void qstr_init() {
+    qstrs_alloc = 400;
+    qstrs_len = 1;
+    qstrs = m_new(const char*, qstrs_alloc);
+    qstrs[0] = "nil";
+}
+
+static qstr qstr_add(const char *str) {
+    if (qstrs_len >= qstrs_alloc) {
+        qstrs_alloc *= 2;
+        qstrs = m_renew(const char*, qstrs, qstrs_alloc);
+    }
+    qstrs[qstrs_len++] = str;
+    return qstrs_len - 1;
+}
+
+qstr qstr_from_str_static(const char *str) {
+    for (int i = 0; i < qstrs_len; i++) {
+        if (strcmp(qstrs[i], str) == 0) {
+            return i;
+        }
+    }
+    return qstr_add(str);
+}
+
+qstr qstr_from_str_take(char *str) {
+    for (int i = 0; i < qstrs_len; i++) {
+        if (strcmp(qstrs[i], str) == 0) {
+            m_free(str);
+            return i;
+        }
+    }
+    return qstr_add(str);
+}
+
+qstr qstr_from_strn_copy(const char *str, int len) {
+    for (int i = 0; i < qstrs_len; i++) {
+        if (strncmp(qstrs[i], str, len) == 0 && qstrs[i][len] == '\0') {
+            return i;
+        }
+    }
+    return qstr_add(strndup(str, len));
+}
+
+const char *qstr_str(qstr qstr) {
+    return qstrs[qstr];
+}