IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an email to s dot adaszewski at gmail dot com. User accounts are meant only to report issues and/or generate pull requests. This is a purpose-specific Git hosting for ADARED projects. Thank you for your understanding!
Vous ne pouvez pas sélectionner plus de 25 sujets Les noms de sujets doivent commencer par une lettre ou un nombre, peuvent contenir des tirets ('-') et peuvent comporter jusqu'à 35 caractères.

212 lignes
8.2KB

  1. #
  2. # Copyright (C) Stanislaw Adaszewski, 2020
  3. # License: GNU General Public License v3.0
  4. # URL: https://github.com/sadaszewski/focker
  5. # URL: https://adared.ch/focker
  6. #
  7. from argparse import ArgumentParser
  8. import argparse
  9. import yaml
  10. import os
  11. from functools import reduce
  12. # from weir import zfs, process
  13. from .image import command_image_build, \
  14. command_image_tag, \
  15. command_image_untag, \
  16. command_image_list, \
  17. command_image_prune, \
  18. command_image_remove
  19. from .volume import command_volume_create, \
  20. command_volume_prune, \
  21. command_volume_list, \
  22. command_volume_tag, \
  23. command_volume_untag
  24. import sys
  25. from .zfs import zfs_init
  26. from .jail import command_jail_create, \
  27. command_jail_start, \
  28. command_jail_stop, \
  29. command_jail_remove, \
  30. command_jail_exec, \
  31. command_jail_oneshot, \
  32. command_jail_list, \
  33. command_jail_tag, \
  34. command_jail_untag, \
  35. command_jail_prune
  36. from .compose import \
  37. command_compose_build, \
  38. command_compose_run
  39. class ListForwarderFunctor(object):
  40. def __init__(self, lst):
  41. self.lst = lst
  42. def __call__(self, *args, **kwargs):
  43. res = []
  44. for elem in self.lst:
  45. res.append(elem(*args, **kwargs))
  46. return ListForwarder(res)
  47. class ListForwarder(object):
  48. def __init__(self, lst):
  49. self.lst = lst
  50. def __getattr__(self, name):
  51. return ListForwarderFunctor(list(map(lambda a: getattr(a, name), self.lst)))
  52. def __setattr__(self, name, value):
  53. if name == 'lst':
  54. super().__setattr__(name, value)
  55. return
  56. # print('setattr(), name:', name, 'value:', value)
  57. for elem in self.lst:
  58. setattr(elem, name, value)
  59. def create_parser():
  60. parser_top = ArgumentParser()
  61. subparsers_top = parser_top.add_subparsers(dest='L1_command')
  62. subparsers_top.required = True
  63. # image
  64. subparsers = ListForwarder([ subparsers_top.add_parser(cmd).add_subparsers(dest='L2_command') \
  65. for cmd in ['image', 'img', 'im', 'i'] ])
  66. subparsers.required = True
  67. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['build', 'b']])
  68. parser.set_defaults(func=command_image_build)
  69. parser.add_argument('focker_dir', type=str)
  70. parser.add_argument('--tags', '-t', type=str, nargs='+', default=[])
  71. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['tag', 't']])
  72. parser.set_defaults(func=command_image_tag)
  73. parser.add_argument('reference', type=str)
  74. parser.add_argument('tags', type=str, nargs='+')
  75. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['untag', 'u']])
  76. parser.set_defaults(func=command_image_untag)
  77. parser.add_argument('tags', type=str, nargs='+', default=[])
  78. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['list', 'ls', 'l']])
  79. parser.set_defaults(func=command_image_list)
  80. parser.add_argument('--full-sha256', '-f', action='store_true')
  81. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['prune', 'p']])
  82. parser.set_defaults(func=command_image_prune)
  83. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['remove', 'r']])
  84. parser.set_defaults(func=command_image_remove)
  85. parser.add_argument('reference', type=str)
  86. # parser.add_argument('--remove-children', '-r', action='store_true')
  87. parser.add_argument('--remove-dependents', '-R', action='store_true')
  88. # jail
  89. subparsers = ListForwarder([ subparsers_top.add_parser(cmd).add_subparsers(dest='L2_command') \
  90. for cmd in ['jail', 'j'] ])
  91. subparsers.required = True
  92. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['create', 'c']])
  93. parser.set_defaults(func=command_jail_create)
  94. parser.add_argument('image', type=str)
  95. parser.add_argument('--command', '-c', type=str, default='/bin/sh')
  96. parser.add_argument('--tags', '-t', type=str, nargs='+', default=[])
  97. parser.add_argument('--env', '-e', type=str, nargs='+', default=[])
  98. parser.add_argument('--mounts', '-m', type=str, nargs='+', default=[])
  99. parser.add_argument('--hostname', '-n', type=str)
  100. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['start', 's']])
  101. parser.set_defaults(func=command_jail_start)
  102. parser.add_argument('reference', type=str)
  103. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['stop', 'S']])
  104. parser.set_defaults(func=command_jail_stop)
  105. parser.add_argument('reference', type=str)
  106. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['remove', 'r']])
  107. parser.set_defaults(func=command_jail_remove)
  108. parser.add_argument('reference', type=str)
  109. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['exec', 'e']])
  110. parser.set_defaults(func=command_jail_exec)
  111. parser.add_argument('reference', type=str)
  112. parser.add_argument('command', type=str, nargs=argparse.REMAINDER, default=['/bin/sh'])
  113. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['oneshot', 'o']])
  114. parser.set_defaults(func=command_jail_oneshot)
  115. parser.add_argument('image', type=str)
  116. parser.add_argument('--env', '-e', type=str, nargs='+', default=[])
  117. parser.add_argument('--mounts', '-m', type=str, nargs='+', default=[])
  118. parser.add_argument('command', type=str, nargs=argparse.REMAINDER, default=['/bin/sh'])
  119. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['list', 'ls', 'l']])
  120. parser.set_defaults(func=command_jail_list)
  121. parser.add_argument('--full-sha256', '-f', action='store_true')
  122. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['tag', 't']])
  123. parser.set_defaults(func=command_jail_tag)
  124. parser.add_argument('reference', type=str)
  125. parser.add_argument('tags', type=str, nargs='+')
  126. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['untag', 'u']])
  127. parser.set_defaults(func=command_jail_untag)
  128. parser.add_argument('tags', type=str, nargs='+')
  129. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['prune', 'p']])
  130. parser.set_defaults(func=command_jail_prune)
  131. parser.add_argument('--force', '-f', action='store_true')
  132. # volume
  133. subparsers = ListForwarder([ subparsers_top.add_parser(cmd).add_subparsers(dest='L2_command') \
  134. for cmd in ['volume', 'vol', 'v'] ])
  135. subparsers.required = True
  136. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['create', 'c']])
  137. parser.set_defaults(func=command_volume_create)
  138. parser.add_argument('--tags', '-t', type=str, nargs='+', default=[])
  139. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['prune', 'p']])
  140. parser.set_defaults(func=command_volume_prune)
  141. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['list', 'ls', 'l']])
  142. parser.set_defaults(func=command_volume_list)
  143. parser.add_argument('--full-sha256', '-f', action='store_true')
  144. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['tag', 't']])
  145. parser.set_defaults(func=command_volume_tag)
  146. parser.add_argument('reference', type=str)
  147. parser.add_argument('tags', type=str, nargs='+')
  148. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['untag', 'u']])
  149. parser.set_defaults(func=command_volume_untag)
  150. parser.add_argument('tags', type=str, nargs='+')
  151. # compose
  152. subparsers = ListForwarder([ subparsers_top.add_parser(cmd).add_subparsers(dest='L2_command') \
  153. for cmd in ['compose', 'comp', 'c'] ])
  154. subparsers.required = True
  155. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['build', 'b']])
  156. parser.set_defaults(func=command_compose_build)
  157. parser.add_argument('filename', type=str)
  158. parser = ListForwarder([subparsers.add_parser(cmd) for cmd in ['run', 'r']])
  159. parser.set_defaults(func=command_compose_run)
  160. parser.add_argument('filename', type=str)
  161. parser.add_argument('command', type=str)
  162. return parser_top
  163. def main():
  164. zfs_init()
  165. parser = create_parser()
  166. args = parser.parse_args()
  167. if not hasattr(args, 'func'):
  168. parser.print_usage()
  169. sys.exit('You must choose a mode')
  170. args.func(args)
  171. if __name__ == '__main__':
  172. main()