Initial commit

Author: Kartatz

src/python-kai.c

@@ -335,6 +335,196 @@ static PyObject* get_enum(const char* const name, const biguint_t value) {
 
 }
 
+static PyObject* get_segments(const struct M3U8Stream* const stream) {
+	
+	size_t index = 0;
+	size_t subindex = 0;
+	
+	int status = 0;
+	int err = M3U8ERR_SUCCESS;
+	
+	const struct M3U8BaseURI* const base_uri = m3u8playlist_geturi(&stream->playlist);
+	
+	char* resolved_uri = NULL;
+	
+	PyObject* key = NULL;
+	
+	PyObject* method = NULL;
+	PyObject* uri = NULL;
+	PyObject* iv = NULL;
+	PyObject* keyformat = NULL;
+	PyObject* keyformatversions = NULL;
+	PyObject* tag = NULL;
+	PyObject* duration = NULL;
+	PyObject* title = NULL;
+	PyObject* byterange = NULL;
+	
+	
+	
+	PyObject* version = NULL;
+	
+	PyObject* segments = PyList_New(0);
+	
+	PyObject* class = NULL;
+	PyObject* args = NULL;
+	
+	PyObject* module = NULL;
+	
+	module = PyImport_ImportModule("kai");
+    
+    if (module == NULL) {
+		return NULL;
+	}
+	
+	if (segments == NULL) {
+		return NULL;
+	}
+	
+	for (index = 0; index < stream->offset; index++) {
+		const struct M3U8StreamItem* const item = &stream->items[index];
+		
+		if (!(item->type == M3U8_STREAM_SEGMENT || item->type == M3U8_STREAM_MAP)) {
+			continue;
+		}
+		
+		switch (item->type) {
+			case M3U8_STREAM_SEGMENT: {
+				const struct M3U8Segment* const segment = ((struct M3U8Segment*) item->item);
+				
+				/* Key */
+				key = Py_None;
+				
+				if (segment->key.uri != NULL) {
+					/* Key -> Method */
+					method = get_enum("M3U8EncryptionMethod", segment->key.method);
+					
+					if (method == NULL) {
+						return NULL;
+					}
+					
+					/* Key -> URI */
+					free(resolved_uri);
+					
+					err = m3u8uri_resolve(base_uri, segment->key.uri, &resolved_uri);
+					
+					uri = get_string((err == M3U8ERR_SUCCESS) ? resolved_uri : segment->key.uri);
+					
+					if (uri == NULL) {
+						return NULL;
+					}
+					
+					/* Key -> IV */
+					iv = Py_None;
+					
+					if (segment->key.iv.offset > 0) {
+						iv = PyByteArray_FromStringAndSize(
+							segment->key.iv.data,
+							segment->key.iv.offset
+						);
+					}
+					
+					if (iv == NULL) {
+						return NULL;
+					}
+					
+					/* Key -> Key format */
+					keyformat = get_string(segment->key.keyformat);
+					
+					if (keyformat == NULL) {
+						return NULL;
+					}
+					
+					/* Key -> Key format versions */
+					keyformatversions = Py_None;
+					
+					if (segment->key.keyformatversions.offset > 0) {
+						keyformatversions = PyList_New(segment->key.keyformatversions.offset);
+						
+						if (keyformatversions == NULL) {
+							return NULL;
+						}
+						
+						for (subindex = 0; subindex < segment->key.keyformatversions.offset; subindex++) {
+							const int value = segment->key.keyformatversions.items[subindex];
+							
+							PyObject* item = PyBool_FromLong(value);
+							
+							if (item == NULL) {
+								return NULL;
+							}
+							
+							status = PyList_SetItem(keyformatversions, subindex, item);
+							
+							if (status != 0) {
+								return NULL;
+							}
+						}
+					}
+					
+					/* Key -> Tag */
+					tag = PyLong_FromLongLong((uintptr_t) segment->key.tag);
+					
+					if (tag == NULL) {
+						return NULL;
+					}
+					
+					class = PyObject_GetAttrString(module, "M3U8Key");
+					
+					if (class == NULL) {
+						return NULL;
+					}
+					
+					args = Py_BuildValue(
+						"(OOOOOO)", 
+						method,
+						uri,
+						iv,
+						keyformat,
+						keyformatversions,
+						tag
+					);
+					
+					if (args == NULL) {
+						return NULL;
+					}
+					
+					key = PyObject_Call(class, args, NULL);
+					
+					Py_DECREF(class);
+					Py_DECREF(args);
+				}
+				
+				if (key == NULL) {
+					return NULL;
+				}
+				
+				/* Duration */
+				duration = PyLong_FromLongLong(segment->duration);
+				
+				if (duration == NULL) {
+					return NULL;
+				}
+				
+				/* Title */
+				title = get_string(segment->title);
+				
+				if (title == NULL) {
+					return NULL;
+				}
+				
+				/* Byte range */
+				byterange = Py_None
+			}
+			case M3U8_STREAM_MAP: {
+				const struct M3U8Map* const map = ((struct M3U8Map*) item->item);
+			}
+			default:
+				break;
+		}
+	}
+	
+}
+
 static PyObject* topy(
 	const struct M3U8Stream* const root,
 	const struct M3U8StreamItem* const item