eServiceCenter::getPrivInstance(sc);
if (sc)
- sc->addServiceFactory(eServiceFactoryFS::id, this);
+ {
+ std::list<std::string> extensions;
+ sc->addServiceFactory(eServiceFactoryFS::id, this, extensions);
+ }
m_service_information = new eStaticServiceFSInformation();
}
RESULT eServiceFactoryFS::list(const eServiceReference &ref, ePtr<iListableService> &ptr)
{
- ptr = new eServiceFS(ref.path.c_str());
+ ptr = new eServiceFS(ref.path.c_str(), ref.getName().length() ? ref.getName().c_str() : 0);
return 0;
}
DEFINE_REF(eServiceFS);
-eServiceFS::eServiceFS(const char *path): path(path)
+eServiceFS::eServiceFS(const char *path, const char *additional_extensions): path(path)
{
m_list_valid = 0;
+ if (additional_extensions)
+ {
+ size_t slen=strlen(additional_extensions);
+ char buf[slen+1];
+ char *tmp=0, *cmds = buf;
+ memcpy(buf, additional_extensions, slen+1);
+
+ // strip spaces at beginning
+ while(cmds[0] == ' ')
+ {
+ ++cmds;
+ --slen;
+ }
+
+ // strip spaces at the end
+ while(slen && cmds[slen-1] == ' ')
+ {
+ cmds[slen-1] = 0;
+ --slen;
+ }
+
+ if (slen)
+ {
+ if (*cmds)
+ {
+ int id;
+ char buf2[16];
+ while(1)
+ {
+ tmp = strchr(cmds, ' ');
+ if (tmp)
+ *tmp = 0;
+ if (strstr(cmds, "0x"))
+ {
+ if (sscanf(cmds, "0x%x:%s", &id, buf2) == 2)
+ m_additional_extensions[id].push_back(buf2);
+ else
+ eDebug("parse additional_extension (%s) failed", cmds);
+ }
+ else
+ {
+ if (sscanf(cmds, "%d:%s", &id, buf2) == 2)
+ m_additional_extensions[id].push_back(buf2);
+ else
+ eDebug("parse additional_extension (%s) failed", cmds);
+ }
+ if (!tmp)
+ break;
+ cmds = tmp+1;
+ while (*cmds && *cmds == ' ')
+ ++cmds;
+ }
+ }
+ }
+ }
}
eServiceFS::~eServiceFS()
{
}
+int lower(char c)
+{
+ return std::tolower(static_cast<unsigned char>(c));
+}
+
RESULT eServiceFS::getContent(std::list<eServiceReference> &list, bool sorted)
{
DIR *d=opendir(path.c_str());
} else
{
size_t e = filename.rfind('.');
- std::string extension = (e != std::string::npos) ? filename.substr(e) : "";
- int type = -1;
-
- if (extension == ".ts")
- type = eServiceFactoryDVB::id;
- else if (extension == ".mp3")
- type = 4097;
- else if (extension == ".ogg")
- type = 4097;
- else if (extension == ".mpg")
- type = 4097;
- else if (extension == ".vob")
- type = 4097;
- else if (extension == ".wav" || extension == ".wave")
- type = 4097;
-
- if (type != -1)
+ if (e != std::string::npos && e+1 < filename.length())
{
- eServiceReference service(type,
- 0,
- filename);
- service.data[0] = 0;
- list.push_back(service);
+ std::string extension = filename.substr(e+1);
+ std::transform(extension.begin(), extension.end(), extension.begin(), lower);
+ int type = getServiceTypeForExtension(extension);
+
+ if (type == -1)
+ {
+ ePtr<eServiceCenter> sc;
+ eServiceCenter::getPrivInstance(sc);
+ type = sc->getServiceTypeForExtension(extension);
+ }
+
+ if (type != -1)
+ {
+ eServiceReference service(type,
+ 0,
+ filename);
+ service.data[0] = 0;
+ list.push_back(service);
+ }
+ else
+ eDebug("unhandled extension %s", extension.c_str());
}
}
}
return 0;
}
-RESULT eServiceFS::getContent(PyObject *list, bool sorted)
+// The first argument of this function is a format string to specify the order and
+// the content of the returned list
+// useable format options are
+// R = Service Reference (as swig object .. this is very slow)
+// S = Service Reference (as python string object .. same as ref.toString())
+// C = Service Reference (as python string object .. same as ref.toCompareString())
+// N = Service Name (as python string object)
+// when exactly one return value per service is selected in the format string,
+// then each value is directly a list entry
+// when more than one value is returned per service, then the list is a list of
+// python tuples
+// unknown format string chars are returned as python None values !
+PyObject *eServiceFS::getContent(const char* format, bool sorted)
{
- if (!list || !PyList_Check(list))
- return -1;
-
+ ePyObject ret;
std::list<eServiceReference> tmplist;
+ int retcount=1;
- getContent(tmplist, sorted);
+ if (!format || !(retcount=strlen(format)))
+ format = "R"; // just return service reference swig object ...
- if (sorted)
- tmplist.sort(iListableServiceCompare(this));
-
- for (std::list<eServiceReference>::iterator it(tmplist.begin());
- it != tmplist.end(); ++it)
+ if (!getContent(tmplist, sorted))
{
- PyObject *refobj = New_eServiceReference(*it);
- PyList_Append(list, refobj);
- Py_DECREF(refobj);
- }
+ int services=tmplist.size();
+ ePtr<iStaticServiceInformation> sptr;
+ eServiceCenterPtr service_center;
- return 0;
+ if (strchr(format, 'N'))
+ eServiceCenter::getPrivInstance(service_center);
+
+ ret = PyList_New(services);
+ std::list<eServiceReference>::iterator it(tmplist.begin());
+
+ for (int cnt=0; cnt < services; ++cnt)
+ {
+ eServiceReference &ref=*it++;
+ ePyObject tuple = retcount > 1 ? PyTuple_New(retcount) : ePyObject();
+ for (int i=0; i < retcount; ++i)
+ {
+ ePyObject tmp;
+ switch(format[i])
+ {
+ case 'R': // service reference (swig)object
+ tmp = NEW_eServiceReference(ref);
+ break;
+ case 'C': // service reference compare string
+ tmp = PyString_FromString(ref.toCompareString().c_str());
+ break;
+ case 'S': // service reference string
+ tmp = PyString_FromString(ref.toString().c_str());
+ break;
+ case 'N': // service name
+ if (service_center)
+ {
+ service_center->info(ref, sptr);
+ if (sptr)
+ {
+ std::string name;
+ sptr->getName(ref, name);
+ if (name.length())
+ tmp = PyString_FromString(name.c_str());
+ }
+ }
+ if (!tmp)
+ tmp = PyString_FromString("<n/a>");
+ break;
+ default:
+ if (tuple)
+ {
+ tmp = Py_None;
+ Py_INCREF(Py_None);
+ }
+ break;
+ }
+ if (tmp)
+ {
+ if (tuple)
+ PyTuple_SET_ITEM(tuple, i, tmp);
+ else
+ PyList_SET_ITEM(ret, cnt, tmp);
+ }
+ }
+ if (tuple)
+ PyList_SET_ITEM(ret, cnt, tuple);
+ }
+ }
+ return ret ? (PyObject*)ret : (PyObject*)PyList_New(0);
}
RESULT eServiceFS::getNext(eServiceReference &ptr)
return -1;
}
+int eServiceFS::getServiceTypeForExtension(const char *str)
+{
+ for (std::map<int, std::list<std::string> >::iterator sit(m_additional_extensions.begin()); sit != m_additional_extensions.end(); ++sit)
+ {
+ for (std::list<std::string>::iterator eit(sit->second.begin()); eit != sit->second.end(); ++eit)
+ {
+ if (*eit == str)
+ return sit->first;
+ }
+ }
+ return -1;
+}
+
+int eServiceFS::getServiceTypeForExtension(const std::string &str)
+{
+ return getServiceTypeForExtension(str.c_str());
+}
+
eAutoInitPtr<eServiceFactoryFS> init_eServiceFactoryFS(eAutoInitNumbers::service+1, "eServiceFactoryFS");
git.cweiske.de / enigma2.git / blobdiff