"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.printDoBlock = exports.printBraceBlock = exports.printBlockVar = void 0;
const prettier_1 = __importDefault(require("../../prettier"));
const utils_1 = require("../../utils");
const { breakParent, group, ifBreak, indent, join, removeLines, softline } = prettier_1.default;
const printBlockVar = (path, opts, print) => {
    const parts = ["|", removeLines(path.call(print, "params"))];
    // The second part of this node is a list of optional block-local variables
    if (path.getValue().locals.length > 0) {
        parts.push("; ", join(", ", path.map(print, "locals")));
    }
    parts.push("| ");
    return parts;
};
exports.printBlockVar = printBlockVar;
// You have to go through the main print function if you could potentially have
// comments attached. So we're doing this weird reflection on the printed docs
// to retroactively change the printed keyword depending on if we're using
// braces or not. Ideally we wouldn't do this, we would instead do this
// reflection in the child printer, but this keeps the logic to just this file
// and contains it, so keeping it here for now.
function printBlockBegin(path, print, useBraces) {
    let docs = print(path);
    const doc = useBraces && !path.getValue().comments ? "{" : "do";
    if (Array.isArray(docs)) {
        docs[1] = doc;
    }
    else {
        docs = doc;
    }
    return docs;
}
function printBlock(braces) {
    return function printBlockWithBraces(path, opts, print) {
        const node = path.getValue();
        const stmts = node.type === "brace_block" ? node.stmts.body : node.bodystmt.stmts.body;
        let doBlockBody = "";
        if (stmts.length !== 1 ||
            stmts[0].type !== "void_stmt" ||
            stmts[0].comments) {
            doBlockBody = indent([
                softline,
                path.call(print, node.type === "brace_block" ? "stmts" : "bodystmt")
            ]);
        }
        // If this block is nested underneath a command or command_call node, then
        // we can't use `do...end` because that will get associated with the parent
        // node as opposed to the current node (because of the difference in
        // operator precedence). Instead, we still use a multi-line format but
        // switch to using braces instead.
        const useBraces = braces && (0, utils_1.hasAncestor)(path, ["command", "command_call"]);
        const doBlock = [
            " ",
            path.call((beginPath) => printBlockBegin(beginPath, print, useBraces), node.type === "brace_block" ? "lbrace" : "keyword"),
            node.block_var ? [" ", path.call(print, "block_var")] : "",
            doBlockBody,
            [softline, useBraces ? "}" : "end"]
        ];
        // We can hit this next pattern if within the block the only statement is a
        // comment.
        if (stmts.length === 1 &&
            stmts[0].type === "void_stmt" &&
            stmts[0].comments) {
            return [breakParent, doBlock];
        }
        const blockReceiver = path.getParentNode().call;
        // If the parent node is a command node, then there are no parentheses
        // around the arguments to that command, so we need to break the block
        if (["command", "command_call"].includes(blockReceiver.type)) {
            return [breakParent, doBlock];
        }
        const hasBody = stmts.some(({ type }) => type !== "void_stmt");
        const braceBlock = [
            " ",
            path.call((beginPath) => printBlockBegin(beginPath, print, true), node.type === "brace_block" ? "lbrace" : "keyword"),
            hasBody || node.block_var ? " " : "",
            node.block_var ? path.call(print, "block_var") : "",
            path.call(print, node.type === "brace_block" ? "stmts" : "bodystmt"),
            hasBody ? " " : "",
            "}"
        ];
        return group(ifBreak(doBlock, braceBlock));
    };
}
exports.printBraceBlock = printBlock(true);
exports.printDoBlock = printBlock(false);