Merge pull request #46 from dimacurrentai/maxminddb

Added `step08` with `maxminddb`.

Author: dkorolev

Dockerfile.template

@@ -4,6 +4,8 @@ FROM alpine AS build
 # Intentionally not installing `cargo`, etc. But it does need `clang`.
 RUN apk add rustup clang musl
 
+# PLACEHOLDER_BUILD
+
 # Init local rust via `rustup`, including static targets builder.
 RUN rustup-init -y -t "$(uname -m)-unknown-linux-musl" --no-modify-path 
 RUN mv /root/.cargo/bin/* /usr/local/bin/
@@ -23,4 +25,5 @@ RUN cargo test --release -- --test-threads=1
 # The resulting container with the static binary only.
 FROM scratch
 COPY --from=build /binary /binary
+# PLACEHOLDER_RUN
 ENTRYPOINT ["/binary"]

step08_maxminddb/.gitignore

@@ -0,0 +1,2 @@
+Dockerfile
+GeoLite2-*.mmdb

step08_maxminddb/code/Cargo.toml

@@ -0,0 +1,8 @@
+[package]
+name = "maxmindb"
+edition = "2021"
+
+[dependencies]
+maxminddb = "0.21.0"
+tokio = { version = "1.0", features = ["full"] }
+trust-dns-resolver = "0.23.2"

step08_maxminddb/code/src/main.rs

@@ -0,0 +1,52 @@
+use maxminddb::Reader;
+use std::net::IpAddr;
+use std::path::Path;
+use trust_dns_resolver::config::{NameServerConfig, Protocol, ResolverConfig, ResolverOpts};
+use trust_dns_resolver::TokioAsyncResolver;
+
+const OPENDNS_SERVER: &str = "208.67.222.222:53";
+const OPENDNS_MYIP_DOMAIN: &str = "myip.opendns.com.";
+const GEOIP_DB_PATH: &str = "GeoLite2-Country.mmdb";
+
+#[tokio::main]
+async fn main() -> Result<(), Box<dyn std::error::Error>> {
+  if let Ok(ip) = get_self_ip().await {
+    println!("Public IP: {}", ip);
+    if let Err(e) = map_ip_to_geo(ip) {
+      println!("Error mapping IP to geo: {}", e);
+    }
+  } else {
+    println!("Failed to get your IP address");
+  }
+  Ok(())
+}
+
+async fn get_self_ip() -> Result<IpAddr, Box<dyn std::error::Error>> {
+  let mut config = ResolverConfig::new();
+  config.add_name_server(NameServerConfig {
+    socket_addr: OPENDNS_SERVER.parse()?,
+    protocol: Protocol::Udp,
+    tls_dns_name: None,
+    trust_negative_responses: false,
+    bind_addr: None,
+  });
+  let resolver = TokioAsyncResolver::tokio(config, ResolverOpts::default());
+  let response = resolver.lookup_ip(OPENDNS_MYIP_DOMAIN).await?;
+  response.iter().next().ok_or_else(|| "No IP found".into())
+}
+
+fn map_ip_to_geo(ip: IpAddr) -> Result<(), Box<dyn std::error::Error>> {
+  let db_path = Path::new(GEOIP_DB_PATH);
+  if !db_path.exists() {
+    return Err(format!("GeoIP database file not found: {}", GEOIP_DB_PATH).into());
+  }
+
+  let reader = Reader::open_readfile(GEOIP_DB_PATH)?;
+  let result: maxminddb::geoip2::City = reader.lookup(ip)?;
+
+  println!("{:#?}", result);
+
+  let iso = result.country.unwrap().iso_code;
+  println!("{iso:?}");
+  Ok(())
+}

step08_maxminddb/run.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+
+set -e
+
+cat ../Dockerfile.template | grep -B 999999 '^# PLACEHOLDER_BUILD$' >Dockerfile
+
+echo 'RUN apk add wget' >>Dockerfile
+echo 'RUN wget https://github.com/P3TERX/GeoLite.mmdb/raw/download/GeoLite2-Country.mmdb' >>Dockerfile
+
+cat ../Dockerfile.template | grep -A 999999 '^# PLACEHOLDER_BUILD$' | grep -B 999999 '^# PLACEHOLDER_RUN$' >>Dockerfile
+
+echo 'COPY --from=build /GeoLite2-Country.mmdb /' >>Dockerfile
+
+cat ../Dockerfile.template | grep -A 999999 '^# PLACEHOLDER_RUN$' ../Dockerfile.template >>Dockerfile
+
+docker build . -t demo
+docker run --rm -t demo