Skip to content

Instantly share code, notes, and snippets.

@p4p1
Last active April 14, 2020 14:55
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save p4p1/a434a526961f4107bcc4b33e5748b97f to your computer and use it in GitHub Desktop.
Save p4p1/a434a526961f4107bcc4b33e5748b97f to your computer and use it in GitHub Desktop.
Bas c'est blih enfaite, changer la variable passwd
#!/usr/bin/env python3
#-
# Copyright 2013-2014 Emmanuel Vadot <elbarto@bocal.org>
# All rights reserved
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted providing that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
# IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
import os
import sys
import getopt
import hmac
import hashlib
import urllib.request
import urllib.parse
import json
import getpass
version = 1.7
class blih:
def __init__(self, baseurl='https://blih.epitech.eu/', user=None, token=None, passwd=None, verbose=False, user_agent='blih-' + str(version)):
self._baseurl = baseurl
self.passwd = passwd
if token:
self._token = token
else:
self.token_calc()
if user == None:
self._user = getpass.getuser()
else:
self._user = user
self._verbose = verbose
self._useragent = user_agent
def token_get(self):
return self._token
def token_set(self, token):
self._token = token
token = property(token_get, token_set)
def token_calc(self):
try:
self._token = bytes(hashlib.sha512(bytes(self.passwd, 'utf8')).hexdigest(), 'utf8')
except:
self._token = bytes(hashlib.sha512(bytes(getpass.getpass(), 'utf8')).hexdigest(), 'utf8')
def sign_data(self, data=None):
signature = hmac.new(self._token, msg=bytes(self._user, 'utf8'), digestmod=hashlib.sha512)
if data:
signature.update(bytes(json.dumps(data, sort_keys=True, indent=4, separators=(',', ': ')), 'utf8'))
signed_data = {'user' : self._user, 'signature' : signature.hexdigest()}
if data != None:
signed_data['data'] = data
return signed_data
def request(self, resource, method='GET', content_type='application/json', data=None, url=None):
signed_data = self.sign_data(data)
if url:
req = urllib.request.Request(url=url, method=method, data=bytes(json.dumps(signed_data), 'utf8'))
else:
req = urllib.request.Request(url=self._baseurl + resource, method=method, data=bytes(json.dumps(signed_data), 'utf8'))
req.add_header('Content-Type', content_type)
req.add_header('User-Agent', self._useragent)
try:
f = urllib.request.urlopen(req)
except urllib.error.HTTPError as e:
print ('HTTP Error ' + str(e.code))
data = json.loads(e.read().decode('utf8'))
print ("Error message : '" + data['error'] + "'")
sys.exit(1)
if f.status == 200:
try:
data = json.loads(f.read().decode('utf8'))
except:
print ("Can't decode data, aborting")
sys.exit(1)
return (f.status, f.reason, f.info(), data)
print ('Unknown error')
sys.exit(1)
def repo_create(self, name, type='git', description=None):
data = {'name' : name, 'type' : type}
if description:
data['description'] = description
status, reason, headers, data = self.request('/repositories', method='POST', data=data)
print (data['message'])
def repo_list(self):
status, reason, headers, data = self.request('/repositories', method='GET')
for i in data['repositories']:
print (i)
def repo_delete(self, name):
status, reason, headers, data = self.request('/repository/' + name, method='DELETE')
print (data['message'])
def repo_info(self, name):
status, reason, headers, data = self.request('/repository/' + name, method='GET')
print (data['message'])
def repo_setacl(self, name, acluser, acl):
data = {'user' : acluser, 'acl' : acl}
status, reason, headers, data = self.request('/repository/' + name + '/acls', method='POST', data=data)
print (data['message'])
def repo_getacl(self, name):
status, reason, headers, data = self.request('/repository/' + name + '/acls', method='GET')
for i in data.keys():
print (i + ':' + data[i])
def sshkey_upload(self, keyfile):
try:
f = open(keyfile, 'r')
except (PermissionError, FileNotFoundError):
print ("Can't open file : " + keyfile)
return
key = urllib.parse.quote(f.read().strip('\n'))
f.close()
data = {'sshkey' : key}
status, reason, headers, data = self.request('/sshkeys', method='POST', data=data)
print (data['message'])
def sshkey_delete(self, sshkey):
status, reason, headers, data = self.request('/sshkey/' + sshkey, method='DELETE')
print (data['message'])
def sshkey_list(self):
status, reason, headers, data = self.request('/sshkeys', method='GET')
for i in data.keys():
print (data[i] + ' ' + i)
def whoami(self):
status, reason, headers, data = self.request('/whoami', method='GET')
print (data['message'])
def usage_repository():
print ('Usage: ' + sys.argv[0] + ' [options] repository command ...')
print ()
print ('Commands :')
print ('\tcreate repo\t\t\t-- Create a repository named "repo"')
print ('\tinfo repo\t\t\t-- Get the repository metadata')
print ('\tgetacl repo\t\t\t-- Get the acls set for the repository')
print ('\tlist\t\t\t\t-- List the repositories created')
print ('\tsetacl repo user [acl]\t\t-- Set (or remove) an acl for "user" on "repo"')
print ('\t\t\t\t\tACL format:')
print ('\t\t\t\t\tr for read')
print ('\t\t\t\t\tw for write')
print ('\t\t\t\t\ta for admin')
sys.exit(1)
def repository(args, baseurl, user, token, passwd, verbose, user_agent):
if len(args) == 0:
usage_repository()
if args[0] == 'create':
if len(args) != 2:
usage_repository()
handle = blih(baseurl=baseurl, user=user, token=token, passwd=passwd, verbose=verbose, user_agent=user_agent)
handle.repo_create(args[1])
elif args[0] == 'list':
if len(args) != 1:
usage_repository()
handle = blih(baseurl=baseurl, user=user, token=token, passwd=passwd, verbose=verbose, user_agent=user_agent)
handle.repo_list()
elif args[0] == 'info':
if len(args) != 2:
usage_repository()
handle = blih(baseurl=baseurl, user=user, token=token, passwd=passwd, verbose=verbose, user_agent=user_agent)
handle.repo_info(args[1])
elif args[0] == 'delete':
if len(args) != 2:
usage_repository()
handle = blih(baseurl=baseurl, user=user, token=token, passwd=passwd, verbose=verbose, user_agent=user_agent)
handle.repo_delete(args[1])
elif args[0] == 'setacl':
if len(args) != 4 and len(args) != 3:
usage_repository()
if len(args) == 3:
acl = ''
else:
acl = args[3]
handle = blih(baseurl=baseurl, user=user, token=token, passwd=passwd, verbose=verbose, user_agent=user_agent)
handle.repo_setacl(args[1], args[2], acl)
elif args[0] == 'getacl':
if len(args) != 2:
usage_repository()
handle = blih(baseurl=baseurl, user=user, token=token, passwd=passwd, verbose=verbose, user_agent=user_agent)
handle.repo_getacl(args[1])
else:
usage_repository()
def usage_sshkey():
print ('Usage: ' + sys.argv[0] + ' [options] sshkey command ...')
print ()
print ('Commands :')
print ('\tupload [file]\t\t\t-- Upload a new ssh-key')
print ('\tlist\t\t\t\t-- List the ssh-keys')
print ('\tdelete <sshkey>\t\t\t-- Delete the sshkey with comment <sshkey>')
sys.exit(1)
def sshkey(args, baseurl, user, token, passwd, verbose, user_agent):
if len(args) == 0:
usage_sshkey()
if args[0] == 'list':
handle = blih(baseurl=baseurl, user=user, token=token, passwd=passwd, verbose=verbose, user_agent=user_agent)
handle.sshkey_list()
elif args[0] == 'upload':
key = None
if len(args) == 1:
key = os.getenv('HOME') + '/.ssh/id_rsa.pub'
elif len(args) == 2:
key = args[1]
else:
usage_sshkey()
handle = blih(baseurl=baseurl, user=user, token=token, passwd=passwd, verbose=verbose, user_agent=user_agent)
handle.sshkey_upload(key)
elif args[0] == 'delete':
if len(args) != 2:
usage_sshkey()
handle = blih(baseurl=baseurl, user=user, token=token, passwd=passwd, verbose=verbose, user_agent=user_agent)
handle.sshkey_delete(args[1])
else:
usage_sshkey()
def whoami(args, baseurl, user, token, passwd, verbose, user_agent):
handle = blih(baseurl=baseurl, user=user, token=token, passwd=passwd, verbose=verbose, user_agent=user_agent)
handle.whoami()
def usage():
print ('Usage: ' + sys.argv[0] + ' [options] command ...')
print ()
print ('Global Options :')
print ('\t-u user | --user=user\t\t-- Run as user')
print ('\t-v | --verbose\t\t\t-- Verbose')
print ('\t-b url | --baseurl=url\t\t-- Base URL for BLIH')
print ('\t-t | --token\t\t\t-- Specify token in the cmdline')
print ()
print ('Commands :')
print ('\trepository\t\t\t-- Repository management')
print ('\tsshkey\t\t\t\t-- SSH-KEYS management')
print ('\twhoami\t\t\t\t-- Print who you are')
sys.exit(1)
if __name__ == "__main__":
try:
opts, args = getopt.getopt(sys.argv[1:], 'hvu:b:t:VU:', ['help', 'verbose', 'user=', 'baseurl=', 'token=', 'version', 'useragent='])
except getopt.GetoptError as e:
print (e)
usage()
verbose = False
user = None
baseurl = 'https://blih.epitech.eu/'
token = None
user_agent = 'blih-' + str(version)
print("%s tes vraiment trop fort <3" % getpass.getuser())
passwd = getpass.getpass()
for o, a in opts:
if o in ('-h', '--help'):
usage()
elif o in ('-v', '--verbose'):
verbose = True
elif o in ('-u', '--user'):
user = a
elif o in ('-b', '--baseurl'):
baseurl = a
elif o in ('-t', '--token'):
token = bytes(a, 'utf8')
elif o in ('-V', '--version'):
print ('blih version ' + str(version))
sys.exit(0)
elif o in ('-U', '--useragent'):
user_agent = a
else:
usage()
if len(args) == 0:
usage()
if args[0] == 'repository':
repository(args[1:], baseurl, user, token, passwd, verbose, user_agent)
elif args[0] == 'sshkey':
sshkey(args[1:], baseurl, user, token, passwd, verbose, user_agent)
elif args[0] == 'whoami':
whoami(args[1:], baseurl, user, token, passwd, verbose, user_agent)
else:
usage()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment