@kui-shell/plugin-kubectl
Version:
Kubernetes visualization plugin for kubernetes
101 lines • 4.58 kB
JavaScript
/*
* Copyright 2019 The Kubernetes Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
import { registerTabCompletionEnumerator } from '@kui-shell/core';
import { getCommandFromArgs } from './util/util';
/**
* Invoke an enumeration command and return the filtered list of matching strings
*
*/
function getMatchingStrings(tab, cmd, spec) {
return __awaiter(this, void 0, void 0, function* () {
const completions = yield tab.REPL.qexec(cmd, undefined, undefined, { raw: true });
const list = completions.split(/[\n\r]/).map(_ => _.replace(/^\w+\//, ''));
return list.filter(name => name.startsWith(spec.toBeCompleted)).map(name => name.substring(spec.toBeCompleted.length));
});
}
/**
* Strip off the ParsedOptions in a way that lets us make an enumeration query safely
*
*/
function optionals(commandLine, filter = () => true) {
const options = commandLine.parsedOptions;
return Object.keys(options)
.filter(filter)
.filter(_ => !/^(-o|--output)/.test(_)) // remove any existing -o, because we want to use -o name
.map(key => `${key.length === 1 ? `-${key}` : `--${key}`} ${options[key]}`)
.join(' ');
}
function isPodSpecific(verb) {
return verb === 'attach' || verb === 'exec' || verb === 'logs';
}
function isNodeSpecific(verb) {
return verb === 'cordon' || verb === 'uncordon' || verb === 'drain' || verb === 'taint';
}
/**
* Tab completion of kube resource names
*
*/
function completeResourceNames(tab, commandLine, spec) {
return __awaiter(this, void 0, void 0, function* () {
const { argvNoOptions, argv, parsedOptions } = commandLine;
const command = getCommandFromArgs({ argvNoOptions });
// index of the arg just before the one to be completed
const previous = spec.toBeCompletedIdx === -1 ? commandLine.argv.length - 1 : spec.toBeCompletedIdx - 1;
const verb = argvNoOptions[1];
if (previous > 0 && (argv[previous] === '-n' || argv[previous] === '--namespace')) {
//
// then we are being asked to complete a namespace
//
const cmd = `${command} get ns ${optionals(commandLine, _ => _ !== '-n' && _ !== '--namespace')} -o name`;
return getMatchingStrings(tab, cmd, spec);
}
else if ((argvNoOptions[0] === 'kubectl' || argvNoOptions[0] === 'k' || argvNoOptions[0] === 'oc') &&
!/^-/.test(argv[previous]) &&
(verb === 'get' ||
verb === 'annotate' ||
verb === 'describe' ||
verb === 'pod' ||
verb === 'edit' ||
verb === 'label' ||
isPodSpecific(verb) ||
isNodeSpecific(verb) ||
(verb === 'delete' && !parsedOptions.f && !parsedOptions.file))) {
//
// then we are being asked to complete a resource name
//
const entityType = isPodSpecific(verb) ? 'pod' : isNodeSpecific(verb) ? 'node' : argvNoOptions[2];
const cmd = `${command} get ${entityType} ${optionals(commandLine)} -o name`;
return getMatchingStrings(tab, cmd, spec);
}
});
}
/**
* Entry point to register kubernetes tab completion
*
*/
export default () => {
registerTabCompletionEnumerator(completeResourceNames);
};
//# sourceMappingURL=tab-completion.js.map