1 files changed, 162 insertions, 0 deletions
diff --git a/sources/pyside6/libpyside/pysideclassinfo.cpp b/sources/pyside6/libpyside/pysideclassinfo.cpp
new file mode 100644
index 000000000..9ab5a7ad0
--- /dev/null
+++ b/sources/pyside6/libpyside/pysideclassinfo.cpp
@@ -0,0 +1,162 @@
+// Copyright (C) 2016 The Qt Company Ltd.
+// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
+
+#include <sbkpython.h>
+
+#include "pysideclassinfo.h"
+#include "pyside_p.h"
+#include "pysideclassinfo_p.h"
+#include "dynamicqmetaobject.h"
+
+#include <shiboken.h>
+#include <signature.h>
+
+extern "C"
+{
+
+static PyTypeObject *createClassInfoType()
+{
+    auto typeSlots =
+        PySide::ClassDecorator::Methods<PySide::ClassInfo::ClassInfoPrivate>::typeSlots();
+
+    PyType_Spec PySideClassInfoType_spec = {
+        "2:PySide6.QtCore.ClassInfo",
+        sizeof(PySideClassDecorator),
+        0,
+        Py_TPFLAGS_DEFAULT,
+        typeSlots.data()};
+    return SbkType_FromSpec(&PySideClassInfoType_spec);
+};
+
+PyTypeObject *PySideClassInfo_TypeF(void)
+{
+    static auto *type = createClassInfoType();
+    return type;
+}
+
+}  // extern "C"
+
+namespace PySide::ClassInfo {
+
+const char *ClassInfoPrivate::name() const
+{
+    return "ClassInfo";
+}
+
+PyObject *ClassInfoPrivate::tp_call(PyObject *self, PyObject *args, PyObject * /* kw */)
+{
+    PyObject *klass = tp_call_check(args, CheckMode::QObjectType);
+    if (klass == nullptr)
+        return nullptr;
+
+    auto *pData = DecoratorPrivate::get<ClassInfoPrivate>(self);
+
+    if (pData->m_alreadyWrapped) {
+        PyErr_SetString(PyExc_TypeError, "This instance of ClassInfo() was already used to wrap an object");
+        return nullptr;
+    }
+
+    PyTypeObject *klassType = reinterpret_cast<PyTypeObject *>(klass);
+    if (!PySide::ClassInfo::setClassInfo(klassType, pData->m_data)) {
+        PyErr_SetString(PyExc_TypeError, "This decorator can only be used on classes that are subclasses of QObject");
+        return nullptr;
+    }
+
+    pData->m_alreadyWrapped = true;
+
+    Py_INCREF(klass);
+    return klass;
+}
+
+int ClassInfoPrivate::tp_init(PyObject *self, PyObject *args, PyObject *kwds)
+{
+    PyObject *infoDict = nullptr;
+    auto size = PyTuple_Size(args);
+    if (size == 1 && kwds == nullptr) {
+        PyObject *tmp = PyTuple_GET_ITEM(args, 0);
+        if (PyDict_Check(tmp))
+            infoDict = tmp;
+    } else if (size == 0 && kwds && PyDict_Check(kwds)) {
+        infoDict = kwds;
+    }
+
+    if (infoDict == nullptr) {
+        PyErr_Format(PyExc_TypeError, "ClassInfo() takes either keyword argument(s) or "
+                                      "a single dictionary argument");
+        return -1;
+    }
+
+    auto *pData = DecoratorPrivate::get<ClassInfoPrivate>(self);
+
+    PyObject *key{};
+    PyObject *value{};
+    Py_ssize_t pos = 0;
+
+    // PyDict_Next causes a segfault if kwds is empty
+    if (PyDict_Size(infoDict) > 0) {
+        while (PyDict_Next(infoDict, &pos, &key, &value)) {
+            if (Shiboken::String::check(key) && Shiboken::String::check(value)) {
+                ClassInfo info{Shiboken::String::toCString(key),
+                               Shiboken::String::toCString(value)};
+                pData->m_data.append(info);
+            } else {
+                PyErr_SetString(PyExc_TypeError, "All keys and values provided to ClassInfo() "
+                                                 "must be strings");
+                return -1;
+            }
+        }
+    }
+
+    return PyErr_Occurred() != nullptr ? -1 : 0;
+}
+
+static const char *ClassInfo_SignatureStrings[] = {
+    "PySide6.QtCore.ClassInfo(self,**info:typing.Dict[str,str])",
+    nullptr}; // Sentinel
+
+void init(PyObject *module)
+{
+    if (InitSignatureStrings(PySideClassInfo_TypeF(), ClassInfo_SignatureStrings) < 0)
+        return;
+
+    Py_INCREF(PySideClassInfo_TypeF());
+    PyModule_AddObject(module, "ClassInfo", reinterpret_cast<PyObject *>(PySideClassInfo_TypeF()));
+}
+
+bool checkType(PyObject *pyObj)
+{
+    return pyObj != nullptr
+        && PyType_IsSubtype(Py_TYPE(pyObj), PySideClassInfo_TypeF()) != 0;
+}
+
+ClassInfoList getClassInfoList(PyObject *decorator)
+{
+    auto *pData = PySide::ClassDecorator::DecoratorPrivate::get<ClassInfoPrivate>(decorator);
+    return pData->m_data;
+}
+
+bool setClassInfo(PyTypeObject *type, const QByteArray &key,
+                  const QByteArray &value)
+{
+    auto *userData = PySide::retrieveTypeUserData(type);
+    const bool result = userData != nullptr;
+    if (result) {
+        PySide::MetaObjectBuilder &mo = userData->mo;
+        mo.addInfo(key, value);
+    }
+    return result;
+}
+
+bool setClassInfo(PyTypeObject *type, const ClassInfoList &list)
+{
+    auto *userData = PySide::retrieveTypeUserData(type);
+    const bool result = userData != nullptr;
+    if (result) {
+        PySide::MetaObjectBuilder &mo = userData->mo;
+        for (const auto &info : list)
+            mo.addInfo(info.key.constData(), info.value.constData());
+    }
+    return result;
+}
+
+} //namespace PySide::ClassInfo