6 from threading
import Timer
7 from urllib
import quote
12 from plugin
import GetPlugin
14 SHARE_TEMPLATE
= '/TiVoConnect?Command=QueryContainer&Container=%s'
15 PLATFORM_MAIN
= 'pyTivo'
16 PLATFORM_VIDEO
= 'pc/pyTivo' # For the nice icon
19 def __init__(self
, names
):
22 def removeService(self
, server
, type, name
):
23 self
.names
.remove(name
.replace('.' + type, ''))
25 def addService(self
, server
, type, name
):
26 self
.names
.append(name
.replace('.' + type, ''))
29 def __init__(self
, logger
):
30 """ Announce our shares via Zeroconf. """
34 self
.rz
= Zeroconf
.Zeroconf()
36 old_titles
= self
.scan()
37 address
= socket
.inet_aton(config
.get_ip())
38 port
= int(config
.getPort())
39 logger
.info('Announcing shares...')
40 for section
, settings
in config
.getShares():
42 ct
= GetPlugin(settings
['type']).CONTENT_TYPE
45 if ct
.startswith('x-container/'):
47 platform
= PLATFORM_VIDEO
49 platform
= PLATFORM_MAIN
50 logger
.info('Registering: %s' % section
)
51 self
.share_names
.append(section
)
52 desc
= {'path': SHARE_TEMPLATE
% quote(section
),
53 'platform': platform
, 'protocol': 'http'}
57 while title
in old_titles
:
59 title
= '%s [%d]' % (section
, count
)
60 self
.renamed
[section
] = title
61 info
= Zeroconf
.ServiceInfo('_%s._tcp.local.' % tt
,
62 '%s._%s._tcp.local.' % (title
, tt
),
63 address
, port
, 0, 0, desc
)
64 self
.rz
.registerService(info
)
65 self
.share_info
.append(info
)
68 """ Look for TiVos using Zeroconf. """
69 VIDS
= '_tivo-videos._tcp.local.'
72 self
.logger
.info('Scanning for TiVos...')
74 # Get the names of servers offering TiVo videos
75 browser
= Zeroconf
.ServiceBrowser(self
.rz
, VIDS
, ZCListener(names
))
77 # Give them a second to respond
80 # Now get the addresses -- this is the slow part
82 info
= self
.rz
.getServiceInfo(VIDS
, name
+ '.' + VIDS
)
83 if info
and 'TSN' in info
.properties
:
84 tsn
= info
.properties
['TSN']
85 address
= socket
.inet_ntoa(info
.getAddress())
87 config
.tivos
[tsn
] = {'name': name
, 'address': address
,
89 config
.tivos
[tsn
].update(info
.properties
)
90 self
.logger
.info(name
)
95 self
.logger
.info('Unregistering: %s' % ' '.join(self
.share_names
))
96 for info
in self
.share_info
:
97 self
.rz
.unregisterService(info
)
102 self
.UDPSock
= socket
.socket(socket
.AF_INET
, socket
.SOCK_DGRAM
)
103 self
.UDPSock
.setsockopt(socket
.SOL_SOCKET
, socket
.SO_BROADCAST
, 1)
106 self
.platform
= PLATFORM_VIDEO
107 for section
, settings
in config
.getShares():
109 ct
= GetPlugin(settings
['type']).CONTENT_TYPE
112 if ct
in ('x-container/tivo-music', 'x-container/tivo-photos'):
113 self
.platform
= PLATFORM_MAIN
117 logger
= logging
.getLogger('pyTivo.beacon')
119 self
.bd
= ZCBroadcast(logger
)
121 logger
.error('Zeroconf failure')
126 def add_service(self
, service
):
127 self
.services
.append(service
)
130 def format_services(self
):
131 return ';'.join(self
.services
)
133 def format_beacon(self
, conntype
, services
=True):
134 beacon
= ['tivoconnect=1',
135 'method=%s' % conntype
,
136 'identity={%s}' % config
.getGUID(),
137 'machine=%s' % socket
.gethostname(),
138 'platform=%s' % self
.platform
]
141 beacon
.append('services=' + self
.format_services())
143 beacon
.append('services=TiVoMediaServer:0/http')
145 return '\n'.join(beacon
) + '\n'
147 def send_beacon(self
):
148 beacon_ips
= config
.getBeaconAddresses()
149 beacon
= self
.format_beacon('broadcast')
150 for beacon_ip
in beacon_ips
.split():
151 if beacon_ip
!= 'listen':
155 result
= self
.UDPSock
.sendto(packet
, (beacon_ip
, 2190))
158 packet
= packet
[result
:]
164 self
.timer
= Timer(60, self
.start
)
172 def recv_bytes(self
, sock
, length
):
174 while len(block
) < length
:
175 add
= sock
.recv(length
- len(block
))
181 def recv_packet(self
, sock
):
182 length
= struct
.unpack('!I', self
.recv_bytes(sock
, 4))[0]
183 return self
.recv_bytes(sock
, length
)
185 def send_packet(self
, sock
, packet
):
186 sock
.sendall(struct
.pack('!I', len(packet
)) + packet
)
189 """ For the direct-connect, TCP-style beacon """
193 TCPSock
= socket
.socket(socket
.AF_INET
, socket
.SOCK_STREAM
)
194 TCPSock
.bind(('', 2190))
198 # Wait for a connection
199 client
, address
= TCPSock
.accept()
201 # Accept (and discard) the client's beacon
202 self
.recv_packet(client
)
205 self
.send_packet(client
, self
.format_beacon('connected'))
209 thread
.start_new_thread(server
, ())
211 def get_name(self
, address
):
212 """ Exchange beacons, and extract the machine name. """
213 our_beacon
= self
.format_beacon('connected', False)
214 machine_name
= re
.compile('machine=(.*)\n').search
217 tsock
= socket
.socket()
218 tsock
.connect((address
, 2190))
219 self
.send_packet(tsock
, our_beacon
)
220 tivo_beacon
= self
.recv_packet(tsock
)
222 name
= machine_name(tivo_beacon
).groups()[0]