uxn-tal 0.2.2

A Rust library for assembling TAL (Tal Assembly Language) files into UXN ROM files
Documentation
use super::{
    downloader::{http_get, write_bytes},
    includes::{parse_includes, resolve_include},
    provider::{FetchResult, Provider, RepoRef},
};
use std::{
    collections::{HashSet, VecDeque},
    fs, path::{Path, PathBuf},
};
use url::Url;

pub struct GitHub;

impl GitHub {
    fn raw_url(r: &RepoRef, repo_rel: &str) -> String {
        format!("https://raw.githubusercontent.com/{}/{}/{}/{}",
            r.owner, r.repo, r.branch, repo_rel)
    }
    // fn try_entry(r: &RepoRef, out: &Path, repo_rel: &str) -> Option<PathBuf> {
    //     let url = Self::raw_url(r, repo_rel);
    //     if let Ok(bytes) = http_get(&url) {
    //         let local = out.join(repo_rel);
    //         if write_bytes(&local, &bytes).is_ok() { return Some(local); }
    //     }
    //     None
    // }
    // fn branch_candidates(branch: &str) -> Vec<String> {
    //     if branch != "HEAD" { return vec![branch.to_string()]; }
    //     vec!["main".into(), "master".into(), "trunk".into()]
    // }
    fn fetch_file(r: &RepoRef, out_root: &Path, repo_rel: &str) -> Result<PathBuf, Box<dyn std::error::Error>> {
        let url = Self::raw_url(r, repo_rel);
        let bytes = http_get(&url)?;
        let local = out_root.join(repo_rel);
        if let Some(p) = local.parent() { fs::create_dir_all(p)?; }
        write_bytes(&local, &bytes)?;
        Ok(local)
    }
}

impl Provider for GitHub {
    fn parse_url(&self, url: &str) -> Option<RepoRef> {
        let u = Url::parse(url).ok()?;
        if u.domain()? != "github.com" { return None; }
        let segs = u.path_segments()?.collect::<Vec<_>>();
        if segs.len() < 2 { return None; }

        let owner = segs[0].to_string();
        let repo  = segs[1].trim_end_matches(".git").to_string();

        if segs.len() == 2 {
            return Some(RepoRef { host: "github.com".into(), owner, repo, branch: "HEAD".into(), path: None });
        }

        // Accept GitHub views: blob/tree/raw/blame
        match segs[2] {
            "blob" | "tree" | "raw" | "blame" if segs.len() >= 4 => {
                // <view>/<ref>/<path...>   where <ref> can be a branch name or a commit SHA
                let branch = segs[3].to_string();
                let path = if segs.len() > 4 { Some(segs[4..].join("/")) } else { None };
                Some(RepoRef { host: "github.com".into(), owner, repo, branch, path })
            }
            _ => {
                // Fallback: treat remainder as repo path with unknown ref
                let path = if segs.len() > 2 { Some(segs[2..].join("/")) } else { None };
                Some(RepoRef { host: "github.com".into(), owner, repo, branch: "HEAD".into(), path })
            }
        }
    }
    fn fetch_tal_tree(&self, r: &RepoRef, out_root: &Path) -> Result<FetchResult, Box<dyn std::error::Error>> {
        let entry_rel = match &r.path {
            Some(p) if p.to_ascii_lowercase().ends_with(".tal") || p.to_ascii_lowercase().ends_with(".rom") => p.replace('\\', "/"),
            _ => return Err("github: URL must point to a .tal or .rom file; not guessing entries".into()),
        };

        let entry_local = Self::fetch_file(r, out_root, &entry_rel)?;
        let mut all = vec![entry_local.clone()];
        // Only walk includes for .tal files
        if entry_rel.to_ascii_lowercase().ends_with(".tal") {
            let mut visited: HashSet<String> = [entry_rel.clone()].into_iter().collect();
            let mut q: VecDeque<(String, PathBuf)> = VecDeque::from([(entry_rel.clone(), entry_local.clone())]);
            while let Some((curr_rel, curr_local)) = q.pop_front() {
                let src = fs::read_to_string(&curr_local).unwrap_or_default();
                for inc in parse_includes(&src) {
                    let target = resolve_include(&curr_rel, &inc);
                    if !visited.insert(target.clone()) { continue; }
                    match Self::fetch_file(r, out_root, &target) {
                        Ok(loc) => { all.push(loc.clone()); q.push_back((target, loc)); }
                        Err(e)  => eprintln!("[github] warn: missing include {} ({})", target, e),
                    }
                }
            }
        }
        Ok(FetchResult { entry_local: all[0].clone(), all_files: all })

        // for b in Self::branch_candidates(&r_in.branch) {
        //     let r = RepoRef { branch: b.clone(), ..r_in.clone() };

        //     let candidates: Vec<String> = match &r.path {
        //         Some(p) if p.ends_with(".tal") => vec![p.clone()],
        //         Some(p) if !p.is_empty()       => repo_entry_guesses().iter().map(|g| format!("{}/{}", p.trim_end_matches('/'), g)).collect(),
        //         _ => repo_entry_guesses().iter().map(|s| s.to_string()).collect(),
        //     };

        //     if let Some((entry_rel, entry_local)) =
        //         candidates.into_iter().find_map(|c| Self::try_entry(&r, out_root, &c).map(|l| (c, l)))
        //     {
        //         let mut visited = HashSet::new();
        //         let mut all = vec![entry_local.clone()];
        //         visited.insert(entry_rel.clone());
        //         let mut q = VecDeque::new();
        //         q.push_back((entry_rel, entry_local));

        //         let mut steps = 0usize;
        //         while let Some((repo_rel, local)) = q.pop_front() {
        //             if steps >= 200 { break; }
        //             steps += 1;

        //             let src = fs::read_to_string(&local).unwrap_or_default();
        //             for inc in parse_includes(&src) {
        //                 let target = resolve_include(&repo_rel, &inc);
        //                 if !visited.insert(target.clone()) { continue; }
        //                 let url = Self::raw_url(&r, &target);
        //                 println!("Fetching include from URL: {}", url);
        //                 match http_get(&url) {
        //                     Ok(bytes) => {
        //                         let loc = out_root.join(&target);
        //                         if write_bytes(&loc, &bytes).is_ok() {
        //                             all.push(loc.clone());
        //                             q.push_back((target, loc));
        //                         }
        //                     }
        //                     Err(_) => eprintln!("warning: include not found in GitHub repo: {} (branch {})", target, r.branch),
        //                 }
        //             }
        //         }
        //         return Ok(FetchResult { entry_local: all[0].clone(), all_files: all });
        //     }
        // }
        // Err("Could not find an entry .tal in GitHub repo across tried branches".into())
    }
}