"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = void 0;

var _core = require("@kui-shell/core");

var _util = require("./util/util");

/*
 * Copyright 2019 IBM Corporation
 *
 * 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 = void 0 && (void 0).__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());
  });
};

/**
 * 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 = (0, _util.getCommandFromArgs)({
      argvNoOptions
    });
    const verb = argvNoOptions[1]; // index of the arg just before the one to be completed

    const previous = spec.toBeCompletedIdx === -1 ? commandLine.argv.length - 1 : spec.toBeCompletedIdx - 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') && (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
 *
 */


var _default = () => {
  (0, _core.registerTabCompletionEnumerator)(completeResourceNames);
};

exports.default = _default;