library(here)
library(tidyverse)
library(gt)
source(here("04-topics/rep-chv2011/Rcode/chv2011-data-prep.R"))
source(here("04-topics/rep-chv2011/Rcode/chv2011-gt-quarto.R"))

raw <- load_chv2011_raw() |> add_chv2011_derived()

# Bundled data already contains corrected AFQT (Hansen-Heckman-Mullen).
# Report schooling-at-test adjustment using school categories as proxies.
prep <- raw |>
  mutate(
    hs_grad = as.integer(school >= 2),
    some_college = as.integer(school >= 3)
  )

table_data <- lm(cafqt ~ hs_grad + some_college + state, data = prep) |>
  broom::tidy() |>
  mutate(
    cell = chv2011_coef_cell(estimate, std.error, chv2011_stars(p.value))
  ) |>
  select(term, cell)

gt_tbl <- table_data |>
  chv2011_quarto_blank_df() |>
  gt() |>
  tab_header(
    title = "Table A-2",
    subtitle = "Regression of AFQT on schooling (proxy; bundled corrected AFQT)"
  ) |>
  cols_label(term = "Variable", cell = "Coefficient (SE)")

save(table_data, gt_tbl, file = here("04-topics/rep-chv2011/Rcode/Table_A2.RData"))