Improved close channel command.
This commit is contained in:
Mariusz B. / mgeeky
parent
6337d9418e
commit
18be9f3d9d
|
|
@ -543,7 +543,7 @@ def _onPing(args):
|
|||
ret = postRequest(f'/api/gateway/{gateway["agentId"]}/relay/{relay["agentId"]}/command', data)
|
||||
|
||||
if type(ret) == dict and 'relayAgentId' in ret.keys() and ret['relayAgentId'] == relay['agentId']:
|
||||
print(f'[.] Pinged relay: {relay["name"]:10s} from gateway {gateway["name"]}')
|
||||
print(f'[.] Pinged relay: {relay["name"]} (id: {relay["agentId"]}) from gateway {gateway["name"]}')
|
||||
pinged += 1
|
||||
|
||||
if pinged == 0:
|
||||
|
|
@ -1245,31 +1245,62 @@ def onCloseNetwork(args):
|
|||
closeNetwork(gateway)
|
||||
|
||||
def onCloseChannel(args):
|
||||
gateway, relay = findAgent(args.agent_id)
|
||||
if not relay and not gateway:
|
||||
Logger.fatal('Could not find agent (Gateway or Relay) which should be used to setup a channel.')
|
||||
gateways = getRequest('/api/gateway')
|
||||
channelsToClose = []
|
||||
|
||||
channelToClose = ''
|
||||
for _gateway in gateways:
|
||||
if len(args.gateway_id) > 0:
|
||||
if _gateway["agentId"].lower() != args.gateway_id.lower() and _gateway["name"].lower() != args.gateway_id.lower():
|
||||
continue
|
||||
|
||||
capability = processCapability(gateway)
|
||||
for chan in capability['channels'].keys():
|
||||
for a in sys.argv:
|
||||
if a.lower() == chan.lower():
|
||||
channelToClose = a
|
||||
break
|
||||
gateway = getRequest(f'/api/gateway/{_gateway["agentId"]}')
|
||||
capability = processCapability(gateway)
|
||||
|
||||
if len(channelToClose) > 0: break
|
||||
if len(args.gateway_id) > 0:
|
||||
if gateway["agentId"].lower() == args.agent_id.lower() or gateway["name"].lower() == args.agent_id.lower():
|
||||
for channel in gateway['channels']:
|
||||
name = list(capability['channels'].keys())[list(capability['channels'].values()).index(channel['type'])]
|
||||
if len(args.channel_id) == 0 or (name.lower() == args.channel_id.lower() or channel['iid'] == args.channel_id):
|
||||
_type = 'non-negotiation'
|
||||
if 'isReturnChannel' in channel.keys() and channel['isReturnChannel']: _type = 'grc'
|
||||
elif 'isNegotiationChannel' in channel.keys() and channel['isNegotiationChannel']: _type = 'negotiation'
|
||||
|
||||
if len(channelToClose) == 0:
|
||||
Logger.fatal('Couldnt identify which channel is to be closed. Specify your channel name in script parameters')
|
||||
channelsToClose.append({
|
||||
'url' : f'/api/gateway/{_gateway["agentId"]}/relay/{relay["agentId"]}/channel/{channel["iid"]}/command',
|
||||
'name' : name,
|
||||
'iid' : channel['iid'],
|
||||
'agent' : relay,
|
||||
'type' : _type,
|
||||
'kind' : 'Relay',
|
||||
})
|
||||
|
||||
channels = collectChannelsToSendCommand(args, channelToClose)
|
||||
for relay in gateway['relays']:
|
||||
if relay["agentId"].lower() != args.agent_id.lower() and relay["name"].lower() != args.agent_id.lower():
|
||||
continue
|
||||
|
||||
if len(channels) == 0:
|
||||
Logger.fatal("Could not find channel to be close. Adjust your agent ID/Name setting and try again.")
|
||||
for channel in relay['channels']:
|
||||
name = list(capability['channels'].keys())[list(capability['channels'].values()).index(channel['type'])]
|
||||
if len(args.channel_id) == 0 or (name.lower() == args.channel_id.lower() or channel['iid'] == args.channel_id):
|
||||
|
||||
for channel in channels:
|
||||
closeChannel(channel, channelToClose)
|
||||
_type = 'non-negotiation'
|
||||
if 'isReturnChannel' in channel.keys() and channel['isReturnChannel']: _type = 'grc'
|
||||
elif 'isNegotiationChannel' in channel.keys() and channel['isNegotiationChannel']: _type = 'negotiation'
|
||||
|
||||
channelsToClose.append({
|
||||
'url' : f'/api/gateway/{_gateway["agentId"]}/relay/{relay["agentId"]}/channel/{channel["iid"]}/command',
|
||||
'name' : name,
|
||||
'iid' : channel['iid'],
|
||||
'agent' : relay,
|
||||
'type' : _type,
|
||||
'kind' : 'Relay',
|
||||
})
|
||||
|
||||
if len(channelsToClose) == 0:
|
||||
Logger.fatal('Could not find channels that should have been closed. Try changing search criteria.')
|
||||
|
||||
for channel in channelsToClose:
|
||||
if channel['type'] == 'grc' and not args.close_grc: continue
|
||||
closeChannel(channel, channel['name'])
|
||||
|
||||
def onMattermostCreate(args):
|
||||
server_url = args.server_url
|
||||
|
|
@ -1856,7 +1887,9 @@ def parseArgs(argv):
|
|||
## Channel
|
||||
close_channel = parser_close_sub.add_parser('channel', help = 'Close a channel.')
|
||||
close_channel.add_argument('agent_id', metavar = 'agent_id', help = 'Gateway or Relay that will be used to find a channel to close. Can be ID or Name.')
|
||||
close_channel.add_argument('-c', '--channel-id', default='', help = 'Specifies ID of the channel to commander. If not given - will issue specified command to all channels in a Gateway/Relay.')
|
||||
close_channel.add_argument('-G', '--close-grc', action='store_true', help = 'Close Gateway-Return Channel (Non-negotiation one) as well. By default the GRC channel (the one marked with violet icon) will not be closed to avoid losing connectivity with relay.')
|
||||
close_channel.add_argument('-c', '--channel-id', default='', help = 'Specifies ID (or Name) of the channel to commander. If not given - will issue specified command to all channels in a Relay. If name is given, will update Jitter on all Channels with that name.')
|
||||
close_channel.add_argument('-g', '--gateway-id', default='', metavar='gateway_id', help = 'ID (or Name) of the Gateway which Relays should be pinged. If not given, will ping all relays in all gateways.')
|
||||
close_channel.set_defaults(func = onCloseChannel)
|
||||
|
||||
## Relay
|
||||
|
|
|
|||
Loading…
Reference in New Issue