Justabot est un Bot écrit en python, il permet une présence sur les salons Jabber. https://www.devosi.org
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

284 lines
7.3KB

  1. # -*- coding: utf-8 -*-
  2. # Copyright Crestetto Kévin, (Novembre 2012)
  3. # checkpoint [at] singularity (dot) fr
  4. # Ce logiciel est un programme informatique
  5. # servant à discuter et executer des commandes sur un salon jabber/XMPP.
  6. # Ce logiciel est régi par la licence CeCILL soumise au droit français et
  7. # respectant les principes de diffusion des logiciels libres. Vous pouvez
  8. # utiliser, modifier et/ou redistribuer ce programme sous les conditions
  9. # de la licence CeCILL telle que diffusée par le CEA, le CNRS et l'INRIA
  10. # sur le site "http://www.cecill.info".
  11. # En contrepartie de l'accessibilité au code source et des droits de copie,
  12. # de modification et de redistribution accordés par cette licence, il n'est
  13. # offert aux utilisateurs qu'une garantie limitée. Pour les mêmes raisons,
  14. # seule une responsabilité restreinte pèse sur l'auteur du programme, le
  15. # titulaire des droits patrimoniaux et les concédants successifs.
  16. # A cet égard l'attention de l'utilisateur est attirée sur les risques
  17. # associés au chargement, à l'utilisation, à la modification et/ou au
  18. # développement et à la reproduction du logiciel par l'utilisateur étant
  19. # donné sa spécificité de logiciel libre, qui peut le rendre complexe à
  20. # manipuler et qui le réserve donc à des développeurs et des professionnels
  21. # avertis possédant des connaissances informatiques approfondies. Les
  22. # utilisateurs sont donc invités à charger et tester l'adéquation du
  23. # logiciel à leurs besoins dans des conditions permettant d'assurer la
  24. # sécurité de leurs systèmes et ou de leurs données et, plus généralement,
  25. # à l'utiliser et l'exploiter dans les mêmes conditions de sécurité.
  26. # Le fait que vous puissiez accéder à cet en-tête signifie que vous avez
  27. # pris connaissance de la licence CeCILL, et que vous en avez accepté les
  28. # termes.
  29. from random import randrange
  30. from os.path import isfile
  31. import mimetypes
  32. import base64
  33. import sleekxmpp as xmpp
  34. import sleekxmpp.jid as streamjid
  35. import sleekxmpp.basexmpp as basexmpp
  36. import xml.etree.cElementTree as libxml
  37. import hashlib
  38. def hashname(name):
  39. return hashlib.sha256(
  40. name.encode('utf-8', 'replace')).hexdigest()
  41. class Avatar:
  42. """manage avatar"""
  43. def card(self, it, i, mime):
  44. """create and send the Vcard"""
  45. base = basexmpp.BaseXMPP()
  46. ET = """
  47. <iq from='%s'
  48. type='set'
  49. id='vc1'>
  50. <vCard xmlns='vcard-temp'>
  51. <N>
  52. <GIVEN>Just</GIVEN>
  53. <FAMILY>Bot</FAMILY>
  54. <MIDDLE>A</MIDDLE>
  55. </N>
  56. <NICKNAME>justabot</NICKNAME>
  57. <URL>http://www.librecase.eu/media/justabot</URL>
  58. <DESC>
  59. Just a bot.
  60. </DESC>
  61. <BDAY>2012-05-13</BDAY>
  62. <TITLE>license CeCILLv2</TITLE>
  63. <ROLE>None given</ROLE>
  64. <PHOTO>
  65. <TYPE>%s</TYPE>
  66. <BINVAL>
  67. %s
  68. </BINVAL>
  69. </PHOTO>
  70. <ADR>
  71. <WORK/>
  72. <LOCALITY>Python3</LOCALITY>
  73. <STREET>SleekXMPP</STREET>
  74. </ADR>
  75. </vCard>
  76. </iq>
  77. """ % (it.cl.boundjid.bare, mime, i)
  78. it.cl.send(ET)
  79. def base64_img(self, img):
  80. """ base64 encode"""
  81. return str(base64.b64encode(img), 'utf-8')
  82. def found_img(self, it):
  83. """return a random line of file"""
  84. cfg = it.DIRS['cfg']
  85. img = open(cfg + 'var/avatar.txt', 'r').readlines()
  86. if len(img) < 1:
  87. return False
  88. ranimg = img[randrange(len(img))][:-1]
  89. if isfile(ranimg):
  90. return ranimg
  91. return False
  92. def get_img(self, i):
  93. """read a binary file"""
  94. return open(i, 'rb').read()
  95. def get_mime(self, i):
  96. """return mime if pict in the list, or False"""
  97. EXTS = ('.jpg', '.gif', '.png', '.jpeg')
  98. mtype = mimetypes.guess_type(i)[0]
  99. if mtype in ('image/jpeg', 'image/gif', 'image/png'):
  100. for x in EXTS:
  101. if i.endswith(x):
  102. return mtype
  103. return False
  104. def sendimg(self, it):
  105. """take an pict in pre-defined list and use it at avatar"""
  106. i = self.found_img(it)
  107. if i != False:
  108. mime = self.get_mime(i)
  109. if mime:
  110. self.card(it, self.base64_img(self.get_img(i)), mime)
  111. def Kickit(n, to, mess, it, visitor=False):
  112. """ kick user, with nick n """
  113. role = 'none'
  114. if visitor:
  115. role = 'visitor'
  116. ET = """
  117. <iq from='%s'
  118. to='%s'
  119. type='set'
  120. id='kick1'>
  121. <query xmlns="http://jabber.org/protocol/muc#admin">
  122. <item role='%s'
  123. nick='%s'>
  124. <reason>
  125. %s
  126. </reason>
  127. </item>
  128. </query>
  129. </iq>
  130. """ % (it.cl.boundjid.bare, to, role, n, mess)
  131. it.cl.send(ET)
  132. def Banit(n, to, mess, it, jid):
  133. """ ban user with jid """
  134. if jid == '':
  135. return
  136. ET = """
  137. <iq from='%s'
  138. to='%s'
  139. type='set'
  140. id='ban1'>
  141. <query xmlns="http://jabber.org/protocol/muc#admin">
  142. <item affiliation='outcast' jid='%s'>
  143. <reason>
  144. %s
  145. </reason>
  146. </item>
  147. </query>
  148. </iq> """ % (it.cl.boundjid.bare, to, jid, mess)
  149. it.cl.send(ET)
  150. class ManagePres:
  151. """ manage user's presence and my role on a room """
  152. def __init__(self):
  153. self.mypres = ('unavailable', 'none')
  154. self.connected = {}
  155. def achange(self, nick, event):
  156. pass
  157. def presence(self, pres, user):
  158. """ save my presence and role """
  159. self.mypres = (pres, user)
  160. def setPres(self, pres, nick, aff):
  161. """ set a user presence """
  162. if pres['muc']['jid'] and pres['muc']['jid'] != '':
  163. self.connected[nick] = {'rname': pres['muc']['jid'].bare,
  164. 'aff': aff}
  165. else:
  166. self.connected[nick] = {'rname': pres['from'], 'aff': aff}
  167. self.achange(nick, 'in')
  168. def getJID(self, it):
  169. """ ask all jid and nick of the room if admin """
  170. count = 0
  171. for role in ['moderator', 'participant', 'visitor']:
  172. iq = it.make_iq_get(ifrom=it.boundjid.bare, ito=self.name)
  173. xml = libxml.fromstring("<query xmlns='http://jabber.org/protocol/muc#admin'><item role='" + role + "'/></query>\n")
  174. iq.set_query('http://jabber.org/protocol/muc#admin')
  175. iq.set_payload(xml)
  176. iq.send(callback=self.cb_getjids)
  177. count += 1
  178. def cb_getjids(self, event):
  179. q = event.get_payload()
  180. for r in q:
  181. for item in r:
  182. nick = None
  183. jid = None
  184. aff = None
  185. for k in item.keys():
  186. if k == 'nick':
  187. nick = item.get(k)
  188. if k == 'jid':
  189. jid = item.get(k)
  190. if k == 'affiliation':
  191. aff = item.get(k)
  192. if nick and jid:
  193. self.connected[nick] = {'rname': streamjid.JID(jid).bare,
  194. 'aff': aff}
  195. def chMyPres(self, user, pres, it):
  196. """ get and apply my role and aff """
  197. aff = pres['muc']['affiliation']
  198. role = pres['muc']['role']
  199. nick = user
  200. if role == 'moderator':
  201. self.presence(aff, role)
  202. self.getJID(it)
  203. self.presence(aff, role)
  204. if role == 'none':
  205. self.it.rejoin(pres['muc']['room'])
  206. return
  207. def removeNick(self, nick):
  208. """ remove a nick from list """
  209. try:
  210. del self.connected[nick]
  211. self.achange(nick, 'out')
  212. except:
  213. pass