spring-shell/src/main/java/com/baeldung/shell/simple/SimpleCLI.java - master - NikolayIvkin/tutorials1

spring-shell/src/main/java/com/baeldung/shell/simple/SimpleCLI.java

70 строк2 KB

Krzysiek

BAEL-19967: Migrate spring-shell module to the com.baeldung package

18 дек 2019, 23:35

18 дек 2019, 23:358b17262

package com.baeldung.shell.simple;

import java.io.BufferedReader;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.URL;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.springframework.shell.Bootstrap;
import org.springframework.shell.core.CommandMarker;
import org.springframework.shell.core.annotation.CliAvailabilityIndicator;
import org.springframework.shell.core.annotation.CliCommand;
import org.springframework.shell.core.annotation.CliOption;
import org.springframework.stereotype.Component;

@Component
public class SimpleCLI implements CommandMarker {

    private String getContentsOfUrlAsString(URL url) {
        StringBuilder sb = new StringBuilder();
        try {
            try (BufferedReader in = new BufferedReader(new InputStreamReader(url.openStream()))) {
                String inputLine;
                while ((inputLine = in.readLine()) != null) {
                    sb.append(inputLine);
                }
            }
        } catch (IOException ex) {
            sb.append("ERROR");
        }
        return sb.toString();
    }

    @CliCommand(value = { "web-get", "wg" }, help = "Displays the contents of a URL.")
    public String webGet(@CliOption(key = { "", "url" }, help = "URL whose contents will be displayed.") URL url) {
        return getContentsOfUrlAsString(url);
    }

    @CliCommand(value = { "web-save", "ws" }, help = "Saves the contents of a URL.")
    public String webSave(@CliOption(key = { "", "url" }, help = "URL whose contents will be saved.") URL url, @CliOption(key = { "out", "file" }, mandatory = true, help = "The name of the file.") String file) {
        String contents = getContentsOfUrlAsString(url);
        try (PrintWriter out = new PrintWriter(file)) {
            out.write(contents);
        } catch (FileNotFoundException ex) {
            // Ignore
        }
        return "Done.";
    }

    private boolean adminEnableExecuted = false;

    @CliAvailabilityIndicator(value = { "web-save" })
    public boolean isAdminEnabled() {
        return adminEnableExecuted;
    }

    @CliCommand(value = "admin-enable")
    public String adminEnable() {
        adminEnableExecuted = true;
        return "Admin commands enabled.";
    }

    @CliCommand(value = "admin-disable")
    public String adminDisable() {
        adminEnableExecuted = false;
        return "Admin commands disabled.";
    }
}