/*      $NetBSD: debug.c,v 1.74 2025/01/04 21:20:59 rillig Exp $        */

/*-
* Copyright (c) 2023 The NetBSD Foundation, Inc.
* All rights reserved.
*
* This code is derived from software contributed to The NetBSD Foundation
* by Roland Illig <[email protected]>.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
*    notice, this list of conditions and the following disclaimer in the
*    documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
* ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/

#include <sys/cdefs.h>
__RCSID("$NetBSD: debug.c,v 1.74 2025/01/04 21:20:59 rillig Exp $");

#include <stdarg.h>
#include <string.h>

#include "indent.h"

#ifdef debug

static struct {
       // false        show only the changes to the parser state
       // true         show unchanged parts of the parser state as well
       bool full_parser_state;
} config = {
       .full_parser_state = false,
};

const char *const lsym_name[] = {
       "lsym_eof",
       "lsym_preprocessing",
       "lsym_newline",
       "lsym_comment",
       "lsym_lparen",
       "lsym_rparen",
       "lsym_lbracket",
       "lsym_rbracket",
       "lsym_lbrace",
       "lsym_rbrace",
       "lsym_period",
       "lsym_unary_op",
       "lsym_sizeof",
       "lsym_offsetof",
       "lsym_postfix_op",
       "lsym_binary_op",
       "lsym_question",
       "lsym_question_colon",
       "lsym_comma",
       "lsym_typedef",
       "lsym_modifier",
       "lsym_tag",
       "lsym_type",
       "lsym_word",
       "lsym_funcname",
       "lsym_label_colon",
       "lsym_other_colon",
       "lsym_semicolon",
       "lsym_case",
       "lsym_default",
       "lsym_do",
       "lsym_else",
       "lsym_for",
       "lsym_if",
       "lsym_switch",
       "lsym_while",
       "lsym_return",
};

const char *const psym_name[] = {
       "psym_0",
       "psym_lbrace_block",
       "psym_lbrace_struct",
       "psym_lbrace_union",
       "psym_lbrace_enum",
       "psym_rbrace",
       "psym_decl",
       "psym_stmt",
       "psym_for_exprs",
       "psym_if_expr",
       "psym_if_expr_stmt",
       "psym_if_expr_stmt_else",
       "psym_else",
       "psym_switch_expr",
       "psym_do",
       "psym_do_stmt",
       "psym_while_expr",
};

static const char *const newline_name[] = {
       "nl_no",
       "nl_unless_if",
       "nl_unless_lbrace",
       "nl_unless_semicolon",
       "nl_yes",
};

static const char *const declaration_name[] = {
       "decl_no",
       "decl_begin",
       "decl_end",
};

static const char *const badp_name[] = {
       "badp_none",
       "badp_seen_lbrace",
       "badp_decl",
       "badp_seen_decl",
       "badp_yes",
};

const char *const paren_level_cast_name[] = {
       "cast_unknown",
       "cast_maybe",
       "cast_no",
};

const char *const line_kind_name[] = {
       "lk_other",
       "lk_blank",
       "lk_pre_if",
       "lk_pre_endif",
       "lk_pre_other",
       "lk_stmt_head",
       "lk_func_end",
       "lk_block_comment",
       "lk_case_or_default",
};

static const char *const extra_expr_indent_name[] = {
       "eei_no",
       "eei_maybe",
       "eei_last"
};

static struct {
       struct parser_state prev_ps;
       bool ps_first;
       const char *heading1;
       const char *heading2;
       unsigned wrote_newlines;
} state = {
       .ps_first = true,
       .wrote_newlines = 1,
};

static FILE *
debug_file(void)
{
       return output == stdout ? stderr : stdout;
}

static void
debug_print_headings(void)
{
       if (state.heading1 != NULL) {
               (void)fprintf(debug_file(), "%s:\n", state.heading1);
               state.heading1 = NULL;
               state.wrote_newlines = 1;
       }
       if (state.heading2 != NULL) {
               (void)fprintf(debug_file(), "\t%s:\n", state.heading2);
               state.heading2 = NULL;
               state.wrote_newlines = 1;
       }
}

void
debug_printf(const char *fmt, ...)
{
       va_list ap;

       debug_print_headings();
       va_start(ap, fmt);
       (void)vfprintf(debug_file(), fmt, ap);
       va_end(ap);
       state.wrote_newlines = 0;
}

void
debug_println(const char *fmt, ...)
{
       va_list ap;

       debug_print_headings();
       va_start(ap, fmt);
       (void)vfprintf(debug_file(), fmt, ap);
       va_end(ap);
       (void)fprintf(debug_file(), "\n");
       state.wrote_newlines = fmt[0] == '\0' ? state.wrote_newlines + 1 : 1;
}

void
debug_blank_line(void)
{
       while (state.wrote_newlines < 2)
               debug_println("");
}

void
debug_vis_range(const char *s, size_t len)
{
       debug_printf("\"");
       for (size_t i = 0; i < len; i++) {
               const char *p = s + i;
               if (*p == '\\' || *p == '"')
                       debug_printf("\\%c", *p);
               else if (isprint((unsigned char)*p))
                       debug_printf("%c", *p);
               else if (*p == '\n')
                       debug_printf("\\n");
               else if (*p == '\t')
                       debug_printf("\\t");
               else
                       debug_printf("\\x%02x", (unsigned char)*p);
       }
       debug_printf("\"");
}

void
debug_print_buf(const char *name, const struct buffer *buf)
{
       if (buf->len > 0) {
               debug_printf(" %s ", name);
               debug_vis_range(buf->s, buf->len);
       }
}

void
debug_buffers(const char *descr)
{
       if (lab.len > 0 || code.len > 0 || com.len > 0) {
               debug_printf("%s:", descr);
               debug_print_buf("label", &lab);
               debug_print_buf("code", &code);
               debug_print_buf("comment", &com);
               debug_blank_line();
       }
}

static void
debug_ps_bool_member(const char *name, bool prev, bool curr)
{
       if (!state.ps_first && curr != prev) {
               char diff = " -+x"[(prev ? 1 : 0) + (curr ? 2 : 0)];
               debug_println("\t\t%s: [%c]", name, diff);
       } else if (config.full_parser_state || state.ps_first)
               debug_println("\t\t%s: [%c]", name, curr ? 'x' : ' ');
}

static void
debug_ps_int_member(const char *name, int prev, int curr)
{
       if (!state.ps_first && curr != prev)
               debug_println("\t\t%s: %d -> %d", name, prev, curr);
       else if (config.full_parser_state || state.ps_first)
               debug_println("\t\t%s: %d", name, curr);
}

static void
debug_ps_enum_member(const char *name, const char *prev, const char *curr)
{
       if (!state.ps_first && strcmp(prev, curr) != 0)
               debug_println("\t\t%s: %s -> %s", name, prev, curr);
       else if (config.full_parser_state || state.ps_first)
               debug_println("\t\t%s: %s", name, curr);
}

static void
debug_ps_str_member(const char *name,
   void (*to_string)(struct buffer *, const struct parser_state *))
{
       static struct buffer prev_buf;
       static struct buffer curr_buf;

       buf_clear(&prev_buf);
       to_string(&prev_buf, &state.prev_ps);
       buf_clear(&curr_buf);
       to_string(&curr_buf, &ps);

       if (!state.ps_first && strcmp(prev_buf.s, curr_buf.s) != 0)
               debug_println("\t\t%s: %s -> %s",
                   name, prev_buf.s, curr_buf.s);
       else if (config.full_parser_state || state.ps_first)
               debug_println("\t\t%s: %s", name, curr_buf.s);
}

static void
ps_di_stack_to_string(struct buffer *buf, const struct parser_state *s)
{
       for (int i = 0; i < s->decl_level; i++) {
               char str[64];
               snprintf(str, sizeof(str), "%s%d",
                        i > 0 ? " " : "", s->di_stack[i]);
               buf_add_str(buf, str);
       }
       if (s->decl_level == 0)
               buf_add_str(buf, "none");
}

static void
ps_paren_to_string(struct buffer *buf, const struct parser_state *s)
{
       for (size_t i = 0; i < s->paren.len; i++) {
               buf_add_str(buf, i > 0 ? "  " : "");
               buf_add_str(buf, paren_level_cast_name[s->paren.item[i].cast]);
               char str[64];
               snprintf(str, sizeof(str), " %d", s->paren.item[i].indent);
               buf_add_str(buf, str);
       }
       if (s->paren.len == 0)
               buf_add_str(buf, "none");
}

void
ps_psyms_to_string(struct buffer *buf, const struct parser_state *s)
{
       for (size_t i = 0; i < s->psyms.len; i++) {
               char num[64];
               snprintf(num, sizeof(num), "%s%d ",
                   i > 0 ? "  " : "", s->psyms.ind_level[i]);
               buf_add_str(buf, num);
               buf_add_str(buf, psym_name[s->psyms.sym[i]]);
       }
}

#define debug_ps_bool(name) \
       debug_ps_bool_member(#name, state.prev_ps.name, ps.name)
#define debug_ps_int(name) \
       debug_ps_int_member(#name, state.prev_ps.name, ps.name)
#define debug_ps_enum(name, names) \
       debug_ps_enum_member(#name, (names)[state.prev_ps.name], \
           (names)[ps.name])
#define debug_ps_str(name, to_string) \
       debug_ps_str_member((/* LINTED 129 */(void)&ps.name, #name), to_string)

void
debug_parser_state(void)
{
       debug_blank_line();

       state.heading1 = "parser state";
       state.heading2 = "token classification";
       debug_ps_enum(prev_lsym, lsym_name);
       debug_ps_bool(in_stmt_or_decl);
       debug_ps_bool(in_decl);
       debug_ps_bool(in_typedef_decl);
       debug_ps_bool(in_var_decl);
       debug_ps_bool(in_init);
       debug_ps_int(init_level);
       debug_ps_bool(line_has_func_def);
       debug_ps_bool(in_func_def_params);
       debug_ps_bool(line_has_decl);
       debug_ps_enum(lbrace_kind, psym_name);
       debug_ps_enum(spaced_expr_psym, psym_name);
       debug_ps_bool(seen_case);
       debug_ps_bool(prev_paren_was_cast);
       debug_ps_int(quest_level);

       state.heading2 = "indentation of statements and declarations";
       debug_ps_int(ind_level);
       debug_ps_int(ind_level_follow);
       debug_ps_str(psyms, ps_psyms_to_string);
       debug_ps_bool(line_is_stmt_cont);
       debug_ps_int(decl_level);
       debug_ps_str(di_stack, ps_di_stack_to_string);
       debug_ps_bool(decl_indent_done);
       debug_ps_int(decl_ind);
       debug_ps_bool(tabs_to_var);
       debug_ps_enum(extra_expr_indent, extra_expr_indent_name);

       state.heading2 = "spacing inside a statement or declaration";
       debug_ps_bool(next_unary);
       debug_ps_bool(want_blank);
       debug_ps_int(ind_paren_level);
       debug_ps_str(paren, ps_paren_to_string);

       state.heading2 = "indentation of comments";
       debug_ps_int(comment_ind);
       debug_ps_int(comment_shift);
       debug_ps_bool(comment_cont);

       state.heading2 = "vertical spacing";
       debug_ps_bool(break_after_comma);
       debug_ps_enum(newline, newline_name);
       debug_ps_enum(declaration, declaration_name);
       debug_ps_bool(blank_line_after_decl);
       debug_ps_enum(badp, badp_name);

       state.heading1 = NULL;
       state.heading2 = NULL;
       debug_blank_line();

       parser_state_free(&state.prev_ps);
       parser_state_back_up(&state.prev_ps);
       state.ps_first = false;
}
#endif