override fun doFunction()

in vim-engine/src/main/kotlin/com/maddyhome/idea/vim/vimscript/model/functions/handlers/collectionFunctions/ExtendFunctionHandlers.kt [29:83]

45 lines of code
13 McCabe index (conditional complexity)


  override fun doFunction(
    arguments: Arguments,
    editor: VimEditor,
    context: ExecutionContext,
    vimContext: VimLContext,
  ): VimDataType {
    val argument1 = arguments[0]
    val argument2 = arguments[1]

    if (argument1 is VimList && argument2 is VimList) {
      val list = if (makeCopy) VimList(argument1.values.toMutableList()) else argument1
      if (list.isLocked) {
        throw exExceptionMessage("E741", "$name() argument")
      }
      val index = arguments.getNumberOrNull(2)?.value ?: list.size
      list.values.addAll(index, argument2.values)
      return list
    }

    if (argument1 is VimDictionary && argument2 is VimDictionary) {
      val argument3 = arguments.getStringOrNull(2)?.value?.lowercase() ?: "force"
      if (argument3 !in listOf("force", "keep", "error")) {
        throw exExceptionMessage("E475", argument3)
      }

      val dict = if (makeCopy) VimDictionary(LinkedHashMap(argument1.dictionary)) else argument1

      if (dict.isLocked) {
        throw exExceptionMessage("E741", "$name() argument")
      }

      argument2.dictionary.forEach { (key, value) ->
        if (dict.dictionary.containsKey(key)) {
          if (dict.dictionary[key]?.isLocked == true) {
            throw exExceptionMessage("E741", "$name() argument")
          }

          when (argument3) {
            "force" -> dict.dictionary[key] = value
            "keep" -> { /* Do nothing. Keep the value in argument1 */
            }

            "error" -> throw exExceptionMessage("E737", key.toOutputString())
          }
        } else {
          dict.dictionary[key] = value
        }
      }

      // TODO: Should return 0 on error. However, IdeaVim does not support this and can only throw errors
      return dict
    }

    throw exExceptionMessage("E712", "$name()")
  }