working filesystem storage
This commit is contained in:
		@@ -3,7 +3,7 @@ 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.19.1"
 | 
					surreal_db = "v1.19.2"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# Crawler config
 | 
					# Crawler config
 | 
				
			||||||
crawl_filter = "en.wikipedia.com" 
 | 
					crawl_filter = "en.wikipedia.com" 
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -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
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -7,15 +7,11 @@ scrape_configs:
 | 
				
			|||||||
    static_configs:
 | 
					    static_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']
 | 
					        #- 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']
 | 
					 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,7 +1,7 @@
 | 
				
			|||||||
use std::{ffi::OsStr, path::PathBuf};
 | 
					use std::{ffi::OsStr, path::PathBuf};
 | 
				
			||||||
 | 
					
 | 
				
			||||||
use tokio::fs;
 | 
					use tokio::fs;
 | 
				
			||||||
use tracing::{error, instrument, trace, warn};
 | 
					use tracing::{debug, error, info, instrument, trace, warn};
 | 
				
			||||||
use url::Url;
 | 
					use url::Url;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
#[instrument(skip(data))]
 | 
					#[instrument(skip(data))]
 | 
				
			||||||
@@ -21,6 +21,8 @@ pub async fn store(data: &str, url: &Url) {
 | 
				
			|||||||
        (url_path.clone(), "index.html".into())
 | 
					        (url_path.clone(), "index.html".into())
 | 
				
			||||||
    };
 | 
					    };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    info!("Writing at: {:?} {:?}", basepath, filename);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    // create the folders
 | 
					    // create the folders
 | 
				
			||||||
    if let Err(err) = fs::create_dir_all(&basepath).await {
 | 
					    if let Err(err) = fs::create_dir_all(&basepath).await {
 | 
				
			||||||
        error!("Dir creation: {err} {:?}", basepath);
 | 
					        error!("Dir creation: {err} {:?}", basepath);
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -19,8 +19,7 @@ impl TokenSink for Website {
 | 
				
			|||||||
            TagToken(tag) => {
 | 
					            TagToken(tag) => {
 | 
				
			||||||
                if tag.kind == StartTag {
 | 
					                if tag.kind == StartTag {
 | 
				
			||||||
                    match tag.name {
 | 
					                    match tag.name {
 | 
				
			||||||
                        // this should be all the html
 | 
					                        // this should be all the html elements that have links
 | 
				
			||||||
                        // elements that have links
 | 
					 | 
				
			||||||
                        local_name!("a")
 | 
					                        local_name!("a")
 | 
				
			||||||
                        | local_name!("audio")
 | 
					                        | local_name!("audio")
 | 
				
			||||||
                        | local_name!("area")
 | 
					                        | local_name!("area")
 | 
				
			||||||
@@ -35,37 +34,9 @@ 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"
 | 
				
			||||||
                                {
 | 
					                                {
 | 
				
			||||||
                                    let url: Option<Url> = match Url::parse(&attr.value) {
 | 
					                                    trace!("Found `{}` in html `{}` tag", &attr.value, tag.name);
 | 
				
			||||||
                                        Ok(ok) => {
 | 
					                                    let url = try_get_url(&self.site, &attr.value);
 | 
				
			||||||
                                            trace!("Found `{}` in the html on `{}` tag", ok.to_string(), tag.name);
 | 
					
 | 
				
			||||||
                                            Some(ok)
 | 
					 | 
				
			||||||
                                        },
 | 
					 | 
				
			||||||
                                        Err(e) => {
 | 
					 | 
				
			||||||
                                            if attr.value.starts_with('#') {
 | 
					 | 
				
			||||||
                                                trace!("Rejecting # url");
 | 
					 | 
				
			||||||
                                                None
 | 
					 | 
				
			||||||
                                            } else {
 | 
					 | 
				
			||||||
                                                match e {
 | 
					 | 
				
			||||||
                                                    url::ParseError::RelativeUrlWithoutBase => {
 | 
					 | 
				
			||||||
                                                        let origin = self.site.origin().ascii_serialization();
 | 
					 | 
				
			||||||
                                                        let url = origin.clone() + &attr.value;
 | 
					 | 
				
			||||||
                                                        trace!("Built `{url}` from `{origin} + {}`", &attr.value.to_string());
 | 
					 | 
				
			||||||
                                                        if let Ok(url) = Url::parse(&url) {
 | 
					 | 
				
			||||||
                                                            trace!("Saved relative url `{}` AS: `{}`", &attr.value, url);
 | 
					 | 
				
			||||||
                                                            Some(url)
 | 
					 | 
				
			||||||
                                                        } else {
 | 
					 | 
				
			||||||
                                                            error!("Failed to reconstruct a url from relative url: `{}` on site: `{}`", &attr.value, self.site.to_string());
 | 
					 | 
				
			||||||
                                                            None
 | 
					 | 
				
			||||||
                                                        }
 | 
					 | 
				
			||||||
                                                    }, 
 | 
					 | 
				
			||||||
                                                    _ => {
 | 
					 | 
				
			||||||
                                                        error!("MISC error: {:?} {:?}", e, &attr.value);
 | 
					 | 
				
			||||||
                                                        None
 | 
					 | 
				
			||||||
                                                    },
 | 
					 | 
				
			||||||
                                                }
 | 
					 | 
				
			||||||
                                            }
 | 
					 | 
				
			||||||
                                        },
 | 
					 | 
				
			||||||
                                    };
 | 
					 | 
				
			||||||
                                    if let Some(mut parsed) = url {
 | 
					                                    if let Some(mut parsed) = url {
 | 
				
			||||||
                                        parsed.set_query(None);
 | 
					                                        parsed.set_query(None);
 | 
				
			||||||
                                        parsed.set_fragment(None);
 | 
					                                        parsed.set_fragment(None);
 | 
				
			||||||
@@ -119,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
 | 
				
			||||||
 | 
					                    }
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        }
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user