move the otk swig module into its own lib/directory
[mikachu/openbox.git] / src / python.cc
1 // -*- mode: C++; indent-tabs-mode: nil; c-basic-offset: 2; -*-
2
3 #include "python.hh"
4 #include "openbox.hh"
5 #include "actions.hh"
6 #include "python.hh"
7 #include "bindings.hh"
8 #include "otk/display.hh"
9 #include "otk/util.hh"
10
11 extern "C" {
12 // The initializer in openbox_wrap.cc / otk_wrap.cc
13 extern void init_ob(void);
14 extern void init_otk(void);
15 }
16
17 namespace ob {
18
19 static PyObject *obdict = NULL;
20
21 void python_init(char *argv0)
22 {
23   // start the python engine
24   Py_SetProgramName(argv0);
25   Py_Initialize();
26   // initialize the C python module
27   init_otk();
28   init_ob();
29   // prepend the openbox directories for python scripts to the sys path
30   PyRun_SimpleString("import sys");
31   PyRun_SimpleString("sys.path.insert(0, '" SCRIPTDIR "')");
32   PyRun_SimpleString(const_cast<char*>(("sys.path.insert(0, '" +
33                                         otk::expandTilde("~/.openbox/python") +
34                                         "')").c_str()));
35   PyRun_SimpleString("import ob; import otk; import config;");
36   // set up convenience global variables
37   PyRun_SimpleString("ob.openbox = ob.Openbox_instance()");
38   PyRun_SimpleString("otk.display = otk.Display_instance()");
39
40   // set up access to the python global variables
41   PyObject *obmodule = PyImport_AddModule("config");
42   obdict = PyModule_GetDict(obmodule);
43 }
44
45 void python_destroy()
46 {
47   Py_Finalize();
48 }
49
50 bool python_exec(const std::string &path)
51 {
52   FILE *rcpyfd = fopen(path.c_str(), "r");
53   if (!rcpyfd) {
54     printf("Failed to load python file %s\n", path.c_str());
55     return false;
56   }
57   PyRun_SimpleFile(rcpyfd, const_cast<char*>(path.c_str()));
58   fclose(rcpyfd);
59   return true;
60 }
61
62 bool python_get_long(const char *name, long *value)
63 {
64   PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
65   if (!(val && PyInt_Check(val))) return false;
66   
67   *value = PyInt_AsLong(val);
68   return true;
69 }
70
71 bool python_get_string(const char *name, otk::ustring *value)
72 {
73   PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
74   if (!(val && PyString_Check(val))) return false;
75   
76   *value = PyString_AsString(val);
77   return true;
78 }
79
80 bool python_get_stringlist(const char *name, std::vector<otk::ustring> *value)
81 {
82   PyObject *val = PyDict_GetItemString(obdict, const_cast<char*>(name));
83   if (!(val && PyList_Check(val))) return false;
84
85   for (int i = 0, end = PyList_Size(val); i < end; ++i) {
86     PyObject *str = PyList_GetItem(val, i);
87     if (PyString_Check(str))
88       value->push_back(PyString_AsString(str));
89   }
90   return true;
91 }
92
93 // ************************************* //
94 // Stuff for calling from Python scripts //
95 // ************************************* //
96
97 PyObject *mbind(const std::string &button, ob::MouseContext::MC context,
98                 ob::MouseAction::MA action, PyObject *func)
99 {
100   if (!PyCallable_Check(func)) {
101     PyErr_SetString(PyExc_TypeError, "Invalid callback function.");
102     return NULL;
103   }
104   if(context < 0 || context >= MouseContext::NUM_MOUSE_CONTEXT) {
105     PyErr_SetString(PyExc_ValueError, "Invalid MouseContext");
106     return NULL;
107   }
108   if(action < 0 || action >= MouseAction::NUM_MOUSE_ACTION) {
109     PyErr_SetString(PyExc_ValueError, "Invalid MouseAction");
110     return NULL;
111   }
112   
113   if (!ob::openbox->bindings()->addButton(button, context,
114                                           action, func)) {
115     PyErr_SetString(PyExc_RuntimeError,"Unable to add binding.");
116     return NULL;
117   }
118   Py_INCREF(Py_None); return Py_None;
119 }
120
121 PyObject *ebind(ob::EventAction::EA action, PyObject *func)
122 {
123   if (!PyCallable_Check(func)) {
124     PyErr_SetString(PyExc_TypeError, "Invalid callback function.");
125     return NULL;
126   }
127   
128   if (!ob::openbox->bindings()->addEvent(action, func)) {
129     PyErr_SetString(PyExc_RuntimeError,"Unable to add binding.");
130     return NULL;
131   }
132   Py_INCREF(Py_None); return Py_None;
133 }
134
135 PyObject *kgrab(int screen, PyObject *func)
136 {
137   if (!PyCallable_Check(func)) {
138     PyErr_SetString(PyExc_TypeError, "Invalid callback function.");
139     return NULL;
140   }
141
142   if (!ob::openbox->bindings()->grabKeyboard(screen, func)) {
143     PyErr_SetString(PyExc_RuntimeError,"Unable to grab keybaord.");
144     return NULL;
145   }
146   Py_INCREF(Py_None); return Py_None;
147 }
148
149 PyObject *kungrab()
150 {
151   ob::openbox->bindings()->ungrabKeyboard();
152   Py_INCREF(Py_None); return Py_None;
153 }
154
155 PyObject *mgrab(int screen)
156 {
157   if (!ob::openbox->bindings()->grabPointer(screen)) {
158     PyErr_SetString(PyExc_RuntimeError,"Unable to grab pointer.");
159     return NULL;
160   }
161   Py_INCREF(Py_None); return Py_None;
162 }
163
164 PyObject *mungrab()
165 {
166   ob::openbox->bindings()->ungrabPointer();
167   Py_INCREF(Py_None); return Py_None;
168 }
169
170 PyObject *kbind(PyObject *keylist, ob::KeyContext::KC context, PyObject *func)
171 {
172   if (!PyCallable_Check(func)) {
173     PyErr_SetString(PyExc_TypeError, "Invalid callback function.");
174     return NULL;
175   }
176   if (!PyList_Check(keylist)) {
177     PyErr_SetString(PyExc_TypeError, "Invalid keylist. Not a list.");
178     return NULL;
179   }
180
181   ob::Bindings::StringVect vectkeylist;
182   for (int i = 0, end = PyList_Size(keylist); i < end; ++i) {
183     PyObject *str = PyList_GetItem(keylist, i);
184     if (!PyString_Check(str)) {
185       PyErr_SetString(PyExc_TypeError,
186                      "Invalid keylist. It must contain only strings.");
187       return NULL;
188     }
189     vectkeylist.push_back(PyString_AsString(str));
190   }
191
192   (void)context; // XXX use this sometime!
193   if (!ob::openbox->bindings()->addKey(vectkeylist, func)) {
194     PyErr_SetString(PyExc_RuntimeError,"Unable to add binding.");
195     return NULL;
196   }
197   Py_INCREF(Py_None); return Py_None;
198 }
199
200 /*
201 PyObject *kunbind(PyObject *keylist, PyObject *func)
202 {
203   if (!PyList_Check(keylist)) {
204     PyErr_SetString(PyExc_TypeError, "Invalid keylist. Not a list.");
205     return NULL;
206   }
207   if (!PyCallable_Check(func)) {
208     PyErr_SetString(PyExc_TypeError, "Invalid callback function.");
209     return NULL;
210   }
211   
212   ob::Bindings::StringVect vectkeylist;
213   for (int i = 0, end = PyList_Size(keylist); i < end; ++i) {
214     PyObject *str = PyList_GetItem(keylist, i);
215     if (!PyString_Check(str)) {
216       PyErr_SetString(PyExc_TypeError,
217                      "Invalid keylist. It must contain only strings.");
218       return NULL;
219     }
220     vectkeylist.push_back(PyString_AsString(str));
221   }
222
223   if (!ob::openbox->bindings()->removeKey(vectkeylist, func)) {
224       PyErr_SetString(PyExc_RuntimeError, "Could not remove callback.");
225       return NULL;
226   }
227   Py_INCREF(Py_None); return Py_None;
228 }
229 */
230
231 void kunbind_all()
232 {
233   ob::openbox->bindings()->removeAllKeys();
234 }
235
236 void set_reset_key(const std::string &key)
237 {
238   ob::openbox->bindings()->setResetKey(key);
239 }
240
241 PyObject *send_client_msg(Window target, Atom type, Window about,
242                           long data, long data1, long data2,
243                           long data3, long data4)
244 {
245   XEvent e;
246   e.xclient.type = ClientMessage;
247   e.xclient.format = 32;
248   e.xclient.message_type = type;
249   e.xclient.window = about;
250   e.xclient.data.l[0] = data;
251   e.xclient.data.l[1] = data1;
252   e.xclient.data.l[2] = data2;
253   e.xclient.data.l[3] = data3;
254   e.xclient.data.l[4] = data4;
255
256   XSendEvent(**otk::display, target, false,
257              SubstructureRedirectMask | SubstructureNotifyMask,
258              &e);
259   Py_INCREF(Py_None); return Py_None;
260 }
261
262 void execute(const std::string &bin, int screen)
263 {
264   if (screen >= ScreenCount(**otk::display))
265     screen = 0;
266   otk::bexec(bin, otk::display->screenInfo(screen)->displayString());
267 }
268
269 }