library(tidyverse)

c1_bots <-"/mmfs1/gscratch/comdata/users/mjilg/mw-repo-lifecycles/case1/event_0430_framework_commit_counts.csv"
c1_input_df <- read.csv(c1_bots , header = TRUE) 

c2_bots <-"/mmfs1/gscratch/comdata/users/mjilg/mw-repo-lifecycles/case2/event_0430_framework_commit_counts.csv"
c2_input_df <- read.csv(c2_bots , header = TRUE) 

c3_bots <-"/mmfs1/gscratch/comdata/users/mjilg/mw-repo-lifecycles/case3/event_0430_framework_commit_counts.csv"
c3_input_df <- read.csv(c3_bots , header = TRUE) 

# Add a column to each dataframe to label them
c1_input_df <- c1_input_df %>% mutate(source = "c1")
c2_input_df <- c2_input_df %>% mutate(source = "c2")
c3_input_df <- c3_input_df %>% mutate(source = "c3")

# Combine the dataframes into one
combined_df <- bind_rows(c1_input_df, c2_input_df, c3_input_df)

combined_df$nonbot_commit_count <- combined_df$commit_count - combined_df$bot_commit_count
combined_df$unaff_commit_count <- combined_df$nonbot_commit_count - combined_df$wikimedia_commit_count

library(scales)
library(ggplot2)

long_df <- combined_df |>
  tidyr::pivot_longer(cols = c(unaff_commit_count, wikimedia_commit_count),
                      names_to = "commit_type",
                      values_to = "lengthened_commit_count")

summed_df <- long_df |>
  group_by(relative_week, project, commit_type) |>
  summarize(total_commits = sum(lengthened_commit_count), .groups = 'drop')

combined_df$returning_unaff_commit_count = combined_df$unaff_commit_count - combined_df$unaff_new_commit_count

new_authors_long_df <- combined_df |>
  tidyr::pivot_longer(cols = c(unaff_new_commit_count, returning_unaff_commit_count),
                      names_to = "commit_seniority",
                      values_to = "lengthened_commit_count")


new_unaff_authors <- new_authors_long_df |>
  ggplot(aes(x=relative_week,
             y=lengthened_commit_count, 
             fill=commit_seniority)) +
  geom_col(position='dodge') +
  labs(x = "Relative Week", y = "Commits", fill="Contributor Tenure (New contributors <= 5 commits before deployment announcement)") +
  geom_vline(data = long_df |> filter(source == "c1"), 
             aes(xintercept = -29), 
             linetype = "dotted", color = "black", linewidth = 1) +
  geom_vline(data = long_df |> filter(source == "c1"), 
             aes(xintercept = -9), 
             linetype = "dotted", color = "black", linewidth = 1) +
  geom_vline(data = long_df |> filter(source == "c2"), 
             aes(xintercept = -99), 
             linetype = "dotted", color = "black", linewidth = 1) +
  geom_vline(data = long_df |> filter(source == "c3"), 
             aes(xintercept = -97), 
             linetype = "dotted", color = "black", linewidth = 1) +
  geom_text(data = data.frame(source = "c1", relative_week = -40, lengthened_commit_count = 90), 
            aes(x = relative_week, y = lengthened_commit_count, label = "Opt-In Testing Deployment"), 
            inherit.aes = FALSE, color = "black", size = 4) +
  geom_vline(xintercept = 0, linetype = "dashed", color = "black", linewidth = 1) + # Add vertical line at week 0
  geom_text(data = data.frame(source = "c1", relative_week = 7, lengthened_commit_count = 90), 
            aes(x = relative_week, y = lengthened_commit_count, label = "Wide Deployment"), 
            inherit.aes = FALSE, color = "black", size = 4) +
  scale_fill_manual(values = c("returning_unaff_commit_count" = "#FFC107",  # Color for "Returning Contributors"
                               "unaff_new_commit_count" = "#004D40"),   
                    labels = c("returning_unaff_commit_count" = "Returning",
                               "unaff_new_commit_count" = "New")
  ) +
  ggtitle("Unaffiliated Commits to Bot Framework Libraries During Feature Deployments") +
  theme_bw() +
  theme(
    legend.position = "top",
    plot.title = element_text(size = 18, face = "bold"), # Increase title font size
    axis.title.x = element_text(size = 16),             # Increase x-axis title font size
    axis.title.y = element_text(size = 16),             # Increase y-axis title font size
    axis.text.x = element_text(size = 16),              # Increase x-axis text font size
    axis.text.y = element_text(size = 16),              # Increase y-axis text font size
    legend.text = element_text(size = 16),              # Increase legend text font size
    legend.title = element_text(size = 16),
    strip.text = element_text(size = 14)# Increase legend title font size
  ) +
  facet_wrap(~source, nrow = 3, labeller = labeller(source = c(
    "c1" = "VisualEditor",
    "c2" = "HTTPS-as-default",
    "c3" = "HTTP-deprecation"
  )))

new_unaff_authors

ggsave(filename = "ww-0501-bot-commits-faceted.png", plot = new_unaff_authors, width = 15, height = 9, dpi = 800)