import { TokenFlow } from "../flow";

export class VerticalFlow extends TokenFlow {
  verticalCandidates(
    direction: "up" | "down",
    fromIndex: number,
    length: number
  ): number[] {
    if (direction === "up") {
      return fromIndex > 0 ? [fromIndex - 1] : [];
    } else {
      return fromIndex < length - 1 ? [fromIndex + 1] : [];
    }
  }
}