io.resys.wrench.assets.flow.spi.hints.task.TaskAutocomplete Maven / Gradle / Ivy

Go to download
package io.resys.wrench.assets.flow.spi.hints.task;

/*-
 * #%L
 * wrench-assets-flow
 * %%
 * Copyright (C) 2016 - 2019 Copyright 2016 ReSys OÜ
 * %%
 * 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.
 * #L%
 */

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import io.resys.wrench.assets.flow.api.FlowAstFactory.Node;
import io.resys.wrench.assets.flow.api.FlowAstFactory.NodeFlow;
import io.resys.wrench.assets.flow.api.model.FlowAst.FlowCommandRange;
import io.resys.wrench.assets.flow.api.model.FlowAst.NodeFlowVisitor;
import io.resys.wrench.assets.flow.api.model.ImmutableFlowAst;
import io.resys.wrench.assets.flow.spi.model.NodeFlowBean;
import io.resys.wrench.assets.flow.spi.support.FlowNodesFactory;

public class TaskAutocomplete implements NodeFlowVisitor {

  @Override
  public void visit(NodeFlow flow, ImmutableFlowAst.Builder modelBuilder) {
    Node tasks = flow.get(NodeFlowBean.KEY_TASKS);
    if(tasks == null) {
      return;
    }

    List taskChildren = new ArrayList<>(flow.getTasks().values());
    Collections.sort(taskChildren);

    List ranges = new ArrayList<>();
    int previous = tasks.getStart();
    for(Node child : taskChildren) {

      int start = child.getStart() - 1;
      if(previous <= start) {
        ranges.add(FlowNodesFactory.range().build(previous, start));
      }
      previous = child.getEnd() + 1;
    }

    if(flow.getEnd() >= previous) {
      ranges.add(FlowNodesFactory.range().build(previous, flow.getEnd()));
    }

    modelBuilder.addAutocomplete(
        FlowNodesFactory.ac()
        .id(TaskAutocomplete.class.getSimpleName())
        .addField(2, "- {name}")
        .addField(6, "id", "{id}")
        .addField(6, "then", "{then}")
        .addField(6, "{type}")
        .addField(8, "ref", "{ref}")
        .addField(8, "collection", "{collection}")
        .addField(8, "inputs")
        .addRange(ranges)
        .build());

  }
}