83 lines
3.3 KiB
Python
83 lines
3.3 KiB
Python
"""
|
||
===
|
||
|
||
MIT License
|
||
|
||
Copyright (c) 2018 Dusty.P https://github.com/dustinpianalto
|
||
|
||
Permission is hereby granted, free of charge, to any person obtaining
|
||
a copy of this software and associated documentation files (the
|
||
"Software"), to deal in the Software without restriction, including
|
||
without limitation the rights to use, copy, modify, merge, publish,
|
||
distribute, sublicense, and/or sell copies of the Software, and to
|
||
permit persons to whom the Software is furnished to do so, subject to
|
||
the following conditions:
|
||
|
||
The above copyright notice and this permission notice shall be
|
||
included in all copies or substantial portions of the Software.
|
||
|
||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
||
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
||
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
||
NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
||
LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
||
OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
||
WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||
|
||
"""
|
||
|
||
|
||
import discord
|
||
from discord.ext import commands
|
||
import logging
|
||
from .imports.utils import paginate, run_command
|
||
import asyncio
|
||
|
||
git_log = logging.getLogger('git')
|
||
|
||
|
||
class Git:
|
||
def __init__(self, bot):
|
||
self.bot = bot
|
||
|
||
@commands.group(case_insensitive=True)
|
||
async def git(self, ctx):
|
||
"""Run help git for more info"""
|
||
pass
|
||
|
||
@git.command()
|
||
@commands.is_owner()
|
||
async def pull(self, ctx):
|
||
async with ctx.typing():
|
||
em = discord.Embed(style='rich',
|
||
title=f'Git Pull',
|
||
color=self.bot.embed_color)
|
||
em.set_thumbnail(url=f'{ctx.guild.me.avatar_url}')
|
||
result = await asyncio.wait_for(self.bot.loop.create_task(run_command('git fetch --all')), 120) + '\n'
|
||
result += await asyncio.wait_for(self.bot.loop.create_task(run_command('git reset --hard '
|
||
'origin/$(git '
|
||
'rev-parse --symbolic-full-name '
|
||
'--abbrev-ref HEAD)')), 120) + '\n\n'
|
||
result += await asyncio.wait_for(self.bot.loop.create_task(run_command('git show --stat | '
|
||
'sed "s/.*@.*[.].*/ /g"')), 10)
|
||
results = paginate(result, maxlen=1014)
|
||
for page in results[:5]:
|
||
em.add_field(name='', value=f'{page}')
|
||
await ctx.send(embed=em)
|
||
|
||
@git.command()
|
||
@commands.is_owner()
|
||
async def status(self, ctx):
|
||
em = discord.Embed(style='rich',
|
||
title=f'Git Pull',
|
||
color=self.bot.embed_color)
|
||
em.set_thumbnail(url=f'{ctx.guild.me.avatar_url}')
|
||
result = await asyncio.wait_for(self.bot.loop.create_task(run_command('git status')), 10)
|
||
results = paginate(result, maxlen=1014)
|
||
for page in results[:5]:
|
||
em.add_field(name='', value=f'{page}')
|
||
await ctx.send(embed=em)
|
||
|
||
|
||
def setup(bot):
|
||
bot.add_cog(Git(bot)) |