aboutsummaryrefslogtreecommitdiff
path: root/lib/python/Plugins/SystemPlugins/Hotplug/plugin.py
blob: 882668f74c6dcd9e84925f07abba4d6665763836 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
from Plugins.Plugin import PluginDescriptor
from twisted.internet.protocol import Protocol, Factory
from twisted.internet import reactor
from Components.Harddisk import harddiskmanager

DEVICEDB =  \
	{ "/devices/pci0000:00/0000:00:14.2/usb1/1-1/1-1:1.0/host0/target0:0:0/0:0:0:0": "CF Slot",
	  "/devices/pci0000:00/0000:00:14.2/usb1/1-1/1-1:1.0/host0/target1:0:0/0:0:0:0": "SD Slot"
	}

class Hotplug(Protocol):
	def getUserfriendlyDeviceName(self, phys):
		return DEVICEDB.get(phys, "USB Storage")

	def connectionMade(self):
		self.received = ""

	def dataReceived(self, data):
		self.received += data

	def connectionLost(self, reason):
		data = self.received.split('\0')[:-1]

		print "hotplug:", data

		if len(data) < 4:
			return

		(action, device, physdev, driver) = data[:4]

		dev = device.split('/')[-1]

		if action == "add":
			print "Medium found in", self.getUserfriendlyDeviceName(dev)
			harddiskmanager.addHotplugPartition(dev, self.getUserfriendlyDeviceName(dev))
		elif action == "remove":
			harddiskmanager.removeHotplugPartition(dev)

def autostart(reason, **kwargs):
	if reason == 0:
		print "starting hotplug handler"
		factory = Factory()
		factory.protocol = Hotplug

		try:
			import os
			os.remove("/tmp/hotplug.socket")
		except OSError:
			pass

		reactor.listenUNIX("/tmp/hotplug.socket", factory)

def Plugins(**kwargs):
	return PluginDescriptor(name = "Hotplug", description = "listens to hotplug events", where = PluginDescriptor.WHERE_AUTOSTART, fnc = autostart)