oc_secret.py 56 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729
  1. #!/usr/bin/env python
  2. # pylint: disable=missing-docstring
  3. # flake8: noqa: T001
  4. # ___ ___ _ _ ___ ___ _ _____ ___ ___
  5. # / __| __| \| | __| _ \ /_\_ _| __| \
  6. # | (_ | _|| .` | _|| / / _ \| | | _|| |) |
  7. # \___|___|_|\_|___|_|_\/_/_\_\_|_|___|___/_ _____
  8. # | \ / _ \ | \| |/ _ \_ _| | __| \_ _|_ _|
  9. # | |) | (_) | | .` | (_) || | | _|| |) | | | |
  10. # |___/ \___/ |_|\_|\___/ |_| |___|___/___| |_|
  11. #
  12. # Copyright 2016 Red Hat, Inc. and/or its affiliates
  13. # and other contributors as indicated by the @author tags.
  14. #
  15. # Licensed under the Apache License, Version 2.0 (the "License");
  16. # you may not use this file except in compliance with the License.
  17. # You may obtain a copy of the License at
  18. #
  19. # http://www.apache.org/licenses/LICENSE-2.0
  20. #
  21. # Unless required by applicable law or agreed to in writing, software
  22. # distributed under the License is distributed on an "AS IS" BASIS,
  23. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  24. # See the License for the specific language governing permissions and
  25. # limitations under the License.
  26. #
  27. # -*- -*- -*- Begin included fragment: lib/import.py -*- -*- -*-
  28. '''
  29. OpenShiftCLI class that wraps the oc commands in a subprocess
  30. '''
  31. # pylint: disable=too-many-lines
  32. from __future__ import print_function
  33. import atexit
  34. import copy
  35. import json
  36. import os
  37. import re
  38. import shutil
  39. import subprocess
  40. import tempfile
  41. # pylint: disable=import-error
  42. try:
  43. import ruamel.yaml as yaml
  44. except ImportError:
  45. import yaml
  46. from ansible.module_utils.basic import AnsibleModule
  47. # -*- -*- -*- End included fragment: lib/import.py -*- -*- -*-
  48. # -*- -*- -*- Begin included fragment: doc/secret -*- -*- -*-
  49. DOCUMENTATION = '''
  50. ---
  51. module: oc_secret
  52. short_description: Module to manage openshift secrets
  53. description:
  54. - Manage openshift secrets programmatically.
  55. options:
  56. state:
  57. description:
  58. - If present, the secret will be created if it doesn't exist or updated if different. If absent, the secret will be removed if present. If list, information about the secret will be gathered and returned as part of the Ansible call results.
  59. required: false
  60. default: present
  61. choices: ["present", "absent", "list"]
  62. aliases: []
  63. kubeconfig:
  64. description:
  65. - The path for the kubeconfig file to use for authentication
  66. required: false
  67. default: /etc/origin/master/admin.kubeconfig
  68. aliases: []
  69. debug:
  70. description:
  71. - Turn on debug output.
  72. required: false
  73. default: False
  74. aliases: []
  75. name:
  76. description:
  77. - Name of the object that is being queried.
  78. required: false
  79. default: None
  80. aliases: []
  81. namespace:
  82. description:
  83. - The namespace where the object lives.
  84. required: false
  85. default: default
  86. aliases: []
  87. files:
  88. description:
  89. - A list of files provided for secrets
  90. required: false
  91. default: None
  92. aliases: []
  93. delete_after:
  94. description:
  95. - Whether or not to delete the files after processing them.
  96. required: false
  97. default: false
  98. aliases: []
  99. contents:
  100. description:
  101. - Content of the secrets
  102. required: false
  103. default: None
  104. aliases: []
  105. force:
  106. description:
  107. - Whether or not to force the operation
  108. required: false
  109. default: false
  110. aliases: []
  111. decode:
  112. description:
  113. - base64 decode the object
  114. required: false
  115. default: false
  116. aliases: []
  117. author:
  118. - "Kenny Woodson <kwoodson@redhat.com>"
  119. extends_documentation_fragment: []
  120. '''
  121. EXAMPLES = '''
  122. - name: create secret
  123. oc_secret:
  124. state: present
  125. namespace: openshift-infra
  126. name: metrics-deployer
  127. files:
  128. - name: nothing
  129. path: /dev/null
  130. register: secretout
  131. run_once: true
  132. - name: get ca from hawkular
  133. oc_secret:
  134. state: list
  135. namespace: openshift-infra
  136. name: hawkular-metrics-certificate
  137. decode: True
  138. register: hawkout
  139. run_once: true
  140. - name: Create secrets
  141. oc_secret:
  142. namespace: mynamespace
  143. name: mysecrets
  144. contents:
  145. - path: data.yml
  146. data: "{{ data_content }}"
  147. - path: auth-keys
  148. data: "{{ auth_keys_content }}"
  149. - path: configdata.yml
  150. data: "{{ configdata_content }}"
  151. - path: cert.crt
  152. data: "{{ cert_content }}"
  153. - path: key.pem
  154. data: "{{ osso_site_key_content }}"
  155. - path: ca.cert.pem
  156. data: "{{ ca_cert_content }}"
  157. register: secretout
  158. '''
  159. # -*- -*- -*- End included fragment: doc/secret -*- -*- -*-
  160. # -*- -*- -*- Begin included fragment: ../../lib_utils/src/class/yedit.py -*- -*- -*-
  161. # pylint: disable=undefined-variable,missing-docstring
  162. # noqa: E301,E302
  163. class YeditException(Exception):
  164. ''' Exception class for Yedit '''
  165. pass
  166. # pylint: disable=too-many-public-methods
  167. class Yedit(object):
  168. ''' Class to modify yaml files '''
  169. re_valid_key = r"(((\[-?\d+\])|([0-9a-zA-Z%s/_-]+)).?)+$"
  170. re_key = r"(?:\[(-?\d+)\])|([0-9a-zA-Z%s/_-]+)"
  171. com_sep = set(['.', '#', '|', ':'])
  172. # pylint: disable=too-many-arguments
  173. def __init__(self,
  174. filename=None,
  175. content=None,
  176. content_type='yaml',
  177. separator='.',
  178. backup=False):
  179. self.content = content
  180. self._separator = separator
  181. self.filename = filename
  182. self.__yaml_dict = content
  183. self.content_type = content_type
  184. self.backup = backup
  185. self.load(content_type=self.content_type)
  186. if self.__yaml_dict is None:
  187. self.__yaml_dict = {}
  188. @property
  189. def separator(self):
  190. ''' getter method for yaml_dict '''
  191. return self._separator
  192. @separator.setter
  193. def separator(self):
  194. ''' getter method for yaml_dict '''
  195. return self._separator
  196. @property
  197. def yaml_dict(self):
  198. ''' getter method for yaml_dict '''
  199. return self.__yaml_dict
  200. @yaml_dict.setter
  201. def yaml_dict(self, value):
  202. ''' setter method for yaml_dict '''
  203. self.__yaml_dict = value
  204. @staticmethod
  205. def parse_key(key, sep='.'):
  206. '''parse the key allowing the appropriate separator'''
  207. common_separators = list(Yedit.com_sep - set([sep]))
  208. return re.findall(Yedit.re_key % ''.join(common_separators), key)
  209. @staticmethod
  210. def valid_key(key, sep='.'):
  211. '''validate the incoming key'''
  212. common_separators = list(Yedit.com_sep - set([sep]))
  213. if not re.match(Yedit.re_valid_key % ''.join(common_separators), key):
  214. return False
  215. return True
  216. @staticmethod
  217. def remove_entry(data, key, sep='.'):
  218. ''' remove data at location key '''
  219. if key == '' and isinstance(data, dict):
  220. data.clear()
  221. return True
  222. elif key == '' and isinstance(data, list):
  223. del data[:]
  224. return True
  225. if not (key and Yedit.valid_key(key, sep)) and \
  226. isinstance(data, (list, dict)):
  227. return None
  228. key_indexes = Yedit.parse_key(key, sep)
  229. for arr_ind, dict_key in key_indexes[:-1]:
  230. if dict_key and isinstance(data, dict):
  231. data = data.get(dict_key, None)
  232. elif (arr_ind and isinstance(data, list) and
  233. int(arr_ind) <= len(data) - 1):
  234. data = data[int(arr_ind)]
  235. else:
  236. return None
  237. # process last index for remove
  238. # expected list entry
  239. if key_indexes[-1][0]:
  240. if isinstance(data, list) and int(key_indexes[-1][0]) <= len(data) - 1: # noqa: E501
  241. del data[int(key_indexes[-1][0])]
  242. return True
  243. # expected dict entry
  244. elif key_indexes[-1][1]:
  245. if isinstance(data, dict):
  246. del data[key_indexes[-1][1]]
  247. return True
  248. @staticmethod
  249. def add_entry(data, key, item=None, sep='.'):
  250. ''' Get an item from a dictionary with key notation a.b.c
  251. d = {'a': {'b': 'c'}}}
  252. key = a#b
  253. return c
  254. '''
  255. if key == '':
  256. pass
  257. elif (not (key and Yedit.valid_key(key, sep)) and
  258. isinstance(data, (list, dict))):
  259. return None
  260. key_indexes = Yedit.parse_key(key, sep)
  261. for arr_ind, dict_key in key_indexes[:-1]:
  262. if dict_key:
  263. if isinstance(data, dict) and dict_key in data and data[dict_key]: # noqa: E501
  264. data = data[dict_key]
  265. continue
  266. elif data and not isinstance(data, dict):
  267. return None
  268. data[dict_key] = {}
  269. data = data[dict_key]
  270. elif (arr_ind and isinstance(data, list) and
  271. int(arr_ind) <= len(data) - 1):
  272. data = data[int(arr_ind)]
  273. else:
  274. return None
  275. if key == '':
  276. data = item
  277. # process last index for add
  278. # expected list entry
  279. elif key_indexes[-1][0] and isinstance(data, list) and int(key_indexes[-1][0]) <= len(data) - 1: # noqa: E501
  280. data[int(key_indexes[-1][0])] = item
  281. # expected dict entry
  282. elif key_indexes[-1][1] and isinstance(data, dict):
  283. data[key_indexes[-1][1]] = item
  284. return data
  285. @staticmethod
  286. def get_entry(data, key, sep='.'):
  287. ''' Get an item from a dictionary with key notation a.b.c
  288. d = {'a': {'b': 'c'}}}
  289. key = a.b
  290. return c
  291. '''
  292. if key == '':
  293. pass
  294. elif (not (key and Yedit.valid_key(key, sep)) and
  295. isinstance(data, (list, dict))):
  296. return None
  297. key_indexes = Yedit.parse_key(key, sep)
  298. for arr_ind, dict_key in key_indexes:
  299. if dict_key and isinstance(data, dict):
  300. data = data.get(dict_key, None)
  301. elif (arr_ind and isinstance(data, list) and
  302. int(arr_ind) <= len(data) - 1):
  303. data = data[int(arr_ind)]
  304. else:
  305. return None
  306. return data
  307. @staticmethod
  308. def _write(filename, contents):
  309. ''' Actually write the file contents to disk. This helps with mocking. '''
  310. tmp_filename = filename + '.yedit'
  311. with open(tmp_filename, 'w') as yfd:
  312. yfd.write(contents)
  313. os.rename(tmp_filename, filename)
  314. def write(self):
  315. ''' write to file '''
  316. if not self.filename:
  317. raise YeditException('Please specify a filename.')
  318. if self.backup and self.file_exists():
  319. shutil.copy(self.filename, self.filename + '.orig')
  320. # Try to set format attributes if supported
  321. try:
  322. self.yaml_dict.fa.set_block_style()
  323. except AttributeError:
  324. pass
  325. # Try to use RoundTripDumper if supported.
  326. try:
  327. Yedit._write(self.filename, yaml.dump(self.yaml_dict, Dumper=yaml.RoundTripDumper))
  328. except AttributeError:
  329. Yedit._write(self.filename, yaml.safe_dump(self.yaml_dict, default_flow_style=False))
  330. return (True, self.yaml_dict)
  331. def read(self):
  332. ''' read from file '''
  333. # check if it exists
  334. if self.filename is None or not self.file_exists():
  335. return None
  336. contents = None
  337. with open(self.filename) as yfd:
  338. contents = yfd.read()
  339. return contents
  340. def file_exists(self):
  341. ''' return whether file exists '''
  342. if os.path.exists(self.filename):
  343. return True
  344. return False
  345. def load(self, content_type='yaml'):
  346. ''' return yaml file '''
  347. contents = self.read()
  348. if not contents and not self.content:
  349. return None
  350. if self.content:
  351. if isinstance(self.content, dict):
  352. self.yaml_dict = self.content
  353. return self.yaml_dict
  354. elif isinstance(self.content, str):
  355. contents = self.content
  356. # check if it is yaml
  357. try:
  358. if content_type == 'yaml' and contents:
  359. # Try to set format attributes if supported
  360. try:
  361. self.yaml_dict.fa.set_block_style()
  362. except AttributeError:
  363. pass
  364. # Try to use RoundTripLoader if supported.
  365. try:
  366. self.yaml_dict = yaml.safe_load(contents, yaml.RoundTripLoader)
  367. except AttributeError:
  368. self.yaml_dict = yaml.safe_load(contents)
  369. # Try to set format attributes if supported
  370. try:
  371. self.yaml_dict.fa.set_block_style()
  372. except AttributeError:
  373. pass
  374. elif content_type == 'json' and contents:
  375. self.yaml_dict = json.loads(contents)
  376. except yaml.YAMLError as err:
  377. # Error loading yaml or json
  378. raise YeditException('Problem with loading yaml file. %s' % err)
  379. return self.yaml_dict
  380. def get(self, key):
  381. ''' get a specified key'''
  382. try:
  383. entry = Yedit.get_entry(self.yaml_dict, key, self.separator)
  384. except KeyError:
  385. entry = None
  386. return entry
  387. def pop(self, path, key_or_item):
  388. ''' remove a key, value pair from a dict or an item for a list'''
  389. try:
  390. entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
  391. except KeyError:
  392. entry = None
  393. if entry is None:
  394. return (False, self.yaml_dict)
  395. if isinstance(entry, dict):
  396. # AUDIT:maybe-no-member makes sense due to fuzzy types
  397. # pylint: disable=maybe-no-member
  398. if key_or_item in entry:
  399. entry.pop(key_or_item)
  400. return (True, self.yaml_dict)
  401. return (False, self.yaml_dict)
  402. elif isinstance(entry, list):
  403. # AUDIT:maybe-no-member makes sense due to fuzzy types
  404. # pylint: disable=maybe-no-member
  405. ind = None
  406. try:
  407. ind = entry.index(key_or_item)
  408. except ValueError:
  409. return (False, self.yaml_dict)
  410. entry.pop(ind)
  411. return (True, self.yaml_dict)
  412. return (False, self.yaml_dict)
  413. def delete(self, path):
  414. ''' remove path from a dict'''
  415. try:
  416. entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
  417. except KeyError:
  418. entry = None
  419. if entry is None:
  420. return (False, self.yaml_dict)
  421. result = Yedit.remove_entry(self.yaml_dict, path, self.separator)
  422. if not result:
  423. return (False, self.yaml_dict)
  424. return (True, self.yaml_dict)
  425. def exists(self, path, value):
  426. ''' check if value exists at path'''
  427. try:
  428. entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
  429. except KeyError:
  430. entry = None
  431. if isinstance(entry, list):
  432. if value in entry:
  433. return True
  434. return False
  435. elif isinstance(entry, dict):
  436. if isinstance(value, dict):
  437. rval = False
  438. for key, val in value.items():
  439. if entry[key] != val:
  440. rval = False
  441. break
  442. else:
  443. rval = True
  444. return rval
  445. return value in entry
  446. return entry == value
  447. def append(self, path, value):
  448. '''append value to a list'''
  449. try:
  450. entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
  451. except KeyError:
  452. entry = None
  453. if entry is None:
  454. self.put(path, [])
  455. entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
  456. if not isinstance(entry, list):
  457. return (False, self.yaml_dict)
  458. # AUDIT:maybe-no-member makes sense due to loading data from
  459. # a serialized format.
  460. # pylint: disable=maybe-no-member
  461. entry.append(value)
  462. return (True, self.yaml_dict)
  463. # pylint: disable=too-many-arguments
  464. def update(self, path, value, index=None, curr_value=None):
  465. ''' put path, value into a dict '''
  466. try:
  467. entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
  468. except KeyError:
  469. entry = None
  470. if isinstance(entry, dict):
  471. # AUDIT:maybe-no-member makes sense due to fuzzy types
  472. # pylint: disable=maybe-no-member
  473. if not isinstance(value, dict):
  474. raise YeditException('Cannot replace key, value entry in ' +
  475. 'dict with non-dict type. value=[%s] [%s]' % (value, type(value))) # noqa: E501
  476. entry.update(value)
  477. return (True, self.yaml_dict)
  478. elif isinstance(entry, list):
  479. # AUDIT:maybe-no-member makes sense due to fuzzy types
  480. # pylint: disable=maybe-no-member
  481. ind = None
  482. if curr_value:
  483. try:
  484. ind = entry.index(curr_value)
  485. except ValueError:
  486. return (False, self.yaml_dict)
  487. elif index is not None:
  488. ind = index
  489. if ind is not None and entry[ind] != value:
  490. entry[ind] = value
  491. return (True, self.yaml_dict)
  492. # see if it exists in the list
  493. try:
  494. ind = entry.index(value)
  495. except ValueError:
  496. # doesn't exist, append it
  497. entry.append(value)
  498. return (True, self.yaml_dict)
  499. # already exists, return
  500. if ind is not None:
  501. return (False, self.yaml_dict)
  502. return (False, self.yaml_dict)
  503. def put(self, path, value):
  504. ''' put path, value into a dict '''
  505. try:
  506. entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
  507. except KeyError:
  508. entry = None
  509. if entry == value:
  510. return (False, self.yaml_dict)
  511. # deepcopy didn't work
  512. # Try to use ruamel.yaml and fallback to pyyaml
  513. try:
  514. tmp_copy = yaml.load(yaml.round_trip_dump(self.yaml_dict,
  515. default_flow_style=False),
  516. yaml.RoundTripLoader)
  517. except AttributeError:
  518. tmp_copy = copy.deepcopy(self.yaml_dict)
  519. # set the format attributes if available
  520. try:
  521. tmp_copy.fa.set_block_style()
  522. except AttributeError:
  523. pass
  524. result = Yedit.add_entry(tmp_copy, path, value, self.separator)
  525. if not result:
  526. return (False, self.yaml_dict)
  527. self.yaml_dict = tmp_copy
  528. return (True, self.yaml_dict)
  529. def create(self, path, value):
  530. ''' create a yaml file '''
  531. if not self.file_exists():
  532. # deepcopy didn't work
  533. # Try to use ruamel.yaml and fallback to pyyaml
  534. try:
  535. tmp_copy = yaml.load(yaml.round_trip_dump(self.yaml_dict,
  536. default_flow_style=False),
  537. yaml.RoundTripLoader)
  538. except AttributeError:
  539. tmp_copy = copy.deepcopy(self.yaml_dict)
  540. # set the format attributes if available
  541. try:
  542. tmp_copy.fa.set_block_style()
  543. except AttributeError:
  544. pass
  545. result = Yedit.add_entry(tmp_copy, path, value, self.separator)
  546. if result:
  547. self.yaml_dict = tmp_copy
  548. return (True, self.yaml_dict)
  549. return (False, self.yaml_dict)
  550. @staticmethod
  551. def get_curr_value(invalue, val_type):
  552. '''return the current value'''
  553. if invalue is None:
  554. return None
  555. curr_value = invalue
  556. if val_type == 'yaml':
  557. curr_value = yaml.load(invalue)
  558. elif val_type == 'json':
  559. curr_value = json.loads(invalue)
  560. return curr_value
  561. @staticmethod
  562. def parse_value(inc_value, vtype=''):
  563. '''determine value type passed'''
  564. true_bools = ['y', 'Y', 'yes', 'Yes', 'YES', 'true', 'True', 'TRUE',
  565. 'on', 'On', 'ON', ]
  566. false_bools = ['n', 'N', 'no', 'No', 'NO', 'false', 'False', 'FALSE',
  567. 'off', 'Off', 'OFF']
  568. # It came in as a string but you didn't specify value_type as string
  569. # we will convert to bool if it matches any of the above cases
  570. if isinstance(inc_value, str) and 'bool' in vtype:
  571. if inc_value not in true_bools and inc_value not in false_bools:
  572. raise YeditException('Not a boolean type. str=[%s] vtype=[%s]'
  573. % (inc_value, vtype))
  574. elif isinstance(inc_value, bool) and 'str' in vtype:
  575. inc_value = str(inc_value)
  576. # If vtype is not str then go ahead and attempt to yaml load it.
  577. if isinstance(inc_value, str) and 'str' not in vtype:
  578. try:
  579. inc_value = yaml.load(inc_value)
  580. except Exception:
  581. raise YeditException('Could not determine type of incoming ' +
  582. 'value. value=[%s] vtype=[%s]'
  583. % (type(inc_value), vtype))
  584. return inc_value
  585. # pylint: disable=too-many-return-statements,too-many-branches
  586. @staticmethod
  587. def run_ansible(module):
  588. '''perform the idempotent crud operations'''
  589. yamlfile = Yedit(filename=module.params['src'],
  590. backup=module.params['backup'],
  591. separator=module.params['separator'])
  592. if module.params['src']:
  593. rval = yamlfile.load()
  594. if yamlfile.yaml_dict is None and \
  595. module.params['state'] != 'present':
  596. return {'failed': True,
  597. 'msg': 'Error opening file [%s]. Verify that the ' +
  598. 'file exists, that it is has correct' +
  599. ' permissions, and is valid yaml.'}
  600. if module.params['state'] == 'list':
  601. if module.params['content']:
  602. content = Yedit.parse_value(module.params['content'],
  603. module.params['content_type'])
  604. yamlfile.yaml_dict = content
  605. if module.params['key']:
  606. rval = yamlfile.get(module.params['key']) or {}
  607. return {'changed': False, 'result': rval, 'state': "list"}
  608. elif module.params['state'] == 'absent':
  609. if module.params['content']:
  610. content = Yedit.parse_value(module.params['content'],
  611. module.params['content_type'])
  612. yamlfile.yaml_dict = content
  613. if module.params['update']:
  614. rval = yamlfile.pop(module.params['key'],
  615. module.params['value'])
  616. else:
  617. rval = yamlfile.delete(module.params['key'])
  618. if rval[0] and module.params['src']:
  619. yamlfile.write()
  620. return {'changed': rval[0], 'result': rval[1], 'state': "absent"}
  621. elif module.params['state'] == 'present':
  622. # check if content is different than what is in the file
  623. if module.params['content']:
  624. content = Yedit.parse_value(module.params['content'],
  625. module.params['content_type'])
  626. # We had no edits to make and the contents are the same
  627. if yamlfile.yaml_dict == content and \
  628. module.params['value'] is None:
  629. return {'changed': False,
  630. 'result': yamlfile.yaml_dict,
  631. 'state': "present"}
  632. yamlfile.yaml_dict = content
  633. # we were passed a value; parse it
  634. if module.params['value']:
  635. value = Yedit.parse_value(module.params['value'],
  636. module.params['value_type'])
  637. key = module.params['key']
  638. if module.params['update']:
  639. # pylint: disable=line-too-long
  640. curr_value = Yedit.get_curr_value(Yedit.parse_value(module.params['curr_value']), # noqa: E501
  641. module.params['curr_value_format']) # noqa: E501
  642. rval = yamlfile.update(key, value, module.params['index'], curr_value) # noqa: E501
  643. elif module.params['append']:
  644. rval = yamlfile.append(key, value)
  645. else:
  646. rval = yamlfile.put(key, value)
  647. if rval[0] and module.params['src']:
  648. yamlfile.write()
  649. return {'changed': rval[0],
  650. 'result': rval[1], 'state': "present"}
  651. # no edits to make
  652. if module.params['src']:
  653. # pylint: disable=redefined-variable-type
  654. rval = yamlfile.write()
  655. return {'changed': rval[0],
  656. 'result': rval[1],
  657. 'state': "present"}
  658. return {'failed': True, 'msg': 'Unkown state passed'}
  659. # -*- -*- -*- End included fragment: ../../lib_utils/src/class/yedit.py -*- -*- -*-
  660. # -*- -*- -*- Begin included fragment: lib/base.py -*- -*- -*-
  661. # pylint: disable=too-many-lines
  662. # noqa: E301,E302,E303,T001
  663. class OpenShiftCLIError(Exception):
  664. '''Exception class for openshiftcli'''
  665. pass
  666. ADDITIONAL_PATH_LOOKUPS = ['/usr/local/bin', os.path.expanduser('~/bin')]
  667. def locate_oc_binary():
  668. ''' Find and return oc binary file '''
  669. # https://github.com/openshift/openshift-ansible/issues/3410
  670. # oc can be in /usr/local/bin in some cases, but that may not
  671. # be in $PATH due to ansible/sudo
  672. paths = os.environ.get("PATH", os.defpath).split(os.pathsep) + ADDITIONAL_PATH_LOOKUPS
  673. oc_binary = 'oc'
  674. # Use shutil.which if it is available, otherwise fallback to a naive path search
  675. try:
  676. which_result = shutil.which(oc_binary, path=os.pathsep.join(paths))
  677. if which_result is not None:
  678. oc_binary = which_result
  679. except AttributeError:
  680. for path in paths:
  681. if os.path.exists(os.path.join(path, oc_binary)):
  682. oc_binary = os.path.join(path, oc_binary)
  683. break
  684. return oc_binary
  685. # pylint: disable=too-few-public-methods
  686. class OpenShiftCLI(object):
  687. ''' Class to wrap the command line tools '''
  688. def __init__(self,
  689. namespace,
  690. kubeconfig='/etc/origin/master/admin.kubeconfig',
  691. verbose=False,
  692. all_namespaces=False):
  693. ''' Constructor for OpenshiftCLI '''
  694. self.namespace = namespace
  695. self.verbose = verbose
  696. self.kubeconfig = Utils.create_tmpfile_copy(kubeconfig)
  697. self.all_namespaces = all_namespaces
  698. self.oc_binary = locate_oc_binary()
  699. # Pylint allows only 5 arguments to be passed.
  700. # pylint: disable=too-many-arguments
  701. def _replace_content(self, resource, rname, content, force=False, sep='.'):
  702. ''' replace the current object with the content '''
  703. res = self._get(resource, rname)
  704. if not res['results']:
  705. return res
  706. fname = Utils.create_tmpfile(rname + '-')
  707. yed = Yedit(fname, res['results'][0], separator=sep)
  708. changes = []
  709. for key, value in content.items():
  710. changes.append(yed.put(key, value))
  711. if any([change[0] for change in changes]):
  712. yed.write()
  713. atexit.register(Utils.cleanup, [fname])
  714. return self._replace(fname, force)
  715. return {'returncode': 0, 'updated': False}
  716. def _replace(self, fname, force=False):
  717. '''replace the current object with oc replace'''
  718. cmd = ['replace', '-f', fname]
  719. if force:
  720. cmd.append('--force')
  721. return self.openshift_cmd(cmd)
  722. def _create_from_content(self, rname, content):
  723. '''create a temporary file and then call oc create on it'''
  724. fname = Utils.create_tmpfile(rname + '-')
  725. yed = Yedit(fname, content=content)
  726. yed.write()
  727. atexit.register(Utils.cleanup, [fname])
  728. return self._create(fname)
  729. def _create(self, fname):
  730. '''call oc create on a filename'''
  731. return self.openshift_cmd(['create', '-f', fname])
  732. def _delete(self, resource, rname, selector=None):
  733. '''call oc delete on a resource'''
  734. cmd = ['delete', resource, rname]
  735. if selector:
  736. cmd.append('--selector=%s' % selector)
  737. return self.openshift_cmd(cmd)
  738. def _process(self, template_name, create=False, params=None, template_data=None): # noqa: E501
  739. '''process a template
  740. template_name: the name of the template to process
  741. create: whether to send to oc create after processing
  742. params: the parameters for the template
  743. template_data: the incoming template's data; instead of a file
  744. '''
  745. cmd = ['process']
  746. if template_data:
  747. cmd.extend(['-f', '-'])
  748. else:
  749. cmd.append(template_name)
  750. if params:
  751. param_str = ["%s=%s" % (key, value) for key, value in params.items()]
  752. cmd.append('-v')
  753. cmd.extend(param_str)
  754. results = self.openshift_cmd(cmd, output=True, input_data=template_data)
  755. if results['returncode'] != 0 or not create:
  756. return results
  757. fname = Utils.create_tmpfile(template_name + '-')
  758. yed = Yedit(fname, results['results'])
  759. yed.write()
  760. atexit.register(Utils.cleanup, [fname])
  761. return self.openshift_cmd(['create', '-f', fname])
  762. def _get(self, resource, rname=None, selector=None):
  763. '''return a resource by name '''
  764. cmd = ['get', resource]
  765. if selector:
  766. cmd.append('--selector=%s' % selector)
  767. elif rname:
  768. cmd.append(rname)
  769. cmd.extend(['-o', 'json'])
  770. rval = self.openshift_cmd(cmd, output=True)
  771. # Ensure results are retuned in an array
  772. if 'items' in rval:
  773. rval['results'] = rval['items']
  774. elif not isinstance(rval['results'], list):
  775. rval['results'] = [rval['results']]
  776. return rval
  777. def _schedulable(self, node=None, selector=None, schedulable=True):
  778. ''' perform oadm manage-node scheduable '''
  779. cmd = ['manage-node']
  780. if node:
  781. cmd.extend(node)
  782. else:
  783. cmd.append('--selector=%s' % selector)
  784. cmd.append('--schedulable=%s' % schedulable)
  785. return self.openshift_cmd(cmd, oadm=True, output=True, output_type='raw') # noqa: E501
  786. def _list_pods(self, node=None, selector=None, pod_selector=None):
  787. ''' perform oadm list pods
  788. node: the node in which to list pods
  789. selector: the label selector filter if provided
  790. pod_selector: the pod selector filter if provided
  791. '''
  792. cmd = ['manage-node']
  793. if node:
  794. cmd.extend(node)
  795. else:
  796. cmd.append('--selector=%s' % selector)
  797. if pod_selector:
  798. cmd.append('--pod-selector=%s' % pod_selector)
  799. cmd.extend(['--list-pods', '-o', 'json'])
  800. return self.openshift_cmd(cmd, oadm=True, output=True, output_type='raw')
  801. # pylint: disable=too-many-arguments
  802. def _evacuate(self, node=None, selector=None, pod_selector=None, dry_run=False, grace_period=None, force=False):
  803. ''' perform oadm manage-node evacuate '''
  804. cmd = ['manage-node']
  805. if node:
  806. cmd.extend(node)
  807. else:
  808. cmd.append('--selector=%s' % selector)
  809. if dry_run:
  810. cmd.append('--dry-run')
  811. if pod_selector:
  812. cmd.append('--pod-selector=%s' % pod_selector)
  813. if grace_period:
  814. cmd.append('--grace-period=%s' % int(grace_period))
  815. if force:
  816. cmd.append('--force')
  817. cmd.append('--evacuate')
  818. return self.openshift_cmd(cmd, oadm=True, output=True, output_type='raw')
  819. def _version(self):
  820. ''' return the openshift version'''
  821. return self.openshift_cmd(['version'], output=True, output_type='raw')
  822. def _import_image(self, url=None, name=None, tag=None):
  823. ''' perform image import '''
  824. cmd = ['import-image']
  825. image = '{0}'.format(name)
  826. if tag:
  827. image += ':{0}'.format(tag)
  828. cmd.append(image)
  829. if url:
  830. cmd.append('--from={0}/{1}'.format(url, image))
  831. cmd.append('-n{0}'.format(self.namespace))
  832. cmd.append('--confirm')
  833. return self.openshift_cmd(cmd)
  834. def _run(self, cmds, input_data):
  835. ''' Actually executes the command. This makes mocking easier. '''
  836. curr_env = os.environ.copy()
  837. curr_env.update({'KUBECONFIG': self.kubeconfig})
  838. proc = subprocess.Popen(cmds,
  839. stdin=subprocess.PIPE,
  840. stdout=subprocess.PIPE,
  841. stderr=subprocess.PIPE,
  842. env=curr_env)
  843. stdout, stderr = proc.communicate(input_data)
  844. return proc.returncode, stdout.decode(), stderr.decode()
  845. # pylint: disable=too-many-arguments,too-many-branches
  846. def openshift_cmd(self, cmd, oadm=False, output=False, output_type='json', input_data=None):
  847. '''Base command for oc '''
  848. cmds = [self.oc_binary]
  849. if oadm:
  850. cmds.append('adm')
  851. cmds.extend(cmd)
  852. if self.all_namespaces:
  853. cmds.extend(['--all-namespaces'])
  854. elif self.namespace is not None and self.namespace.lower() not in ['none', 'emtpy']: # E501
  855. cmds.extend(['-n', self.namespace])
  856. rval = {}
  857. results = ''
  858. err = None
  859. if self.verbose:
  860. print(' '.join(cmds))
  861. try:
  862. returncode, stdout, stderr = self._run(cmds, input_data)
  863. except OSError as ex:
  864. returncode, stdout, stderr = 1, '', 'Failed to execute {}: {}'.format(subprocess.list2cmdline(cmds), ex)
  865. rval = {"returncode": returncode,
  866. "results": results,
  867. "cmd": ' '.join(cmds)}
  868. if returncode == 0:
  869. if output:
  870. if output_type == 'json':
  871. try:
  872. rval['results'] = json.loads(stdout)
  873. except ValueError as err:
  874. if "No JSON object could be decoded" in err.args:
  875. err = err.args
  876. elif output_type == 'raw':
  877. rval['results'] = stdout
  878. if self.verbose:
  879. print("STDOUT: {0}".format(stdout))
  880. print("STDERR: {0}".format(stderr))
  881. if err:
  882. rval.update({"err": err,
  883. "stderr": stderr,
  884. "stdout": stdout,
  885. "cmd": cmds})
  886. else:
  887. rval.update({"stderr": stderr,
  888. "stdout": stdout,
  889. "results": {}})
  890. return rval
  891. class Utils(object):
  892. ''' utilities for openshiftcli modules '''
  893. @staticmethod
  894. def _write(filename, contents):
  895. ''' Actually write the file contents to disk. This helps with mocking. '''
  896. with open(filename, 'w') as sfd:
  897. sfd.write(contents)
  898. @staticmethod
  899. def create_tmp_file_from_contents(rname, data, ftype='yaml'):
  900. ''' create a file in tmp with name and contents'''
  901. tmp = Utils.create_tmpfile(prefix=rname)
  902. if ftype == 'yaml':
  903. # AUDIT:no-member makes sense here due to ruamel.YAML/PyYAML usage
  904. # pylint: disable=no-member
  905. if hasattr(yaml, 'RoundTripDumper'):
  906. Utils._write(tmp, yaml.dump(data, Dumper=yaml.RoundTripDumper))
  907. else:
  908. Utils._write(tmp, yaml.safe_dump(data, default_flow_style=False))
  909. elif ftype == 'json':
  910. Utils._write(tmp, json.dumps(data))
  911. else:
  912. Utils._write(tmp, data)
  913. # Register cleanup when module is done
  914. atexit.register(Utils.cleanup, [tmp])
  915. return tmp
  916. @staticmethod
  917. def create_tmpfile_copy(inc_file):
  918. '''create a temporary copy of a file'''
  919. tmpfile = Utils.create_tmpfile('lib_openshift-')
  920. Utils._write(tmpfile, open(inc_file).read())
  921. # Cleanup the tmpfile
  922. atexit.register(Utils.cleanup, [tmpfile])
  923. return tmpfile
  924. @staticmethod
  925. def create_tmpfile(prefix='tmp'):
  926. ''' Generates and returns a temporary file name '''
  927. with tempfile.NamedTemporaryFile(prefix=prefix, delete=False) as tmp:
  928. return tmp.name
  929. @staticmethod
  930. def create_tmp_files_from_contents(content, content_type=None):
  931. '''Turn an array of dict: filename, content into a files array'''
  932. if not isinstance(content, list):
  933. content = [content]
  934. files = []
  935. for item in content:
  936. path = Utils.create_tmp_file_from_contents(item['path'] + '-',
  937. item['data'],
  938. ftype=content_type)
  939. files.append({'name': os.path.basename(item['path']),
  940. 'path': path})
  941. return files
  942. @staticmethod
  943. def cleanup(files):
  944. '''Clean up on exit '''
  945. for sfile in files:
  946. if os.path.exists(sfile):
  947. if os.path.isdir(sfile):
  948. shutil.rmtree(sfile)
  949. elif os.path.isfile(sfile):
  950. os.remove(sfile)
  951. @staticmethod
  952. def exists(results, _name):
  953. ''' Check to see if the results include the name '''
  954. if not results:
  955. return False
  956. if Utils.find_result(results, _name):
  957. return True
  958. return False
  959. @staticmethod
  960. def find_result(results, _name):
  961. ''' Find the specified result by name'''
  962. rval = None
  963. for result in results:
  964. if 'metadata' in result and result['metadata']['name'] == _name:
  965. rval = result
  966. break
  967. return rval
  968. @staticmethod
  969. def get_resource_file(sfile, sfile_type='yaml'):
  970. ''' return the service file '''
  971. contents = None
  972. with open(sfile) as sfd:
  973. contents = sfd.read()
  974. if sfile_type == 'yaml':
  975. # AUDIT:no-member makes sense here due to ruamel.YAML/PyYAML usage
  976. # pylint: disable=no-member
  977. if hasattr(yaml, 'RoundTripLoader'):
  978. contents = yaml.load(contents, yaml.RoundTripLoader)
  979. else:
  980. contents = yaml.safe_load(contents)
  981. elif sfile_type == 'json':
  982. contents = json.loads(contents)
  983. return contents
  984. @staticmethod
  985. def filter_versions(stdout):
  986. ''' filter the oc version output '''
  987. version_dict = {}
  988. version_search = ['oc', 'openshift', 'kubernetes']
  989. for line in stdout.strip().split('\n'):
  990. for term in version_search:
  991. if not line:
  992. continue
  993. if line.startswith(term):
  994. version_dict[term] = line.split()[-1]
  995. # horrible hack to get openshift version in Openshift 3.2
  996. # By default "oc version in 3.2 does not return an "openshift" version
  997. if "openshift" not in version_dict:
  998. version_dict["openshift"] = version_dict["oc"]
  999. return version_dict
  1000. @staticmethod
  1001. def add_custom_versions(versions):
  1002. ''' create custom versions strings '''
  1003. versions_dict = {}
  1004. for tech, version in versions.items():
  1005. # clean up "-" from version
  1006. if "-" in version:
  1007. version = version.split("-")[0]
  1008. if version.startswith('v'):
  1009. versions_dict[tech + '_numeric'] = version[1:].split('+')[0]
  1010. # "v3.3.0.33" is what we have, we want "3.3"
  1011. versions_dict[tech + '_short'] = version[1:4]
  1012. return versions_dict
  1013. @staticmethod
  1014. def openshift_installed():
  1015. ''' check if openshift is installed '''
  1016. import yum
  1017. yum_base = yum.YumBase()
  1018. if yum_base.rpmdb.searchNevra(name='atomic-openshift'):
  1019. return True
  1020. return False
  1021. # Disabling too-many-branches. This is a yaml dictionary comparison function
  1022. # pylint: disable=too-many-branches,too-many-return-statements,too-many-statements
  1023. @staticmethod
  1024. def check_def_equal(user_def, result_def, skip_keys=None, debug=False):
  1025. ''' Given a user defined definition, compare it with the results given back by our query. '''
  1026. # Currently these values are autogenerated and we do not need to check them
  1027. skip = ['metadata', 'status']
  1028. if skip_keys:
  1029. skip.extend(skip_keys)
  1030. for key, value in result_def.items():
  1031. if key in skip:
  1032. continue
  1033. # Both are lists
  1034. if isinstance(value, list):
  1035. if key not in user_def:
  1036. if debug:
  1037. print('User data does not have key [%s]' % key)
  1038. print('User data: %s' % user_def)
  1039. return False
  1040. if not isinstance(user_def[key], list):
  1041. if debug:
  1042. print('user_def[key] is not a list key=[%s] user_def[key]=%s' % (key, user_def[key]))
  1043. return False
  1044. if len(user_def[key]) != len(value):
  1045. if debug:
  1046. print("List lengths are not equal.")
  1047. print("key=[%s]: user_def[%s] != value[%s]" % (key, len(user_def[key]), len(value)))
  1048. print("user_def: %s" % user_def[key])
  1049. print("value: %s" % value)
  1050. return False
  1051. for values in zip(user_def[key], value):
  1052. if isinstance(values[0], dict) and isinstance(values[1], dict):
  1053. if debug:
  1054. print('sending list - list')
  1055. print(type(values[0]))
  1056. print(type(values[1]))
  1057. result = Utils.check_def_equal(values[0], values[1], skip_keys=skip_keys, debug=debug)
  1058. if not result:
  1059. print('list compare returned false')
  1060. return False
  1061. elif value != user_def[key]:
  1062. if debug:
  1063. print('value should be identical')
  1064. print(value)
  1065. print(user_def[key])
  1066. return False
  1067. # recurse on a dictionary
  1068. elif isinstance(value, dict):
  1069. if key not in user_def:
  1070. if debug:
  1071. print("user_def does not have key [%s]" % key)
  1072. return False
  1073. if not isinstance(user_def[key], dict):
  1074. if debug:
  1075. print("dict returned false: not instance of dict")
  1076. return False
  1077. # before passing ensure keys match
  1078. api_values = set(value.keys()) - set(skip)
  1079. user_values = set(user_def[key].keys()) - set(skip)
  1080. if api_values != user_values:
  1081. if debug:
  1082. print("keys are not equal in dict")
  1083. print(api_values)
  1084. print(user_values)
  1085. return False
  1086. result = Utils.check_def_equal(user_def[key], value, skip_keys=skip_keys, debug=debug)
  1087. if not result:
  1088. if debug:
  1089. print("dict returned false")
  1090. print(result)
  1091. return False
  1092. # Verify each key, value pair is the same
  1093. else:
  1094. if key not in user_def or value != user_def[key]:
  1095. if debug:
  1096. print("value not equal; user_def does not have key")
  1097. print(key)
  1098. print(value)
  1099. if key in user_def:
  1100. print(user_def[key])
  1101. return False
  1102. if debug:
  1103. print('returning true')
  1104. return True
  1105. class OpenShiftCLIConfig(object):
  1106. '''Generic Config'''
  1107. def __init__(self, rname, namespace, kubeconfig, options):
  1108. self.kubeconfig = kubeconfig
  1109. self.name = rname
  1110. self.namespace = namespace
  1111. self._options = options
  1112. @property
  1113. def config_options(self):
  1114. ''' return config options '''
  1115. return self._options
  1116. def to_option_list(self):
  1117. '''return all options as a string'''
  1118. return self.stringify()
  1119. def stringify(self):
  1120. ''' return the options hash as cli params in a string '''
  1121. rval = []
  1122. for key, data in self.config_options.items():
  1123. if data['include'] \
  1124. and (data['value'] or isinstance(data['value'], int)):
  1125. rval.append('--%s=%s' % (key.replace('_', '-'), data['value']))
  1126. return rval
  1127. # -*- -*- -*- End included fragment: lib/base.py -*- -*- -*-
  1128. # -*- -*- -*- Begin included fragment: lib/secret.py -*- -*- -*-
  1129. # pylint: disable=too-many-instance-attributes
  1130. class SecretConfig(object):
  1131. ''' Handle secret options '''
  1132. # pylint: disable=too-many-arguments
  1133. def __init__(self,
  1134. sname,
  1135. namespace,
  1136. kubeconfig,
  1137. secrets=None):
  1138. ''' constructor for handling secret options '''
  1139. self.kubeconfig = kubeconfig
  1140. self.name = sname
  1141. self.namespace = namespace
  1142. self.secrets = secrets
  1143. self.data = {}
  1144. self.create_dict()
  1145. def create_dict(self):
  1146. ''' assign the correct properties for a secret dict '''
  1147. self.data['apiVersion'] = 'v1'
  1148. self.data['kind'] = 'Secret'
  1149. self.data['metadata'] = {}
  1150. self.data['metadata']['name'] = self.name
  1151. self.data['metadata']['namespace'] = self.namespace
  1152. self.data['data'] = {}
  1153. if self.secrets:
  1154. for key, value in self.secrets.items():
  1155. self.data['data'][key] = value
  1156. # pylint: disable=too-many-instance-attributes
  1157. class Secret(Yedit):
  1158. ''' Class to wrap the oc command line tools '''
  1159. secret_path = "data"
  1160. kind = 'secret'
  1161. def __init__(self, content):
  1162. '''secret constructor'''
  1163. super(Secret, self).__init__(content=content)
  1164. self._secrets = None
  1165. @property
  1166. def secrets(self):
  1167. '''secret property getter'''
  1168. if self._secrets is None:
  1169. self._secrets = self.get_secrets()
  1170. return self._secrets
  1171. @secrets.setter
  1172. def secrets(self):
  1173. '''secret property setter'''
  1174. if self._secrets is None:
  1175. self._secrets = self.get_secrets()
  1176. return self._secrets
  1177. def get_secrets(self):
  1178. ''' returns all of the defined secrets '''
  1179. return self.get(Secret.secret_path) or {}
  1180. def add_secret(self, key, value):
  1181. ''' add a secret '''
  1182. if self.secrets:
  1183. self.secrets[key] = value
  1184. else:
  1185. self.put(Secret.secret_path, {key: value})
  1186. return True
  1187. def delete_secret(self, key):
  1188. ''' delete secret'''
  1189. try:
  1190. del self.secrets[key]
  1191. except KeyError as _:
  1192. return False
  1193. return True
  1194. def find_secret(self, key):
  1195. ''' find secret'''
  1196. rval = None
  1197. try:
  1198. rval = self.secrets[key]
  1199. except KeyError as _:
  1200. return None
  1201. return {'key': key, 'value': rval}
  1202. def update_secret(self, key, value):
  1203. ''' update a secret'''
  1204. if key in self.secrets:
  1205. self.secrets[key] = value
  1206. else:
  1207. self.add_secret(key, value)
  1208. return True
  1209. # -*- -*- -*- End included fragment: lib/secret.py -*- -*- -*-
  1210. # -*- -*- -*- Begin included fragment: class/oc_secret.py -*- -*- -*-
  1211. # pylint: disable=wrong-import-position,wrong-import-order
  1212. import base64
  1213. # pylint: disable=too-many-arguments
  1214. class OCSecret(OpenShiftCLI):
  1215. ''' Class to wrap the oc command line tools
  1216. '''
  1217. def __init__(self,
  1218. namespace,
  1219. secret_name=None,
  1220. decode=False,
  1221. kubeconfig='/etc/origin/master/admin.kubeconfig',
  1222. verbose=False):
  1223. ''' Constructor for OpenshiftOC '''
  1224. super(OCSecret, self).__init__(namespace, kubeconfig=kubeconfig, verbose=verbose)
  1225. self.name = secret_name
  1226. self.decode = decode
  1227. def get(self):
  1228. '''return a secret by name '''
  1229. results = self._get('secrets', self.name)
  1230. results['decoded'] = {}
  1231. results['exists'] = False
  1232. if results['returncode'] == 0 and results['results'][0]:
  1233. results['exists'] = True
  1234. if self.decode:
  1235. if 'data' in results['results'][0]:
  1236. for sname, value in results['results'][0]['data'].items():
  1237. results['decoded'][sname] = base64.b64decode(value)
  1238. if results['returncode'] != 0 and '"%s" not found' % self.name in results['stderr']:
  1239. results['returncode'] = 0
  1240. return results
  1241. def delete(self):
  1242. '''delete a secret by name'''
  1243. return self._delete('secrets', self.name)
  1244. def create(self, files=None, contents=None):
  1245. '''Create a secret '''
  1246. if not files:
  1247. files = Utils.create_tmp_files_from_contents(contents)
  1248. secrets = ["%s=%s" % (sfile['name'], sfile['path']) for sfile in files]
  1249. cmd = ['secrets', 'new', self.name]
  1250. cmd.extend(secrets)
  1251. results = self.openshift_cmd(cmd)
  1252. return results
  1253. def update(self, files, force=False):
  1254. '''run update secret
  1255. This receives a list of file names and converts it into a secret.
  1256. The secret is then written to disk and passed into the `oc replace` command.
  1257. '''
  1258. secret = self.prep_secret(files)
  1259. if secret['returncode'] != 0:
  1260. return secret
  1261. sfile_path = '/tmp/%s' % self.name
  1262. with open(sfile_path, 'w') as sfd:
  1263. sfd.write(json.dumps(secret['results']))
  1264. atexit.register(Utils.cleanup, [sfile_path])
  1265. return self._replace(sfile_path, force=force)
  1266. def prep_secret(self, files=None, contents=None):
  1267. ''' return what the secret would look like if created
  1268. This is accomplished by passing -ojson. This will most likely change in the future
  1269. '''
  1270. if not files:
  1271. files = Utils.create_tmp_files_from_contents(contents)
  1272. secrets = ["%s=%s" % (sfile['name'], sfile['path']) for sfile in files]
  1273. cmd = ['-ojson', 'secrets', 'new', self.name]
  1274. cmd.extend(secrets)
  1275. return self.openshift_cmd(cmd, output=True)
  1276. @staticmethod
  1277. # pylint: disable=too-many-return-statements,too-many-branches
  1278. # TODO: This function should be refactored into its individual parts.
  1279. def run_ansible(params, check_mode):
  1280. '''run the ansible idempotent code'''
  1281. ocsecret = OCSecret(params['namespace'],
  1282. params['name'],
  1283. params['decode'],
  1284. kubeconfig=params['kubeconfig'],
  1285. verbose=params['debug'])
  1286. state = params['state']
  1287. api_rval = ocsecret.get()
  1288. #####
  1289. # Get
  1290. #####
  1291. if state == 'list':
  1292. return {'changed': False, 'results': api_rval, state: 'list'}
  1293. if not params['name']:
  1294. return {'failed': True,
  1295. 'msg': 'Please specify a name when state is absent|present.'}
  1296. ########
  1297. # Delete
  1298. ########
  1299. if state == 'absent':
  1300. if not Utils.exists(api_rval['results'], params['name']):
  1301. return {'changed': False, 'state': 'absent'}
  1302. if check_mode:
  1303. return {'changed': True, 'msg': 'Would have performed a delete.'}
  1304. api_rval = ocsecret.delete()
  1305. return {'changed': True, 'results': api_rval, 'state': 'absent'}
  1306. if state == 'present':
  1307. if params['files']:
  1308. files = params['files']
  1309. elif params['contents']:
  1310. files = Utils.create_tmp_files_from_contents(params['contents'])
  1311. else:
  1312. return {'failed': True,
  1313. 'msg': 'Either specify files or contents.'}
  1314. ########
  1315. # Create
  1316. ########
  1317. if not Utils.exists(api_rval['results'], params['name']):
  1318. if check_mode:
  1319. return {'changed': True,
  1320. 'msg': 'Would have performed a create.'}
  1321. api_rval = ocsecret.create(files, params['contents'])
  1322. # Remove files
  1323. if files and params['delete_after']:
  1324. Utils.cleanup([ftmp['path'] for ftmp in files])
  1325. if api_rval['returncode'] != 0:
  1326. return {'failed': True,
  1327. 'msg': api_rval}
  1328. return {'changed': True,
  1329. 'results': api_rval,
  1330. 'state': 'present'}
  1331. ########
  1332. # Update
  1333. ########
  1334. secret = ocsecret.prep_secret(params['files'], params['contents'])
  1335. if secret['returncode'] != 0:
  1336. return {'failed': True, 'msg': secret}
  1337. if Utils.check_def_equal(secret['results'], api_rval['results'][0]):
  1338. # Remove files
  1339. if files and params['delete_after']:
  1340. Utils.cleanup([ftmp['path'] for ftmp in files])
  1341. return {'changed': False,
  1342. 'results': secret['results'],
  1343. 'state': 'present'}
  1344. if check_mode:
  1345. return {'changed': True,
  1346. 'msg': 'Would have performed an update.'}
  1347. api_rval = ocsecret.update(files, force=params['force'])
  1348. # Remove files
  1349. if secret and params['delete_after']:
  1350. Utils.cleanup([ftmp['path'] for ftmp in files])
  1351. if api_rval['returncode'] != 0:
  1352. return {'failed': True,
  1353. 'msg': api_rval}
  1354. return {'changed': True,
  1355. 'results': api_rval,
  1356. 'state': 'present'}
  1357. return {'failed': True,
  1358. 'changed': False,
  1359. 'msg': 'Unknown state passed. %s' % state,
  1360. 'state': 'unknown'}
  1361. # -*- -*- -*- End included fragment: class/oc_secret.py -*- -*- -*-
  1362. # -*- -*- -*- Begin included fragment: ansible/oc_secret.py -*- -*- -*-
  1363. def main():
  1364. '''
  1365. ansible oc module for managing OpenShift Secrets
  1366. '''
  1367. module = AnsibleModule(
  1368. argument_spec=dict(
  1369. kubeconfig=dict(default='/etc/origin/master/admin.kubeconfig', type='str'),
  1370. state=dict(default='present', type='str',
  1371. choices=['present', 'absent', 'list']),
  1372. debug=dict(default=False, type='bool'),
  1373. namespace=dict(default='default', type='str'),
  1374. name=dict(default=None, type='str'),
  1375. files=dict(default=None, type='list'),
  1376. delete_after=dict(default=False, type='bool'),
  1377. contents=dict(default=None, type='list'),
  1378. force=dict(default=False, type='bool'),
  1379. decode=dict(default=False, type='bool'),
  1380. ),
  1381. mutually_exclusive=[["contents", "files"]],
  1382. supports_check_mode=True,
  1383. )
  1384. rval = OCSecret.run_ansible(module.params, module.check_mode)
  1385. if 'failed' in rval:
  1386. module.fail_json(**rval)
  1387. module.exit_json(**rval)
  1388. if __name__ == '__main__':
  1389. main()
  1390. # -*- -*- -*- End included fragment: ansible/oc_secret.py -*- -*- -*-