refact: fix bugs and make export work

Author: frytg

.github/dependabot.yml

@@ -0,0 +1,9 @@
+# https://docs.github.com/en/code-security/dependabot/dependabot-version-updates/configuration-options-for-the-dependabot.yml-file
+
+version: 2
+updates:
+  - package-ecosystem: cargo
+    directory: "/"
+    schedule:
+      interval: weekly
+      day: friday

.github/exporter.sh

@@ -1,7 +1,4 @@
-pkg/*.gem
-coverage/
-.coveralls.yml
-.bundle
-Gemfile.lock
 target/
 Cargo.lock
+.DS_Store
+*.svg

.github/workflows/build.yml

@@ -1,11 +1,17 @@
 [package]
 name = "githubchart"
-version = "0.1.0"
-authors = ["Les Aker <[email protected]>"]
-edition = "2018"
+version = "5.0.0"
+authors = ["frytg"]
+edition = "2021"
 
 [dependencies]
-serde = { version = "1.0", features = ["derive"] }
-serde_json = "1.0"
-svg = "0.8"
-chrono = "0.4"
+reqwest = { version = "0.11", features = ["blocking", "json"] }
+scraper = "0.17"
+
+[profile.release]
+# optimizations from https://github.com/johnthagen/min-sized-rust
+opt-level = 'z' # optimize for size
+lto = true # enable link time optimization
+codegen-units = 1 # reduce parallel codegen units to increase optimizations
+panic = 'abort' # abort on panic
+strip = true # strip symbols from binary

.gitignore

@@ -1,35 +1,51 @@
-GithubChart
-============
+# GitHubChart (The Rusty Version)
 
-[![Crate Version](https://img.shields.io/crates/v/githubchart.svg)](https://crates.io/crates/githubchart)
-[![GitHub Workflow Status](https://img.shields.io/github/actions/workflow/status/akerl/githubchart/build.yml?branch=main)](https://github.com/akerl/githubchart/actions)
-[![MIT Licensed](https://img.shields.io/badge/license-MIT-green.svg)](https://tldrlegal.com/license/mit-license)
+Generates an SVG of your GitHub contributions:
 
-Generates an SVG of your Github contributions:
+![Example image](./assets/frytg.svg)
 
-![Example image](http://akerl.github.io/githubchart/chart.svg)
+## Fork
 
-![Other user example](http://akerl.github.io/githubchart/other_user.svg)
+This is forked from [githubchart](https://github.com/akerl/githubchart) and ported from Ruby to Rust. It does not provide 100% of the same functionality, but it does generate a similar SVG.
 
-## Usage
+## Usage with `cargo`
 
-Run `githubchart path/to/svg` to generate an SVG. To override the default username (pulled from your local shell or .gitconfig), use `githubchart -u username path/to/svg`
+If you have Rust installed and are familiar with cargo, you can install and run this directly:
 
-GithubChart also allows you to provide input from a file instead of pulling data from Github. You can pass JSON to GithubChart by using `githubchart -i /path/to/file /path/to/svg`, or use '-' to use STDIN. See spec/examples/input.json for example data.
+```sh
+cargo run -- output.svg -u frytg
+```
 
-If you don't provide a file path, the resulting SVG will be printed to stdout.
+This compiles and runs the program directly (using dev profile and debug symbols). This would also be the command when developing locally.
 
-To modify the color scheme used, you can provide `-c SCHEME`. For example, `githubchart -c halloween` uses GitHub's halloween colors. Use `-s` to list the available schemes.
+To modify the color scheme used, you can provide `-c SCHEME`. For example, `cargo run -- output.svg -u frytg -c halloween` uses GitHub's halloween colors.
 
-### Hosted SVG
+Use `cargo fmt` to format the code.
 
-A hosted service for loading these SVGs was made by [2016rshah](https://github.com/2016rshah): http://ghchart.rshah.org/ ([source code](https://github.com/2016rshah/githubchart-api))
+## Usage with binary
 
-## Installation
+Alternatively, you can download a release binary from the [releases page](https://github.com/frytg/githubchart-rust/releases) and run it directly:
 
-    cargo install githubchart
+```sh
+./githubchart output.svg -u frytg
+```
 
-## License
+## Build
+
+You can build a release binary with:
+
+```sh
+cargo build --release
+```
+
+[`Cargo.toml`](./Cargo.toml) is configured to optimize for size.
 
-githubchart is released under the MIT License. See the bundled LICENSE file for details.
+Test the binary with:
+
+```sh
+./target/release/githubchart release.svg -u frytg
+```
+
+## License
 
+This `githubchart-rust` fork (like the upstream repo) is released under the MIT License. See the bundled [LICENSE](./LICENSE) file for details.

Cargo.toml

@@ -1,3 +1,6 @@
+use reqwest;
+use scraper::{Html, Selector};
+
 pub mod svg;
 
 pub struct Chart {
@@ -14,17 +17,75 @@ impl Chart {
         }
     }
 
-    pub fn render(&self, format: &str) -> Result<String, String> {
-        match format {
-            "svg" => Ok(svg::render_svg(&self)),
-            "svg_square" => Ok(svg::render_svg_square(&self)),
-            _ => Err(format!("Format {} is unsupported.", format)),
-        }
+    pub fn render(&self) -> Result<String, String> {
+        Ok(svg::render_svg(&self))
     }
 }
 
 pub const COLOR_SCHEMES: &[(&str, &[&str])] = &[
-    ("default", &["#eeeeee", "#c6e48b", "#7bc96f", "#239a3b", "#196127"]),
-    ("old", &["#eeeeee", "#d6e685", "#8cc665", "#44a340", "#1e6823"]),
-    ("halloween", &["#EEEEEE", "#FFEE4A", "#FFC501", "#FE9600", "#03001C"]),
+    (
+        "default",
+        &["#eeeeee", "#c6e48b", "#7bc96f", "#239a3b", "#196127"],
+    ),
+    (
+        "old",
+        &["#eeeeee", "#d6e685", "#8cc665", "#44a340", "#1e6823"],
+    ),
+    (
+        "halloween",
+        &["#EEEEEE", "#FFEE4A", "#FFC501", "#FE9600", "#03001C"],
+    ),
 ];
+
+pub fn fetch_github_stats(
+    username: &str,
+) -> Result<Vec<(String, i32)>, Box<dyn std::error::Error>> {
+    // GitHub's contribution graph data endpoint
+    let url = format!("https://github.com/users/{}/contributions", username);
+
+    // Fetch the HTML content
+    let client = reqwest::blocking::Client::new();
+    let response = client
+        .get(&url)
+        .header("User-Agent", "githubchart-rust")
+        .send()?;
+
+    // Check status code
+    if !response.status().is_success() {
+        return Err(format!(
+            "Failed loading data from GitHub: {} {}",
+            url,
+            response.status()
+        )
+        .into());
+    }
+
+    let html_content = response.text()?;
+    let document = Html::parse_document(&html_content);
+
+    // Select contribution calendar cells
+    let cell_selector = Selector::parse("td.ContributionCalendar-day").unwrap();
+
+    let mut stats = Vec::new();
+
+    for element in document.select(&cell_selector) {
+        if let (Some(date), Some(count_str)) = (
+            element.value().attr("data-date"),
+            element.value().attr("data-level"),
+        ) {
+            if let Ok(count) = count_str.parse::<i32>() {
+                // println!("fetch_github_stats > date: {}, count: {}", date, count);
+                stats.push((date.to_string(), count));
+            }
+        }
+    }
+
+    // Sort by date
+    stats.sort_by(|a, b| a.0.cmp(&b.0));
+
+    if stats.is_empty() {
+        println!("Warning: No contribution data found for user {}", username);
+    }
+
+    Ok(stats)
+}