fix: A/B test 16 (pageHeaderRenderer on playlist pages)

codegen/src/abtest.rs

@@ -35,6 +35,7 @@ pub enum ABTest {
     MusicPlaylistTwoColumn = 13,
     CommentsFrameworkUpdate = 14,
     ChannelShortsLockup = 15,
+    PlaylistPageHeader = 16,
 }
 
 /// List of active A/B tests that are run when none is manually specified
@@ -112,6 +113,7 @@ pub async fn run_test(
                     ABTest::MusicPlaylistTwoColumn => music_playlist_two_column(&query).await,
                     ABTest::CommentsFrameworkUpdate => comments_framework_update(&query).await,
                     ABTest::ChannelShortsLockup => channel_shorts_lockup(&query).await,
+                    ABTest::PlaylistPageHeader => playlist_page_header_renderer(&query).await,
                 }
                 .unwrap();
                 pb.inc(1);
@@ -382,3 +384,20 @@ pub async fn channel_shorts_lockup(rp: &RustyPipeQuery) -> Result<bool> {
         .unwrap();
     Ok(res.contains("\"shortsLockupViewModel\""))
 }
+
+pub async fn playlist_page_header_renderer(rp: &RustyPipeQuery) -> Result<bool> {
+    let id = "VLPLZN_exA7d4RVmCQrG5VlWIjMOkMFZVVOc";
+    let res = rp
+        .raw(
+            ClientType::Desktop,
+            "browse",
+            &QBrowse {
+                context: rp.get_context(ClientType::Desktop, true, None).await,
+                browse_id: id,
+                params: None,
+            },
+        )
+        .await
+        .unwrap();
+    Ok(res.contains("\"pageHeaderRenderer\""))
+}

codegen/src/collect_chan_prefixes.rs

@@ -0,0 +1,75 @@
+use std::{collections::BTreeMap, fs::File, io::BufReader};
+
+use path_macro::path;
+use rustypipe::{
+    client::RustyPipe,
+    param::{Language, LANGUAGES},
+};
+use serde::{Deserialize, Serialize};
+
+use crate::util::{self, DICT_DIR};
+
+#[derive(Debug, Serialize, Deserialize)]
+struct Entry {
+    prefix: String,
+    suffix: String,
+}
+
+pub async fn collect_chan_prefixes() {
+    let cname = "kiernanchrisman";
+    let json_path = path!(*DICT_DIR / "chan_prefixes.json");
+    let mut res = BTreeMap::new();
+
+    let rp = RustyPipe::new();
+
+    for lang in LANGUAGES {
+        let playlist = rp
+            .query()
+            .lang(lang)
+            .playlist("PLZN_exA7d4RVmCQrG5VlWIjMOkMFZVVOc")
+            .await
+            .unwrap();
+        let n = playlist.channel.unwrap().name;
+        let offset = n.find(cname).unwrap();
+        let prefix = &n[..offset];
+        let suffix = &n[(offset + cname.len())..];
+
+        res.insert(
+            lang,
+            Entry {
+                prefix: prefix.to_owned(),
+                suffix: suffix.to_owned(),
+            },
+        );
+    }
+
+    let file = File::create(json_path).unwrap();
+    serde_json::to_writer_pretty(file, &res).unwrap();
+}
+
+pub fn write_samples_to_dict() {
+    let json_path = path!(*DICT_DIR / "chan_prefixes.json");
+
+    let json_file = File::open(json_path).unwrap();
+    let collected: BTreeMap<Language, Entry> =
+        serde_json::from_reader(BufReader::new(json_file)).unwrap();
+    let mut dict = util::read_dict();
+    let langs = dict.keys().copied().collect::<Vec<_>>();
+
+    for lang in langs {
+        let dict_entry = dict.entry(lang).or_default();
+
+        let e = collected.get(&lang).unwrap();
+        dict_entry.chan_prefix = e.prefix.trim().to_owned();
+        dict_entry.chan_suffix = e.suffix.trim().to_owned();
+
+        for lang in &dict_entry.equivalent {
+            let ee = collected.get(lang).unwrap();
+            if ee.prefix != e.prefix || ee.suffix != e.suffix {
+                panic!("equivalent lang conflict, lang: {lang}");
+            }
+        }
+    }
+
+    util::write_dict(dict);
+}

codegen/src/gen_dictionary.rs

@@ -85,6 +85,10 @@ pub(crate) struct Entry {
     ///
     /// Format: Parsed text -> Album type
     pub album_types: phf::Map<&'static str, AlbumType>,
+    /// Channel name prefix on playlist pages (e.g. `by`)
+    pub chan_prefix: &'static str,
+    /// Channel name suffix on playlist pages
+    pub chan_suffix: &'static str,
 }
 "#;
 
@@ -180,8 +184,8 @@ pub(crate) fn entry(lang: Language) -> Entry {
             .to_string()
             .replace('\n', "\n            ");
 
-        write!(code_timeago_tokens, "{} => Entry {{\n            timeago_tokens: {},\n            date_order: {},\n            months: {},\n            timeago_nd_tokens: {},\n            comma_decimal: {:?},\n            number_tokens: {},\n            number_nd_tokens: {},\n            album_types: {},\n        }},\n        ",
-        selector, code_ta_tokens, date_order, code_months, code_ta_nd_tokens, entry.comma_decimal, code_number_tokens, code_number_nd_tokens, code_album_types).unwrap();
+        write!(code_timeago_tokens, "{} => Entry {{\n            timeago_tokens: {},\n            date_order: {},\n            months: {},\n            timeago_nd_tokens: {},\n            comma_decimal: {:?},\n            number_tokens: {},\n            number_nd_tokens: {},\n            album_types: {},\n            chan_prefix: {:?},\n            chan_suffix: {:?},\n        }},\n        ",
+        selector, code_ta_tokens, date_order, code_months, code_ta_nd_tokens, entry.comma_decimal, code_number_tokens, code_number_nd_tokens, code_album_types, entry.chan_prefix, entry.chan_suffix).unwrap();
     }
 
     code_timeago_tokens = code_timeago_tokens.trim_end().to_owned() + "\n    }\n}\n";

codegen/src/main.rs

@@ -2,6 +2,7 @@
 
 mod abtest;
 mod collect_album_types;
+mod collect_chan_prefixes;
 mod collect_large_numbers;
 mod collect_playlist_dates;
 mod collect_video_dates;
@@ -29,10 +30,12 @@ enum Commands {
     CollectAlbumTypes,
     CollectVideoDurations,
     CollectVideoDates,
+    CollectChanPrefixes,
     ParsePlaylistDates,
     ParseLargeNumbers,
     ParseAlbumTypes,
     ParseVideoDurations,
+    ParseChanPrefixes,
     GenLocales,
     GenDict,
     DownloadTestfiles,
@@ -65,10 +68,14 @@ async fn main() {
         Commands::CollectVideoDates => {
             collect_video_dates::collect_video_dates(cli.concurrency).await;
         }
+        Commands::CollectChanPrefixes => {
+            collect_chan_prefixes::collect_chan_prefixes().await;
+        }
         Commands::ParsePlaylistDates => collect_playlist_dates::write_samples_to_dict(),
         Commands::ParseLargeNumbers => collect_large_numbers::write_samples_to_dict(),
         Commands::ParseAlbumTypes => collect_album_types::write_samples_to_dict(),
         Commands::ParseVideoDurations => collect_video_durations::parse_video_durations(),
+        Commands::ParseChanPrefixes => collect_chan_prefixes::write_samples_to_dict(),
         Commands::GenLocales => {
             gen_locales::generate_locales().await;
         }

codegen/src/model.rs

@@ -50,10 +50,10 @@ pub struct DictEntry {
     ///
     /// Format: Parsed text -> Album type
     pub album_types: BTreeMap<String, AlbumType>,
-    /// Names of item types (Song, Video, Artist, Playlist)
-    ///
-    /// Format: Parsed text -> Item type
-    pub item_types: BTreeMap<String, ExtItemType>,
+    /// Channel name prefix on playlist pages (e.g. `by`)
+    pub chan_prefix: String,
+    /// Channel name suffix on playlist pages
+    pub chan_suffix: String,
 }
 
 /// Parsed TimeAgo string, contains amount and time unit.