1 files changed, 68 insertions, 0 deletions
diff --git a/src/python/root.cpp b/src/python/root.cpp
new file mode 100644
index 0000000..d3ac021
--- /dev/null
+++ b/src/python/root.cpp
@@ -0,0 +1,68 @@
+/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+/*
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+#include "root.hpp"
+#include "document.hpp"
+#include "global.hpp"
+
+#include "orcus/format_detection.hpp"
+#include "orcus/info.hpp"
+
+#include <iostream>
+#include <sstream>
+
+#include <object.h>
+
+using namespace std;
+
+namespace orcus { namespace python {
+
+PyObject* detect_format(PyObject* /*module*/, PyObject* args, PyObject* kwargs)
+{
+    py_unique_ptr stream = read_stream_from_args(args, kwargs);
+    if (!stream)
+        return nullptr;
+
+    char* p = nullptr;
+    Py_ssize_t n = 0;
+    if (PyBytes_AsStringAndSize(stream.get(), &p, &n) < 0)
+        return nullptr;
+
+    try
+    {
+        format_t ft = orcus::detect({p, (size_t)n});
+
+        switch (ft)
+        {
+            case format_t::ods:
+                return get_python_enum_value("FormatType", "ODS");
+            case format_t::xlsx:
+                return get_python_enum_value("FormatType", "XLSX");
+            case format_t::gnumeric:
+                return get_python_enum_value("FormatType", "GNUMERIC");
+            case format_t::xls_xml:
+                return get_python_enum_value("FormatType", "XLS_XML");
+            case format_t::parquet:
+                return get_python_enum_value("FormatType", "PARQUET");
+            case format_t::csv:
+                return get_python_enum_value("FormatType", "CSV");
+            case format_t::unknown:
+            default:
+                return get_python_enum_value("FormatType", "UNKNOWN");
+        }
+    }
+    catch (const std::exception&)
+    {
+        PyErr_SetString(PyExc_ValueError, "failed to perform deep detection on this file.");
+        return nullptr;
+    }
+}
+
+}}
+
+/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
+