Compare commits
	
		
			15 Commits
		
	
	
		
			7df19a480f
			...
			4b557a923c
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 4b557a923c | |||
| c08a20ac00 | |||
| 94912e9125 | |||
| a9465dda6e | |||
| add6f00ed6 | |||
| 4a433a1a77 | |||
| 03cbcd9ae0 | |||
| 6fc71c7a78 | |||
| 96a3ca092a | |||
| b750d88d48 | |||
| 808790a7c3 | |||
| 2de01b2a0e | |||
| be0fd5505b | |||
| a23429104c | |||
| 66581cc453 | 
							
								
								
									
										3
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										3
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							@@ -4,4 +4,5 @@
 | 
				
			|||||||
perf.data
 | 
					perf.data
 | 
				
			||||||
flamegraph.svg
 | 
					flamegraph.svg
 | 
				
			||||||
perf.data.old
 | 
					perf.data.old
 | 
				
			||||||
/docker/logs/*
 | 
					/docker/logs/*
 | 
				
			||||||
 | 
					/downloaded
 | 
				
			||||||
							
								
								
									
										13
									
								
								Crawler.toml
									
									
									
									
									
								
							
							
						
						
									
										13
									
								
								Crawler.toml
									
									
									
									
									
								
							@@ -3,14 +3,9 @@ surreal_url = "localhost:8000"
 | 
				
			|||||||
surreal_username = "root"
 | 
					surreal_username = "root"
 | 
				
			||||||
surreal_password = "root"
 | 
					surreal_password = "root"
 | 
				
			||||||
surreal_ns = "test"
 | 
					surreal_ns = "test"
 | 
				
			||||||
surreal_db = "v1.17"
 | 
					surreal_db = "v1.19.5"
 | 
				
			||||||
 | 
					 | 
				
			||||||
# Minio config
 | 
					 | 
				
			||||||
s3_bucket = "v1.17"
 | 
					 | 
				
			||||||
s3_url = "http://localhost:9000"
 | 
					 | 
				
			||||||
s3_access_key = "Ok6s9uQEvKrqRoGZdacm"
 | 
					 | 
				
			||||||
s3_secret_key = "qubeSkP787c7QZu4TvtnuwPTGIAq6ETPupCxvv6K"
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
# Crawler config
 | 
					# Crawler config
 | 
				
			||||||
crawl_filter = "en.wikipedia.com" 
 | 
					crawl_filter = "en.wikipedia.org" 
 | 
				
			||||||
budget = 1000
 | 
					start_url = "https://en.wikipedia.org"
 | 
				
			||||||
 | 
					budget = 100
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										36
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										36
									
								
								README.md
									
									
									
									
									
								
							@@ -2,13 +2,43 @@
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
Crawls sites saving all the found links to a surrealdb database. It then proceeds to take batches of 100 uncrawled links untill the crawl budget is reached. It saves the data of each site in a minio database.
 | 
					Crawls sites saving all the found links to a surrealdb database. It then proceeds to take batches of 100 uncrawled links untill the crawl budget is reached. It saves the data of each site in a minio database.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					## How to use
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					1. Clone the repo and `cd` into it.
 | 
				
			||||||
 | 
					2. Build the repo with `cargo build -r`
 | 
				
			||||||
 | 
					3. Start the docker conatiners
 | 
				
			||||||
 | 
						1. cd into the docker folder `cd docker`
 | 
				
			||||||
 | 
						2. Bring up the docker containers `docker compose up -d`
 | 
				
			||||||
 | 
					4. From the project's root, edit the `Crawler.toml` file to your liking.
 | 
				
			||||||
 | 
					5. Run with `./target/release/internet_mapper`
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					You can view stats of the project at `http://<your-ip>:3000/dashboards`
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					```bash
 | 
				
			||||||
 | 
					# Untested script but probably works
 | 
				
			||||||
 | 
					git clone https://git.oliveratkinson.net/Oliver/internet_mapper.git
 | 
				
			||||||
 | 
					cd internet_mapper
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					cargo build -r
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					cd docker
 | 
				
			||||||
 | 
					docker compose up -d
 | 
				
			||||||
 | 
					cd ..
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					$EDITOR Crawler.toml
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					./target/release/internet_mapper
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					```
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### TODO
 | 
					### TODO
 | 
				
			||||||
 | 
					
 | 
				
			||||||
- [ ] Domain filtering - prevent the crawler from going on alternate versions of wikipedia.
 | 
					- [x] Domain filtering - prevent the crawler from going on alternate versions of wikipedia.
 | 
				
			||||||
- [ ] Conditionally save content - based on filename or file contents
 | 
					- [ ] Conditionally save content - based on filename or file contents
 | 
				
			||||||
- [x] GUI / TUI ? - Graphana
 | 
					- [x] GUI / TUI ? - Graphana
 | 
				
			||||||
- [x] Better asynchronous getting of the sites. Currently it all happens serially.
 | 
					- [x] Better asynchronous getting of the sites. Currently it all happens serially.
 | 
				
			||||||
- [ ] Allow for storing asynchronously
 | 
					- [x] Allow for storing asynchronously - dropping the "links to" logic fixes this need
 | 
				
			||||||
 | 
					- [x] Control crawler via config file (no recompliation needed)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
3/17/25: Took >1hr to crawl 100 pages
 | 
					3/17/25: Took >1hr to crawl 100 pages
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -17,6 +47,8 @@ This ment we stored 1000 pages, 142,997 urls, and 1,425,798 links between the tw
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
3/20/25: Took 5min to crawl 1000 pages
 | 
					3/20/25: Took 5min to crawl 1000 pages
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					3/21/25: Took 3min to crawl 1000 pages
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# About
 | 
					# About
 | 
				
			||||||
 | 
					
 | 
				
			||||||

 | 
					
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -14,22 +14,6 @@ services:
 | 
				
			|||||||
      - --pass
 | 
					      - --pass
 | 
				
			||||||
      - root
 | 
					      - root
 | 
				
			||||||
      - rocksdb:/mydata/database.db
 | 
					      - rocksdb:/mydata/database.db
 | 
				
			||||||
  minio:
 | 
					 | 
				
			||||||
    image: quay.io/minio/minio
 | 
					 | 
				
			||||||
    ports:
 | 
					 | 
				
			||||||
      - 9000:9000
 | 
					 | 
				
			||||||
      - 9001:9001
 | 
					 | 
				
			||||||
    environment:
 | 
					 | 
				
			||||||
      - MINIO_ROOT_USER=root
 | 
					 | 
				
			||||||
      - MINIO_ROOT_PASSWORD=an8charpassword
 | 
					 | 
				
			||||||
      - MINIO_PROMETHEUS_AUTH_TYPE=public
 | 
					 | 
				
			||||||
    volumes:
 | 
					 | 
				
			||||||
      - minio_storage:/data
 | 
					 | 
				
			||||||
    command:
 | 
					 | 
				
			||||||
      - server
 | 
					 | 
				
			||||||
      - /data
 | 
					 | 
				
			||||||
      - --console-address
 | 
					 | 
				
			||||||
      - ":9001"
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
  alloy:
 | 
					  alloy:
 | 
				
			||||||
    image: grafana/alloy:latest
 | 
					    image: grafana/alloy:latest
 | 
				
			||||||
@@ -82,4 +66,3 @@ volumes:
 | 
				
			|||||||
  grafana_storage:
 | 
					  grafana_storage:
 | 
				
			||||||
  alloy_storage:
 | 
					  alloy_storage:
 | 
				
			||||||
  surrealdb_storage:
 | 
					  surrealdb_storage:
 | 
				
			||||||
  minio_storage:
 | 
					 | 
				
			||||||
 
 | 
				
			|||||||
@@ -8,13 +8,10 @@ scrape_configs:
 | 
				
			|||||||
    # change this your machine's ip, localhost won't work
 | 
					    # change this your machine's ip, localhost won't work
 | 
				
			||||||
    # because localhost refers to the docker container.
 | 
					    # because localhost refers to the docker container.
 | 
				
			||||||
      - targets: ['172.20.239.48:2500']
 | 
					      - targets: ['172.20.239.48:2500']
 | 
				
			||||||
 | 
					        #- targets: ['192.168.8.209:2500']
 | 
				
			||||||
  - job_name: loki
 | 
					  - job_name: loki
 | 
				
			||||||
    static_configs:
 | 
					    static_configs:
 | 
				
			||||||
      - targets: ['loki:3100']
 | 
					      - targets: ['loki:3100']
 | 
				
			||||||
  - job_name: prometheus
 | 
					  - job_name: prometheus
 | 
				
			||||||
    static_configs:
 | 
					    static_configs:
 | 
				
			||||||
      - targets: ['localhost:9090']
 | 
					      - targets: ['localhost:9090']
 | 
				
			||||||
  - job_name: minio
 | 
					 | 
				
			||||||
    metrics_path: /minio/v2/metrics/cluster
 | 
					 | 
				
			||||||
    static_configs:
 | 
					 | 
				
			||||||
      - targets: ['minio:9000']
 | 
					 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										108
									
								
								src/db.rs
									
									
									
									
									
								
							
							
						
						
									
										108
									
								
								src/db.rs
									
									
									
									
									
								
							@@ -1,34 +1,20 @@
 | 
				
			|||||||
use base64::{
 | 
					 | 
				
			||||||
    alphabet,
 | 
					 | 
				
			||||||
    engine::{self, general_purpose},
 | 
					 | 
				
			||||||
    Engine,
 | 
					 | 
				
			||||||
};
 | 
					 | 
				
			||||||
use metrics::counter;
 | 
					use metrics::counter;
 | 
				
			||||||
use serde::{ser::SerializeStruct, Deserialize, Serialize};
 | 
					use std::fmt::Debug;
 | 
				
			||||||
use std::{fmt::Debug, sync::LazyLock, time::Instant};
 | 
					use serde::{Deserialize, Serialize};
 | 
				
			||||||
use surrealdb::{
 | 
					use surrealdb::{
 | 
				
			||||||
    engine::remote::ws::{Client, Ws},
 | 
					    engine::remote::ws::{Client, Ws},
 | 
				
			||||||
    opt::auth::Root,
 | 
					    opt::auth::Root,
 | 
				
			||||||
    sql::Thing,
 | 
					    sql::Thing,
 | 
				
			||||||
    Response, Surreal,
 | 
					    Surreal,
 | 
				
			||||||
};
 | 
					};
 | 
				
			||||||
use tokio::sync::Mutex;
 | 
					use tracing::{error, instrument, trace};
 | 
				
			||||||
use tracing::{error, instrument, trace, warn};
 | 
					 | 
				
			||||||
use url::Url;
 | 
					use url::Url;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
use crate::Config;
 | 
					use crate::Config;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// static LOCK: LazyLock<Arc<Mutex<bool>>> = LazyLock::new(|| Arc::new(Mutex::new(true)));
 | 
					 | 
				
			||||||
static LOCK: LazyLock<Mutex<bool>> = LazyLock::new(|| Mutex::new(true));
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
const CUSTOM_ENGINE: engine::GeneralPurpose =
 | 
					 | 
				
			||||||
    engine::GeneralPurpose::new(&alphabet::URL_SAFE, general_purpose::NO_PAD);
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
const TIME_SPENT_ON_LOCK: &str = "surql_lock_waiting_ms";
 | 
					 | 
				
			||||||
const STORE: &str = "surql_store_calls";
 | 
					const STORE: &str = "surql_store_calls";
 | 
				
			||||||
const LINK: &str = "surql_link_calls";
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
#[derive(Deserialize, Clone)]
 | 
					#[derive(Serialize, Deserialize, Clone, Eq, PartialEq, Hash)]
 | 
				
			||||||
pub struct Website {
 | 
					pub struct Website {
 | 
				
			||||||
    /// The url that this data is found at
 | 
					    /// The url that this data is found at
 | 
				
			||||||
    pub site: Url,
 | 
					    pub site: Url,
 | 
				
			||||||
@@ -36,18 +22,6 @@ pub struct Website {
 | 
				
			|||||||
    pub crawled: bool,
 | 
					    pub crawled: bool,
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
impl Serialize for Website {
 | 
					 | 
				
			||||||
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
 | 
					 | 
				
			||||||
    where
 | 
					 | 
				
			||||||
        S: serde::Serializer {
 | 
					 | 
				
			||||||
            let mut state = serializer.serialize_struct("Website", 2)?;
 | 
					 | 
				
			||||||
            state.serialize_field("crawled", &self.crawled)?;
 | 
					 | 
				
			||||||
            // to_string() calls the correct naming of site
 | 
					 | 
				
			||||||
            state.serialize_field("site", &self.site.to_string())?;
 | 
					 | 
				
			||||||
            state.end()
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// manual impl to make tracing look nicer
 | 
					// manual impl to make tracing look nicer
 | 
				
			||||||
impl Debug for Website {
 | 
					impl Debug for Website {
 | 
				
			||||||
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
 | 
					    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
 | 
				
			||||||
@@ -73,82 +47,13 @@ impl Website {
 | 
				
			|||||||
        self.crawled = true
 | 
					        self.crawled = true
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    pub fn get_url_as_string(site: &Url) -> String {
 | 
					 | 
				
			||||||
        let domain = match site.domain() {
 | 
					 | 
				
			||||||
            Some(s) => s.to_string(),
 | 
					 | 
				
			||||||
            None => {
 | 
					 | 
				
			||||||
                warn!("Failed to get domain of URL: {}, falling back to 'localhost'", site.to_string());
 | 
					 | 
				
			||||||
                "localhost".to_string()
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
        let path = site.path();
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        domain + path
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    pub fn get_url_as_b64_path(site: &Url) -> String {
 | 
					 | 
				
			||||||
        let domain = site.domain().unwrap_or("DOMAIN").to_string();
 | 
					 | 
				
			||||||
        let path = &CUSTOM_ENGINE.encode(site.path());
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        domain + path
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    #[instrument(skip_all)]
 | 
					 | 
				
			||||||
    pub async fn links_to(&self, other: Vec<Thing>, db: &Surreal<Client>) {
 | 
					 | 
				
			||||||
        let len = other.len();
 | 
					 | 
				
			||||||
        if len == 0 {
 | 
					 | 
				
			||||||
            return;
 | 
					 | 
				
			||||||
        }
 | 
					 | 
				
			||||||
        
 | 
					 | 
				
			||||||
        let from = &self.site;
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        // let to = other.site.to_string();
 | 
					 | 
				
			||||||
        trace!("Linking {} pages to {from}", other.len());
 | 
					 | 
				
			||||||
        counter!(LINK).increment(1);
 | 
					 | 
				
			||||||
        match db
 | 
					 | 
				
			||||||
            .query("COUNT(RELATE (SELECT id FROM website WHERE site = $in) -> links_to -> $out)")
 | 
					 | 
				
			||||||
            .bind(("in", from.clone()))
 | 
					 | 
				
			||||||
            .bind(("out", other))
 | 
					 | 
				
			||||||
            .await
 | 
					 | 
				
			||||||
        {
 | 
					 | 
				
			||||||
            Ok(mut e) => {
 | 
					 | 
				
			||||||
                // The relate could technically "fail" (not relate anything), this just means that
 | 
					 | 
				
			||||||
                // the query was ok.
 | 
					 | 
				
			||||||
                let _: Response = e;
 | 
					 | 
				
			||||||
                if let Ok(vec) = e.take(0) {
 | 
					 | 
				
			||||||
                    let _: Vec<usize> = vec;
 | 
					 | 
				
			||||||
                    if let Some(num) = vec.first() {
 | 
					 | 
				
			||||||
                        if *num == len {
 | 
					 | 
				
			||||||
                            trace!("Link for {from} OK - {num}/{len}");
 | 
					 | 
				
			||||||
                            return;
 | 
					 | 
				
			||||||
                        } else {
 | 
					 | 
				
			||||||
                            error!("Didn't link all the records. {num}/{len}. Surreal response: {:?}", e);
 | 
					 | 
				
			||||||
                            return;
 | 
					 | 
				
			||||||
                        }
 | 
					 | 
				
			||||||
                    }
 | 
					 | 
				
			||||||
                }
 | 
					 | 
				
			||||||
                error!("Linking request succeeded but couldn't verify the results.");
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
            Err(e) => {
 | 
					 | 
				
			||||||
                error!("{}", e.to_string());
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        }
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    // Insert ever item in the vec into surreal, crawled state will be preserved as TRUE
 | 
					    // Insert ever item in the vec into surreal, crawled state will be preserved as TRUE
 | 
				
			||||||
    // if already in the database as such or incoming data is TRUE.
 | 
					    // if already in the database as such or incoming data is TRUE.
 | 
				
			||||||
 | 
					    #[instrument(skip(db))]
 | 
				
			||||||
    pub async fn store_all(all: Vec<Self>, db: &Surreal<Client>) -> Vec<Thing> {
 | 
					    pub async fn store_all(all: Vec<Self>, db: &Surreal<Client>) -> Vec<Thing> {
 | 
				
			||||||
        counter!(STORE).increment(1);
 | 
					        counter!(STORE).increment(1);
 | 
				
			||||||
        let mut things = Vec::with_capacity(all.len());
 | 
					        let mut things = Vec::with_capacity(all.len());
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        // TODO this only allows for one thread to be in the database at a time.
 | 
					 | 
				
			||||||
        // This is currently required since otherwise we get write errors.
 | 
					 | 
				
			||||||
        // If the default `crawled` is set to false, we might not need to write any more
 | 
					 | 
				
			||||||
        // than just the name. `accessed_at` is fun but not needed.
 | 
					 | 
				
			||||||
        let now = Instant::now();
 | 
					 | 
				
			||||||
        let lock = LOCK.lock().await;
 | 
					 | 
				
			||||||
        counter!(TIME_SPENT_ON_LOCK).increment(now.elapsed().as_millis() as u64);
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        match db
 | 
					        match db
 | 
				
			||||||
            .query(
 | 
					            .query(
 | 
				
			||||||
                "INSERT INTO website $array
 | 
					                "INSERT INTO website $array
 | 
				
			||||||
@@ -169,7 +74,6 @@ impl Website {
 | 
				
			|||||||
                error!("{:?}", err);
 | 
					                error!("{:?}", err);
 | 
				
			||||||
            }
 | 
					            }
 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
        drop(lock);
 | 
					 | 
				
			||||||
        things
 | 
					        things
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										60
									
								
								src/filesystem.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										60
									
								
								src/filesystem.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,60 @@
 | 
				
			|||||||
 | 
					use std::{ffi::OsStr, path::PathBuf};
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					use tokio::fs;
 | 
				
			||||||
 | 
					use tracing::{debug, error, instrument, trace, warn};
 | 
				
			||||||
 | 
					use url::Url;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					#[instrument(skip(data))]
 | 
				
			||||||
 | 
					/// Returns whether or not the saved file should be parsed.
 | 
				
			||||||
 | 
					/// If the file is just data, like an image, it doesn't need to be parsed.
 | 
				
			||||||
 | 
					/// If it's html, then it does need to be parsed.
 | 
				
			||||||
 | 
					pub async fn store(data: &str, url: &Url) -> bool {
 | 
				
			||||||
 | 
					    // extract data from url to save it accurately
 | 
				
			||||||
 | 
					    let url_path = PathBuf::from("./downloaded/".to_string() + url.domain().unwrap_or("UnknownDomain") + url.path());
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    // if it's a file
 | 
				
			||||||
 | 
					    let (basepath, filename) = if url_path.extension().filter(valid_file_extension).is_some() {
 | 
				
			||||||
 | 
					        // get everything up till the file
 | 
				
			||||||
 | 
					        let basepath = url_path.ancestors().skip(1).take(1).collect::<PathBuf>();
 | 
				
			||||||
 | 
					        // get the file name
 | 
				
			||||||
 | 
					        let filename = url_path.file_name().expect("This should exist").to_string_lossy();
 | 
				
			||||||
 | 
					        trace!("Save path: {:?} and base path: {:?}", &url_path, &basepath);
 | 
				
			||||||
 | 
					        (basepath, filename.to_string())
 | 
				
			||||||
 | 
					    } else {
 | 
				
			||||||
 | 
					        (url_path.clone(), "index.html".into())
 | 
				
			||||||
 | 
					    };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    let should_parse = filename.ends_with(".html");
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    debug!("Writing at: {:?} {:?}", basepath, filename);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    // create the folders
 | 
				
			||||||
 | 
					    if let Err(err) = fs::create_dir_all(&basepath).await {
 | 
				
			||||||
 | 
					        error!("Dir creation: {err} {:?}", basepath);
 | 
				
			||||||
 | 
					    } else {
 | 
				
			||||||
 | 
					        if let Err(err) = fs::write(&basepath.join(filename), data).await {
 | 
				
			||||||
 | 
					            error!("File creation: {err} {:?}", url_path);
 | 
				
			||||||
 | 
					        }
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    should_parse
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					fn valid_file_extension(take: &&OsStr) -> bool {
 | 
				
			||||||
 | 
					    let los = take.to_string_lossy();
 | 
				
			||||||
 | 
					    let all = los.split('.');
 | 
				
			||||||
 | 
					    match all.last() {
 | 
				
			||||||
 | 
					        Some(s) => {
 | 
				
			||||||
 | 
					            // FIXME it's worth noting that the dumb tlds like .zip are in here,
 | 
				
			||||||
 | 
					            // which could cause problems
 | 
				
			||||||
 | 
					            let all_domains = include_str!("tlds-alpha-by-domain.txt");
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            // check if it is a domain
 | 
				
			||||||
 | 
					            match all_domains.lines().map(str::to_lowercase).find(|x| x==s.to_lowercase().as_str()) {
 | 
				
			||||||
 | 
					                Some(_) => false,
 | 
				
			||||||
 | 
					                None => true
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        None => false,
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										111
									
								
								src/main.rs
									
									
									
									
									
								
							
							
						
						
									
										111
									
								
								src/main.rs
									
									
									
									
									
								
							@@ -3,30 +3,29 @@
 | 
				
			|||||||
extern crate html5ever;
 | 
					extern crate html5ever;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
use std::{
 | 
					use std::{
 | 
				
			||||||
    fs::File,
 | 
					    collections::HashSet, fs::File, io::Read, net::{IpAddr, Ipv4Addr}
 | 
				
			||||||
    io::Read,
 | 
					 | 
				
			||||||
    net::{IpAddr, Ipv4Addr},
 | 
					 | 
				
			||||||
};
 | 
					};
 | 
				
			||||||
 | 
					
 | 
				
			||||||
use db::{connect, Website};
 | 
					use db::{connect, Website};
 | 
				
			||||||
use metrics::{counter, gauge};
 | 
					use metrics::{counter, gauge};
 | 
				
			||||||
use metrics_exporter_prometheus::PrometheusBuilder;
 | 
					use metrics_exporter_prometheus::PrometheusBuilder;
 | 
				
			||||||
use s3::S3;
 | 
					 | 
				
			||||||
use serde::Deserialize;
 | 
					use serde::Deserialize;
 | 
				
			||||||
use surrealdb::{engine::remote::ws::Client, Surreal};
 | 
					use surrealdb::{engine::remote::ws::Client, Surreal};
 | 
				
			||||||
use tokio::task::JoinSet;
 | 
					use tokio::task::JoinSet;
 | 
				
			||||||
use tracing::{debug, error, info, instrument, trace, trace_span};
 | 
					use tracing::{debug, error, info, instrument, level_filters::LevelFilter, trace, trace_span};
 | 
				
			||||||
use tracing_subscriber::{fmt, layer::SubscriberExt, EnvFilter, Layer, Registry};
 | 
					use tracing_subscriber::{fmt, layer::SubscriberExt, EnvFilter, Layer, Registry};
 | 
				
			||||||
 | 
					
 | 
				
			||||||
mod db;
 | 
					mod db;
 | 
				
			||||||
mod parser;
 | 
					mod parser;
 | 
				
			||||||
mod s3;
 | 
					mod filesystem;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
const GET_METRIC: &str = "total_gets";
 | 
					const GET_METRIC: &str = "total_gets";
 | 
				
			||||||
const GET_IN_FLIGHT: &str = "gets_in_flight";
 | 
					const GET_IN_FLIGHT: &str = "gets_in_flight";
 | 
				
			||||||
const SITES_CRAWLED: &str = "pages_crawled";
 | 
					const SITES_CRAWLED: &str = "pages_crawled";
 | 
				
			||||||
const BEING_PROCESSED: &str = "pages_being_processed";
 | 
					const BEING_PROCESSED: &str = "pages_being_processed";
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					const BATCH_SIZE: usize = 2;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
#[derive(Deserialize)]
 | 
					#[derive(Deserialize)]
 | 
				
			||||||
struct Config {
 | 
					struct Config {
 | 
				
			||||||
    surreal_ns: String,
 | 
					    surreal_ns: String,
 | 
				
			||||||
@@ -35,24 +34,24 @@ struct Config {
 | 
				
			|||||||
    surreal_username: String,
 | 
					    surreal_username: String,
 | 
				
			||||||
    surreal_password: String,
 | 
					    surreal_password: String,
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    s3_url: String,
 | 
					 | 
				
			||||||
    s3_bucket: String,
 | 
					 | 
				
			||||||
    s3_access_key: String,
 | 
					 | 
				
			||||||
    s3_secret_key: String,
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    crawl_filter: String,
 | 
					    crawl_filter: String,
 | 
				
			||||||
 | 
					    start_url: String,
 | 
				
			||||||
    budget: usize,
 | 
					    budget: usize,
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
#[tokio::main]
 | 
					#[tokio::main]
 | 
				
			||||||
async fn main() {
 | 
					async fn main() {
 | 
				
			||||||
 | 
					    println!("Logs and metrics are provided to the Grafana dashboard");
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    let writer = std::fs::OpenOptions::new()
 | 
					    let writer = std::fs::OpenOptions::new()
 | 
				
			||||||
        .append(true)
 | 
					        .append(true)
 | 
				
			||||||
        .create(true)
 | 
					        .create(true)
 | 
				
			||||||
        .open("./docker/logs/tracing.log")
 | 
					        .open("./docker/logs/tracing.log")
 | 
				
			||||||
        .expect("Couldn't make log file!");
 | 
					        .expect("Couldn't make log file!");
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    let filter = EnvFilter::from_default_env();
 | 
					    let filter = EnvFilter::builder()
 | 
				
			||||||
 | 
					        .with_default_directive(LevelFilter::DEBUG.into())
 | 
				
			||||||
 | 
					        .from_env_lossy();
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    let registry = Registry::default().with(
 | 
					    let registry = Registry::default().with(
 | 
				
			||||||
        fmt::layer()
 | 
					        fmt::layer()
 | 
				
			||||||
@@ -75,9 +74,8 @@ async fn main() {
 | 
				
			|||||||
        .install()
 | 
					        .install()
 | 
				
			||||||
        .expect("failed to install recorder/exporter");
 | 
					        .expect("failed to install recorder/exporter");
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    debug!("Starting...");
 | 
					    info!("Starting...");
 | 
				
			||||||
    // Would probably take these in as parameters from a cli
 | 
					
 | 
				
			||||||
    let starting_url = "https://en.wikipedia.org/";
 | 
					 | 
				
			||||||
    // When getting uncrawled pages, name must contain this variable. "" will effectively get ignored.
 | 
					    // When getting uncrawled pages, name must contain this variable. "" will effectively get ignored.
 | 
				
			||||||
    // let crawl_filter = "en.wikipedia.org/";
 | 
					    // let crawl_filter = "en.wikipedia.org/";
 | 
				
			||||||
    // let budget = 50;
 | 
					    // let budget = 50;
 | 
				
			||||||
@@ -88,13 +86,11 @@ async fn main() {
 | 
				
			|||||||
    let _ = file.read_to_string(&mut buf);
 | 
					    let _ = file.read_to_string(&mut buf);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    let config: Config = toml::from_str(&buf).expect("Failed to parse Crawler.toml");
 | 
					    let config: Config = toml::from_str(&buf).expect("Failed to parse Crawler.toml");
 | 
				
			||||||
 | 
					    let starting_url = &config.start_url;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    let db = connect(&config)
 | 
					    let db = connect(&config)
 | 
				
			||||||
        .await
 | 
					        .await
 | 
				
			||||||
        .expect("Failed to connect to surreal, aborting.");
 | 
					        .expect("Failed to connect to surreal, aborting.");
 | 
				
			||||||
    let s3 = S3::connect(&config)
 | 
					 | 
				
			||||||
        .await
 | 
					 | 
				
			||||||
        .expect("Failed to connect to minio, aborting.\n\nThis probably means you need to login to the minio console and get a new access key!\n\n(Probably here) http://localhost:9001/access-keys/new-account\n\n");
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
    let reqwest = reqwest::Client::builder()
 | 
					    let reqwest = reqwest::Client::builder()
 | 
				
			||||||
        // .use_rustls_tls()
 | 
					        // .use_rustls_tls()
 | 
				
			||||||
@@ -108,38 +104,27 @@ async fn main() {
 | 
				
			|||||||
    let pre_loop_span = span.enter();
 | 
					    let pre_loop_span = span.enter();
 | 
				
			||||||
    // Download the site
 | 
					    // Download the site
 | 
				
			||||||
    let site = Website::new(starting_url, false);
 | 
					    let site = Website::new(starting_url, false);
 | 
				
			||||||
    process(site, db.clone(), reqwest.clone(), s3.clone()).await;
 | 
					    process(site, db.clone(), reqwest.clone()).await;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    drop(pre_loop_span);
 | 
					    drop(pre_loop_span);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    let span = trace_span!("Loop");
 | 
					    let span = trace_span!("Loop");
 | 
				
			||||||
    let span = span.enter();
 | 
					    let span = span.enter();
 | 
				
			||||||
    while crawled < config.budget {
 | 
					    while crawled < config.budget {
 | 
				
			||||||
        let get_num = if config.budget - crawled < 100 {
 | 
					        let uncrawled = get_uncrawled_links(&db, config.budget - crawled, config.crawl_filter.clone()).await;
 | 
				
			||||||
            config.budget - crawled
 | 
					 | 
				
			||||||
        } else {
 | 
					 | 
				
			||||||
            100
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        let uncrawled = get_uncrawled_links(&db, get_num, config.crawl_filter.clone()).await;
 | 
					 | 
				
			||||||
        if uncrawled.is_empty() {
 | 
					        if uncrawled.is_empty() {
 | 
				
			||||||
            info!("Had more budget but finished crawling everything.");
 | 
					            info!("Had more budget but finished crawling everything.");
 | 
				
			||||||
            return;
 | 
					            return;
 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
        debug!("Crawling {} pages...", uncrawled.len());
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        let span = trace_span!("Crawling");
 | 
					 | 
				
			||||||
        let _ = span.enter();
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
        {
 | 
					        {
 | 
				
			||||||
            let mut futures = JoinSet::new();
 | 
					            let mut futures = JoinSet::new();
 | 
				
			||||||
            for site in uncrawled {
 | 
					            for site in uncrawled {
 | 
				
			||||||
                gauge!(BEING_PROCESSED).increment(1);
 | 
					                gauge!(BEING_PROCESSED).increment(1);
 | 
				
			||||||
                futures.spawn(process(site, db.clone(), reqwest.clone(), s3.clone()));
 | 
					                futures.spawn(process(site, db.clone(), reqwest.clone()));
 | 
				
			||||||
                // let percent = format!("{:.2}%", (crawled as f32 / budget as f32) * 100f32);
 | 
					                // let percent = format!("{:.2}%", (crawled as f32 / budget as f32) * 100f32);
 | 
				
			||||||
                // info!("Crawled {crawled} out of {budget} pages. ({percent})");
 | 
					                // info!("Crawled {crawled} out of {budget} pages. ({percent})");
 | 
				
			||||||
            }
 | 
					            }
 | 
				
			||||||
            debug!("Joining {} futures...", futures.len());
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
            let c = counter!(SITES_CRAWLED);
 | 
					            let c = counter!(SITES_CRAWLED);
 | 
				
			||||||
            // As futures complete runs code in while block
 | 
					            // As futures complete runs code in while block
 | 
				
			||||||
@@ -152,13 +137,22 @@ async fn main() {
 | 
				
			|||||||
    }
 | 
					    }
 | 
				
			||||||
    drop(span);
 | 
					    drop(span);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    debug!("Done");
 | 
					    if let Ok(mut ok) = db.query("count(select id from website where crawled = true)").await {
 | 
				
			||||||
 | 
					        let res = ok.take::<Option<usize>>(0);
 | 
				
			||||||
 | 
					        if let Ok(i) = res {
 | 
				
			||||||
 | 
					            if let Some(n) = i {
 | 
				
			||||||
 | 
					                info!("Total crawled pages now equals {n}");
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        }
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    info!("Done");
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
#[instrument(skip(db, s3, reqwest))]
 | 
					#[instrument(skip(db, reqwest))]
 | 
				
			||||||
/// Downloads and crawls and stores a webpage.
 | 
					/// Downloads and crawls and stores a webpage.
 | 
				
			||||||
/// It is acceptable to clone `db`, `reqwest`, and `s3` because they all use `Arc`s internally. - Noted by Oliver
 | 
					/// It is acceptable to clone `db`, `reqwest`, and `s3` because they all use `Arc`s internally. - Noted by Oliver
 | 
				
			||||||
async fn process(mut site: Website, db: Surreal<Client>, reqwest: reqwest::Client, s3: S3) {
 | 
					async fn process(mut site: Website, db: Surreal<Client>, reqwest: reqwest::Client) {
 | 
				
			||||||
    
 | 
					    
 | 
				
			||||||
    // METRICS
 | 
					    // METRICS
 | 
				
			||||||
    trace!("Process: {}", &site.site);
 | 
					    trace!("Process: {}", &site.site);
 | 
				
			||||||
@@ -172,31 +166,44 @@ async fn process(mut site: Website, db: Surreal<Client>, reqwest: reqwest::Clien
 | 
				
			|||||||
    // Send the http request (get)
 | 
					    // Send the http request (get)
 | 
				
			||||||
    if let Ok(response) = request_builder.send().await {
 | 
					    if let Ok(response) = request_builder.send().await {
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        // METRICS
 | 
					        // TODO if this will fail if the object we are downloading is
 | 
				
			||||||
        g.decrement(1);
 | 
					        // larger than the memory of the device it's running on.
 | 
				
			||||||
        counter!(GET_METRIC).increment(1);
 | 
					        // We should store it *as* we download it then parse it in-place.
 | 
				
			||||||
 | 
					 | 
				
			||||||
        // Get body from response
 | 
					        // Get body from response
 | 
				
			||||||
        let data = response
 | 
					        let data = response
 | 
				
			||||||
            .text()
 | 
					            .text()
 | 
				
			||||||
            .await
 | 
					            .await
 | 
				
			||||||
            .expect("Failed to read http response's body!");
 | 
					            .expect("Failed to read http response's body!");
 | 
				
			||||||
        // Store document
 | 
					 | 
				
			||||||
        s3.store(&data, &site.site).await;
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
        // Parse document and get relationships
 | 
					        // METRICS
 | 
				
			||||||
        let sites = parser::parse(&site, &data).await;
 | 
					        g.decrement(1);
 | 
				
			||||||
 | 
					        counter!(GET_METRIC).increment(1);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        // Store document
 | 
				
			||||||
 | 
					        let should_parse = filesystem::store(&data, &site.site).await;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        if should_parse {
 | 
				
			||||||
 | 
					            // Parse document and get relationships
 | 
				
			||||||
 | 
					            let sites = parser::parse(&site, &data).await;
 | 
				
			||||||
        
 | 
					        
 | 
				
			||||||
 | 
					            // De-duplicate this list
 | 
				
			||||||
 | 
					            let prev_len = sites.len();
 | 
				
			||||||
 | 
					            let set = sites.into_iter().fold(HashSet::new(), |mut set,item| {
 | 
				
			||||||
 | 
					                set.insert(item);
 | 
				
			||||||
 | 
					                set
 | 
				
			||||||
 | 
					            });
 | 
				
			||||||
 | 
					            let de_dupe_sites: Vec<Website> = set.into_iter().collect();
 | 
				
			||||||
 | 
					            let diff = prev_len - de_dupe_sites.len();
 | 
				
			||||||
 | 
					            trace!("Saved {diff} from being entered into the db by de-duping");
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            // Store all the other sites so that we can link to them.
 | 
				
			||||||
 | 
					            let _ = Website::store_all(de_dupe_sites, &db).await;
 | 
				
			||||||
 | 
					        }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        // update self in db
 | 
					        // update self in db
 | 
				
			||||||
        site.set_crawled();
 | 
					        site.set_crawled();
 | 
				
			||||||
        Website::store_all(vec![site], &db).await;
 | 
					        Website::store_all(vec![site], &db).await;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        // Store all the other sites so that we can link to them.
 | 
					 | 
				
			||||||
        // let mut links_to = Vec::new();
 | 
					 | 
				
			||||||
        let _ = Website::store_all(sites, &db).await;
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        // Make the database's links reflect the html links between sites
 | 
					 | 
				
			||||||
        // site.links_to(others, &db).await;
 | 
					 | 
				
			||||||
    } else {
 | 
					    } else {
 | 
				
			||||||
        error!("Failed to get: {}", &site.site);
 | 
					        error!("Failed to get: {}", &site.site);
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
@@ -209,9 +216,11 @@ async fn get_uncrawled_links(
 | 
				
			|||||||
    mut count: usize,
 | 
					    mut count: usize,
 | 
				
			||||||
    filter: String,
 | 
					    filter: String,
 | 
				
			||||||
) -> Vec<Website> {
 | 
					) -> Vec<Website> {
 | 
				
			||||||
    if count > 100 {
 | 
					
 | 
				
			||||||
        count = 100
 | 
					    if count > BATCH_SIZE {
 | 
				
			||||||
 | 
					        count = BATCH_SIZE;
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    debug!("Getting uncrawled links");
 | 
					    debug!("Getting uncrawled links");
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    let mut response = db
 | 
					    let mut response = db
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -5,18 +5,21 @@ use html5ever::tokenizer::{BufferQueue, TokenizerResult};
 | 
				
			|||||||
use html5ever::tokenizer::{StartTag, TagToken};
 | 
					use html5ever::tokenizer::{StartTag, TagToken};
 | 
				
			||||||
use html5ever::tokenizer::{Token, TokenSink, TokenSinkResult, Tokenizer, TokenizerOpts};
 | 
					use html5ever::tokenizer::{Token, TokenSink, TokenSinkResult, Tokenizer, TokenizerOpts};
 | 
				
			||||||
use html5ever::{local_name, tendril::*};
 | 
					use html5ever::{local_name, tendril::*};
 | 
				
			||||||
use tracing::instrument;
 | 
					use tracing::{debug, error, instrument, trace, warn};
 | 
				
			||||||
 | 
					use url::Url;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
use crate::db::Website;
 | 
					use crate::db::Website;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
impl TokenSink for Website {
 | 
					impl TokenSink for Website {
 | 
				
			||||||
    type Handle = Vec<Website>;
 | 
					    type Handle = Vec<Website>;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    #[instrument(skip(token, _line_number))]
 | 
				
			||||||
    fn process_token(&self, token: Token, _line_number: u64) -> TokenSinkResult<Self::Handle> {
 | 
					    fn process_token(&self, token: Token, _line_number: u64) -> TokenSinkResult<Self::Handle> {
 | 
				
			||||||
        match token {
 | 
					        match token {
 | 
				
			||||||
            TagToken(tag) => {
 | 
					            TagToken(tag) => {
 | 
				
			||||||
                if tag.kind == StartTag {
 | 
					                if tag.kind == StartTag {
 | 
				
			||||||
                    match tag.name {
 | 
					                    match tag.name {
 | 
				
			||||||
 | 
					                        // this should be all the html elements that have links
 | 
				
			||||||
                        local_name!("a")
 | 
					                        local_name!("a")
 | 
				
			||||||
                        | local_name!("audio")
 | 
					                        | local_name!("audio")
 | 
				
			||||||
                        | local_name!("area")
 | 
					                        | local_name!("area")
 | 
				
			||||||
@@ -31,23 +34,18 @@ impl TokenSink for Website {
 | 
				
			|||||||
                                let attr_name = attr.name.local.to_string();
 | 
					                                let attr_name = attr.name.local.to_string();
 | 
				
			||||||
                                if attr_name == "src" || attr_name == "href" || attr_name == "data"
 | 
					                                if attr_name == "src" || attr_name == "href" || attr_name == "data"
 | 
				
			||||||
                                {
 | 
					                                {
 | 
				
			||||||
                                    // Get clone of the current site object
 | 
					                                    trace!("Found `{}` in html `{}` tag", &attr.value, tag.name);
 | 
				
			||||||
                                    let mut web = self.clone();
 | 
					                                    let url = try_get_url(&self.site, &attr.value);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                                    // Set url
 | 
					                                    if let Some(mut parsed) = url {
 | 
				
			||||||
                                    let mut url = web.site;
 | 
					                                        parsed.set_query(None);
 | 
				
			||||||
                                    url.set_fragment(None); // removes #xyz
 | 
					                                        parsed.set_fragment(None);
 | 
				
			||||||
                                    let joined = url
 | 
					                                        debug!("Final cleaned URL: `{}`", parsed.to_string());
 | 
				
			||||||
                                        .join(&attr.value)
 | 
					                                        let web = Website::new(&parsed.to_string(), false);
 | 
				
			||||||
                                        .expect("Failed to join url during parsing!");
 | 
					                                        links.push(web);
 | 
				
			||||||
                                    web.site = joined;
 | 
					                                    }
 | 
				
			||||||
 | 
					 | 
				
			||||||
                                    web.crawled = false;
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
                                    links.push(web);
 | 
					 | 
				
			||||||
                                }
 | 
					                                }
 | 
				
			||||||
                            }
 | 
					                            }
 | 
				
			||||||
 | 
					 | 
				
			||||||
                            return TokenSinkResult::Script(links);
 | 
					                            return TokenSinkResult::Script(links);
 | 
				
			||||||
                        }
 | 
					                        }
 | 
				
			||||||
                        local_name!("button") | local_name!("meta") | local_name!("iframe") => {
 | 
					                        local_name!("button") | local_name!("meta") | local_name!("iframe") => {
 | 
				
			||||||
@@ -76,7 +74,7 @@ pub async fn parse(site: &Website, data: &str) -> Vec<Website> {
 | 
				
			|||||||
    token_buffer.push_back(
 | 
					    token_buffer.push_back(
 | 
				
			||||||
        chunk
 | 
					        chunk
 | 
				
			||||||
            .try_reinterpret::<fmt::UTF8>()
 | 
					            .try_reinterpret::<fmt::UTF8>()
 | 
				
			||||||
            .expect("Failed to reinterprt chunk!"),
 | 
					            .expect("Failed to reinterpret chunk!"),
 | 
				
			||||||
    );
 | 
					    );
 | 
				
			||||||
    // create the tokenizer
 | 
					    // create the tokenizer
 | 
				
			||||||
    let tokenizer = Tokenizer::new(site.clone(), TokenizerOpts::default());
 | 
					    let tokenizer = Tokenizer::new(site.clone(), TokenizerOpts::default());
 | 
				
			||||||
@@ -92,3 +90,56 @@ pub async fn parse(site: &Website, data: &str) -> Vec<Website> {
 | 
				
			|||||||
    other_sites
 | 
					    other_sites
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					#[instrument]
 | 
				
			||||||
 | 
					fn try_get_url(parent: &Url, link: &str) -> Option<Url> {
 | 
				
			||||||
 | 
					    match Url::parse(link) {
 | 
				
			||||||
 | 
					        Ok(ok) => Some(ok),
 | 
				
			||||||
 | 
					        Err(e) => {
 | 
				
			||||||
 | 
					            if link.starts_with('#') {
 | 
				
			||||||
 | 
					                trace!("Rejecting # url");
 | 
				
			||||||
 | 
					                None
 | 
				
			||||||
 | 
					            } else if link.starts_with("//") {
 | 
				
			||||||
 | 
					                // if a url starts with "//" is assumed that it will adopt
 | 
				
			||||||
 | 
					                // the same scheme as it's parent
 | 
				
			||||||
 | 
					                // https://stackoverflow.com/questions/9646407/two-forward-slashes-in-a-url-src-href-attribute
 | 
				
			||||||
 | 
					                let scheme = parent.scheme();
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                match Url::parse(&format!("{scheme}://{}", link)) {
 | 
				
			||||||
 | 
					                    Ok(url) => Some(url),
 | 
				
			||||||
 | 
					                    Err(err) => {
 | 
				
			||||||
 | 
					                        error!("Failed parsing realative scheme url: {}", err);
 | 
				
			||||||
 | 
					                        None
 | 
				
			||||||
 | 
					                    }
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					            } else {
 | 
				
			||||||
 | 
					                // # This is some sort of realative url, gonna try patching it up into an absolute
 | 
				
			||||||
 | 
					                // url
 | 
				
			||||||
 | 
					                match e {
 | 
				
			||||||
 | 
					                    url::ParseError::RelativeUrlWithoutBase => {
 | 
				
			||||||
 | 
					                        // Is: scheme://host:port
 | 
				
			||||||
 | 
					                        let origin = parent.origin().ascii_serialization();
 | 
				
			||||||
 | 
					                        let url = origin.clone() + link;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                        trace!("Built `{url}` from `{origin} + {}`", link.to_string());
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                        if let Ok(url) = Url::parse(&url) {
 | 
				
			||||||
 | 
					                            trace!("Saved relative url `{}` AS: `{}`", link, url);
 | 
				
			||||||
 | 
					                            Some(url)
 | 
				
			||||||
 | 
					                        } else {
 | 
				
			||||||
 | 
					                            error!(
 | 
				
			||||||
 | 
					                                "Failed to reconstruct a url from relative url: `{}` on site: `{}`",
 | 
				
			||||||
 | 
					                                link,
 | 
				
			||||||
 | 
					                                parent.to_string()
 | 
				
			||||||
 | 
					                            );
 | 
				
			||||||
 | 
					                            None
 | 
				
			||||||
 | 
					                        }
 | 
				
			||||||
 | 
					                    }
 | 
				
			||||||
 | 
					                    _ => {
 | 
				
			||||||
 | 
					                        error!("MISC error: {:?} {:?}", e, link);
 | 
				
			||||||
 | 
					                        None
 | 
				
			||||||
 | 
					                    }
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        }
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										100
									
								
								src/s3.rs
									
									
									
									
									
								
							
							
						
						
									
										100
									
								
								src/s3.rs
									
									
									
									
									
								
							@@ -1,100 +0,0 @@
 | 
				
			|||||||
use metrics::counter;
 | 
					 | 
				
			||||||
use minio::s3::{
 | 
					 | 
				
			||||||
    args::{BucketExistsArgs, MakeBucketArgs},
 | 
					 | 
				
			||||||
    client::ClientBuilder,
 | 
					 | 
				
			||||||
    creds::StaticProvider,
 | 
					 | 
				
			||||||
    error::Error,
 | 
					 | 
				
			||||||
    http::BaseUrl,
 | 
					 | 
				
			||||||
    Client,
 | 
					 | 
				
			||||||
};
 | 
					 | 
				
			||||||
use tracing::{instrument, trace, warn};
 | 
					 | 
				
			||||||
use url::Url;
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
use crate::{db::Website, Config};
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
const S3_ROUND_TRIP_METRIC: &str = "s3_trips";
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
#[derive(Clone)]
 | 
					 | 
				
			||||||
pub struct S3 {
 | 
					 | 
				
			||||||
    bucket_name: String,
 | 
					 | 
				
			||||||
    client: Client,
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
impl S3 {
 | 
					 | 
				
			||||||
    #[instrument(skip_all, name = "S3")]
 | 
					 | 
				
			||||||
    pub async fn connect(config: &Config) -> Result<Self, Error> {
 | 
					 | 
				
			||||||
        let base_url = config
 | 
					 | 
				
			||||||
            .s3_url
 | 
					 | 
				
			||||||
            .parse::<BaseUrl>()
 | 
					 | 
				
			||||||
            .expect("Failed to parse url into BaseUrl");
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        let static_provider =
 | 
					 | 
				
			||||||
            StaticProvider::new(&config.s3_access_key, &config.s3_secret_key, None);
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        let client = ClientBuilder::new(base_url)
 | 
					 | 
				
			||||||
            .provider(Some(Box::new(static_provider)))
 | 
					 | 
				
			||||||
            .build()?;
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        trace!("Checking bucket...");
 | 
					 | 
				
			||||||
        let exists = client
 | 
					 | 
				
			||||||
            .bucket_exists(
 | 
					 | 
				
			||||||
                &BucketExistsArgs::new(&config.s3_bucket)
 | 
					 | 
				
			||||||
                    .expect("Failed to check if bucket exists"),
 | 
					 | 
				
			||||||
            )
 | 
					 | 
				
			||||||
            .await?;
 | 
					 | 
				
			||||||
        counter!(S3_ROUND_TRIP_METRIC).increment(1);
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if !exists {
 | 
					 | 
				
			||||||
            trace!("Creating bucket...");
 | 
					 | 
				
			||||||
            client
 | 
					 | 
				
			||||||
                .make_bucket(
 | 
					 | 
				
			||||||
                    &MakeBucketArgs::new(&config.s3_bucket).expect("Failed to create bucket!"),
 | 
					 | 
				
			||||||
                )
 | 
					 | 
				
			||||||
                .await?;
 | 
					 | 
				
			||||||
        }
 | 
					 | 
				
			||||||
        counter!(S3_ROUND_TRIP_METRIC).increment(1);
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        trace!("Connection successful");
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        Ok(Self {
 | 
					 | 
				
			||||||
            bucket_name: config.s3_bucket.to_owned(),
 | 
					 | 
				
			||||||
            client,
 | 
					 | 
				
			||||||
        })
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    #[instrument(name = "s3_store", skip_all)]
 | 
					 | 
				
			||||||
    pub async fn store(&self, data: &str, url: &Url) {
 | 
					 | 
				
			||||||
        let counter = counter!(S3_ROUND_TRIP_METRIC);
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        let filename = Website::get_url_as_string(url);
 | 
					 | 
				
			||||||
        trace!("Storing {} as {filename}", url.to_string());
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        counter.increment(1);
 | 
					 | 
				
			||||||
        match &self
 | 
					 | 
				
			||||||
            .client
 | 
					 | 
				
			||||||
            .put_object_content(&self.bucket_name, &filename, data.to_owned())
 | 
					 | 
				
			||||||
            .send()
 | 
					 | 
				
			||||||
            .await
 | 
					 | 
				
			||||||
        {
 | 
					 | 
				
			||||||
            Ok(_) => {}
 | 
					 | 
				
			||||||
            Err(err) => match err {
 | 
					 | 
				
			||||||
                Error::InvalidObjectName(_) => {
 | 
					 | 
				
			||||||
                    // This code will really only run if the url has non-english chars
 | 
					 | 
				
			||||||
                    warn!("Tried storing invalid object name, retrying with Base64 encoding. Last try.");
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
                    let filename: String = Website::get_url_as_b64_path(url);
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
                    counter.increment(1);
 | 
					 | 
				
			||||||
                    let _ = &self
 | 
					 | 
				
			||||||
                        .client
 | 
					 | 
				
			||||||
                        .put_object_content(&self.bucket_name, &filename, data.to_owned())
 | 
					 | 
				
			||||||
                        .send()
 | 
					 | 
				
			||||||
                        .await
 | 
					 | 
				
			||||||
                        .unwrap();
 | 
					 | 
				
			||||||
                }
 | 
					 | 
				
			||||||
                _ => {}
 | 
					 | 
				
			||||||
            },
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
							
								
								
									
										1444
									
								
								src/tlds-alpha-by-domain.txt
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1444
									
								
								src/tlds-alpha-by-domain.txt
									
									
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
		Reference in New Issue
	
	Block a user