net.oneandone.stool.cli.Select Maven / Gradle / Ivy

Go to download
/*
 * Copyright 1&1 Internet AG, https://github.com/1and1/
 *
 * 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.
 */
package net.oneandone.stool.cli;

import net.oneandone.stool.locking.Mode;
import net.oneandone.stool.util.Session;
import net.oneandone.sushi.fs.file.FileNode;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

public class Select extends SessionCommand {
    private final boolean fuzzy;
    private final String stageName;

    public Select(Session session, boolean fuzzy, String stageName) {
        super(session, Mode.NONE);
        this.fuzzy = fuzzy;
        this.stageName = stageName;
    }

    @Override
    public void doRun() throws Exception {
        FileNode dir;

        if (stageName == null) {
            console.info.println("currently selected: "
                    + (session.getSelectedStageId() == null ? "none" : session.getSelectedStageId()));
            return;
        }
        if ("none".equals(stageName)) {
            if (session.getSelectedStageId() == null) {
                console.info.println("already selected: none");
            } else {
                console.verbose.println("selecting none");
                session.cd(world.getWorking().getParent());
            }
            return;
        }
        dir = session.lookup(stageName);
        if (dir == null) {
            List candidates;

            candidates = candidates(session.stageNames(), stageName);
            switch (candidates.size()) {
                case 0:
                    throw new IOException("No such stage: " + stageName);
                case 1:
                    if (fuzzy) {
                        dir = session.lookup(candidates.get(0));
                        console.info.println("assuming you mean " + candidates.get(0));
                        break;
                    } else {
                        throw new IOException("No such stage: " + stageName + "\nDid you mean " + candidates.get(0) + "?");
                    }
                default:
                    throw new IOException("No such stage: " + stageName + "\nDid you mean one of " + candidates + "?");
            }
        }
        session.cd(dir);
    }

    public static List candidates(List names, String search) {
        String reduced;
        List result;

        result = new ArrayList<>();
        reduced = reduce(search);
        for (String name : names) {
            if (reduce(name).contains(reduced)) {
                result.add(name);
            }
        }
        return result;
    }

    private static String reduce(String name) {
        StringBuilder builder;
        char c;

        builder = new StringBuilder();
        for (int i = 0, max = name.length(); i < max; i++) {
            c = name.charAt(i);
            if (c >= '0' && c <= '9') {
                builder.append(c);
            } else if (c >= 'a' && c <= 'z') {
                builder.append(c);
            } else if (c >= 'A' && c <= 'Z') {
                builder.append(Character.toLowerCase(c));
            }
        }
        return builder.toString();
    }
}