network-infra-utility 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 960f1593215711ad0837083bf676f357760fab6d2cf22947ce482328d1be0cb8
-  data.tar.gz: 051314c8fe9620ee9a0540e05559e92358002ef6da53dd848ce0d7a62cae1258
+  metadata.gz: e8a2b1b8402f7fdb0ba39e7a6d01b384d624e6906530a27c575ae6371bedabfd
+  data.tar.gz: 7f0cb17756a53dcc31386fa5aebc786ba3640ca0cd86443c539237c41239d0b8
 SHA512:
-  metadata.gz: d7a36a0695efcc11cc60dae6529502ceff7cad727ec4315e45e8966ba28b127baafe76ddc349f49bb859c03b0c33f7ec84a95580905e89b5b470d223bdaff68e
-  data.tar.gz: b1af08569c4a1a722d09e27e4658c7e7376217d50479cc0bcd3f15c7fe7bde35e9be9be2b30301c8d63dd4d58ab30065bb1b9cc8d825214360e78a1183aa3b74
+  metadata.gz: 75901331bb496d6bbd54935256bf01990ddd2221b97415a9210a1133ba03516abbb6ebc4d4550197e17ebeee9593dc0920f90b146fd0ae7f546836bd9a92f4e0
+  data.tar.gz: 66e83826b05f0029b781c93d03daf2f7faaf01058d1a404f34bf25a1bf2a68064c43ac95912148e1a9793d7ff05fd3fe80def8265d18f2aa71ceb69f9bc602ce

data/README.md

@@ -1,13 +1,3 @@
----
-AIGC:
-  ContentProducer: '001191110102MAD55U9H0F10002'
-  ContentPropagator: '001191110102MAD55U9H0F10002'
-  Label: '1'
-  ProduceID: 'c63508f8-64ef-4fa8-b091-e1efa3d7ce89'
-  PropagateID: 'c63508f8-64ef-4fa8-b091-e1efa3d7ce89'
-  ReservedCode1: '145627d0-3e13-4b6f-ac83-23d7f58c195c'
-  ReservedCode2: '145627d0-3e13-4b6f-ac83-23d7f58c195c'
----
 
 # Network Infrastructure Utility
 
@@ -53,5 +43,3 @@ bundle exec rake           # 两者都跑（默认）
 ## License
 
 采用 [GNU Affero General Public License v3.0 或更高版本](https://www.gnu.org/licenses/agpl-3.0.html)（AGPL-3.0-or-later），详见 [LICENSE.txt](LICENSE.txt)。
-
-> AI生成

data/bin/geo-api

@@ -0,0 +1,114 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+# coding: utf-8
+#
+# geo-api — network-infra-utility 的 IP 归属信息查询命令行服务
+#
+# 把 service/geodb 的 GeoAPI 封装为可直接调用的命令行服务，默认监听
+# 0.0.0.0:9292，供本地或同网段主机通过 HTTP 查询 IP 归属信息
+# (asn / city / country 三个接口，详见 service/geodb/GeoAPI.md)。
+#
+# 用法:
+#   geo-api                              # 默认 ./geodb/ + 0.0.0.0:9292
+#   geo-api -d /data/ipdb                # 指定数据文件目录
+#   geo-api -p 8080                      # 指定端口
+#   geo-api -b 127.0.0.1 -p 9292 -d ./ipdb
+#   geo-api --server puma                # 指定 Rack 服务器
+#   GEODB_DATA_DIR=/data/ipdb geo-api    # 通过环境变量指定数据目录
+#
+# 选项:
+#   -d, --data-dir DIR   IP 数据文件目录，默认 ./geodb/
+#   -b, --host HOST      监听地址，默认 0.0.0.0
+#   -p, --port PORT      监听端口，默认 9292
+#   -s, --server NAME    Rack 服务器，默认 puma
+#   -h, --help           显示帮助
+#   -v, --version        显示版本
+#
+# 数据目录说明:
+#   目录下应至少存在 geoapi 期望的 JSON 文件之一 (asn.json / city-IPv4.json /
+#   country-IPv4.json / country-IPv6.json / city-IPv6.json / geo-city.json /
+#   geo-country.json)。目录不存在或没有任何 *.json 文件将直接报错退出。
+#   查询时若具体文件缺失，对应接口容错返回 404 无结果，不影响其他接口。
+
+require "optparse"
+require "rackup"
+
+# 找到 gem 自身根目录 (本文件在 <root>/bin/geo-api)
+ROOT = File.expand_path("..", __dir__)
+$LOAD_PATH.unshift File.join(ROOT, "service", "geodb")
+
+# 命令行默认值
+options = {
+  data_dir: nil,
+  host: "0.0.0.0",
+  port: 9292,
+  server: "puma"
+}
+
+parser = OptionParser.new do |opts|
+  opts.banner = <<~BAN
+    geo-api — IP 归属信息查询服务 (network-infra-utility)
+
+    用法:
+      geo-api [options]
+      GEODB_DATA_DIR=/data/ipdb geo-api
+
+    选项:
+  BAN
+  opts.on("-d", "--data-dir DIR", "IP 数据文件目录，默认 ./geodb/") { |v| options[:data_dir] = v }
+  opts.on("-b", "--host HOST", "监听地址，默认 0.0.0.0") { |v| options[:host] = v }
+  opts.on("-p", "--port PORT", Integer, "监听端口，默认 9292") { |v| options[:port] = v }
+  opts.on("-s", "--server NAME", "Rack 服务器，默认 puma") { |v| options[:server] = v }
+  opts.on("-v", "--version", "显示版本") do
+    require_relative "../version"
+    puts "geo-api #{NetworkInfraUtility::VERSION}"
+    exit 0
+  end
+  opts.on("-h", "--help", "显示帮助") do
+    puts opts
+    exit 0
+  end
+end
+parser.parse!
+
+# 数据目录: 命令行 > 环境变量 > 默认 ./geodb
+data_dir = options[:data_dir] || ENV["GEODB_DATA_DIR"] || "geodb"
+data_dir = File.expand_path(data_dir)
+ENV["GEODB_DATA_DIR"] = data_dir
+
+# ---- 启动前校验数据目录 ---------------------------------------------------
+# 目录不存在直接报错退出；目录存在但没有任何 *.json 也视为错误。
+unless File.directory?(data_dir)
+  warn "✗ 数据目录不存在: #{data_dir}"
+  warn "  请用 -d/--data-dir 指定正确的数据文件目录，或在该位置创建 geodb/ 目录。"
+  warn "  期望目录结构: #{data_dir}/ 下含 asn.json / city-IPv4.json /"
+  warn "                country-IPv4.json / geo-city.json 等 JSON 数据文件。"
+  exit 1
+end
+
+json_files = Dir.glob(File.join(data_dir, "*.json"))
+if json_files.empty?
+  warn "✗ 数据目录 #{data_dir} 下没有任何 JSON 数据文件。"
+  warn "  请先将 GeoLite2 CSV 数据通过 service/geodb/geodb.rb 生成为 JSON，"
+  warn "  或用 -d/--data-dir 指向已生成 JSON 的目录。"
+  exit 1
+end
+
+require_relative "../version"
+puts "╭─ geo-api #{NetworkInfraUtility::VERSION} ───────────────────────────────"
+puts "│ 数据目录: #{data_dir} "
+puts "│ #{json_files.length}个数据文件：#{json_files.map { |f| File.basename(f) }.join(' ')}"
+puts "│ 监听: http://#{options[:host]}:#{options[:port]}"
+puts "│ 服务: #{options[:server]}"
+puts "╰─ Ctrl+C 退出 ─────────────────────────────────────"
+$stdout.flush
+
+# ---- 启动服务 -------------------------------------------------------------
+require "api"   # 即 service/geodb/api.rb
+
+Rackup::Server.start(
+  app:    GeoAPI.app,
+  server: options[:server],
+  Host:   options[:host],
+  Port:   options[:port]
+)

data/bin/geo-get

@@ -0,0 +1,135 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+# coding: utf-8
+#
+# geo-get — network-infra-utility 的 IP 归属信息命令行查询工具
+#
+# 向运行中的 geo-api 服务发起查询，一次输入 IP，同时拉取 country / city /
+# asn 三类归属信息并汇总输出。默认展示文字，加 -j 则打印 JSON。
+#
+# 用法:
+#   geo-get 1.181.240.251              # 默认三接口齐查, 文字格式
+#   geo-get 8.8.8.8 -j                 # JSON 格式输出
+#   geo-get 8.8.8.8 --asn              # 仅查 asn, 文字
+#   geo-get 1.1.1.1 --country --json   # 仅查 country, JSON
+#
+# 选项:
+#   -t, --text         文字格式输出 (默认)
+#   -j, --json         JSON 格式输出
+#   --country          仅查询国家接口
+#   --city             仅查询城市接口
+#   --asn              仅查询 ASN 接口
+#   (不指定 --country/--city/--asn 时三接口齐查)
+#   -h, --help         显示帮助
+#   -v, --version      显示版本
+#
+# 默认连 http://127.0.0.1:9292 (geo-api 默认监听)。
+# 服务未运行时给出明确提示与启动命令。
+
+require "optparse"
+require "net/http"
+require "json"
+require "uri"
+
+HOST = "127.0.0.1"
+PORT = 9292
+BASE = "http://#{HOST}:#{PORT}"
+
+# ---- 命令行解析 -------------------------------------------------------------
+options = { format: :text, only: nil }
+parser = OptionParser.new do |opts|
+  opts.banner = <<~BAN
+    geo-get — IP 归属信息查询工具 (network-infra-utility)
+
+    用法:
+      geo-get <IP> [options]
+      geo-get                    # 查默认 IP 1.181.240.251, 三接口齐查, 文字
+      geo-get 8.8.8.8 -j         # JSON 格式
+      geo-get 8.8.8.8 --asn      # 仅查 asn
+
+    选项:
+  BAN
+  opts.on("-t", "--text", "文字格式输出 (默认)")      { options[:format] = :text }
+  opts.on("-j", "--json", "JSON 格式输出")            { options[:format] = :json }
+  opts.on("--country", "仅查询国家接口")              { options[:only] = :country }
+  opts.on("--city", "仅查询城市接口")                 { options[:only] = :city }
+  opts.on("--asn", "仅查询 ASN 接口")                 { options[:only] = :asn }
+  opts.on("-v", "--version", "显示版本") do
+    require_relative "../version"
+    puts "geo-get #{NetworkInfraUtility::VERSION}"
+    exit 0
+  end
+  opts.on("-h", "--help", "显示帮助") do
+    puts opts
+    exit 0
+  end
+end
+parser.parse!
+
+# 第一个非 -- 参数当 IP; 不给就用默认演示 IP
+ip = ARGV.find { |a| !a.start_with?("-") } || "1.181.240.251"
+
+only = options[:only]
+do_country = only.nil? || only == :country
+do_city    = only.nil? || only == :city
+do_asn     = only.nil? || only == :asn
+
+# ---- HTTP 取数 --------------------------------------------------------------
+# 发起一次 GET, 返回 {code:, body:}
+def fetch(base, path, params)
+  uri = URI("#{base}#{path}")
+  uri.query = URI.encode_www_form(params) unless params.empty?
+  res = Net::HTTP.get_response(uri)
+  { code: res.code.to_i, body: (JSON.parse(res.body) rescue res.body) }
+rescue Errno::ECONNREFUSED
+  abort "✗ 无法连接 #{base} —— geo-api 服务未启动。\n  启动命令: geo-api -d E:/workspace/momentum/twinklite/data/geodb"
+rescue => e
+  abort "✗ 请求失败: #{e.class}: #{e.message}"
+end
+
+# ---- 按接口拉取 + 收集 ------------------------------------------------------
+results = {}
+results[:country] = fetch(BASE, "/geo/country", addr: ip) if do_country
+results[:city]    = fetch(BASE, "/geo/city",    addr: ip) if do_city
+results[:asn]     = fetch(BASE, "/geo/asn",     addr: ip) if do_asn
+
+# ---- 输出: --json / --text 两路分发 -----------------------------------------
+case options[:format]
+when :json
+  # JSON 格式: 把每个接口的响应原样放进一个对象打印。
+  # 命中的接口放解析后的 body; 404/无结果放 null, 不逐条刷错误。
+  payload = {}
+  results.each do |key, r|
+    payload[key] = (r[:code] == 200 ? r[:body] : nil)
+  end
+  puts JSON.pretty_generate(payload)
+when :text
+  puts "查询 IP: #{ip}    范围: #{only ? "仅 #{only}" : 'country / city / asn 三接口齐查'}    服务: #{BASE}"
+  puts "-" * 60
+
+  hits = []
+  if do_country && (r = results[:country]) && r[:code] == 200
+    d = r[:body]
+    g = d["geoname"]; rc = d["registered_country"]
+    country = (g && g["country_name"]) || (rc && rc["country_name"]) || "未知"
+    hits << "[国家] #{country} (#{(g && g["country_iso_code"]) || (rc && rc["country_iso_code"])})  网段 #{d["network"]}"
+  end
+  if do_city && (r = results[:city]) && r[:code] == 200
+    d = r[:body]
+    g = d["geoname"]
+    city = (g && (g["city_namezh"] || g["city_name"])) || "未知"
+    coord = g && g["latitude"] && g["longitude"] ? "#{g["latitude"]}, #{g["longitude"]}" : "无"
+    hits << "[城市] #{city}    坐标 #{coord}    网段 #{d["network"]}"
+  end
+  if do_asn && (r = results[:asn]) && r[:code] == 200
+    d = r[:body]
+    hits << "[ASN]  AS#{d["autonomous_system_number"]} #{d["autonomous_system_organization"]}    网段 #{d["network"]}"
+  end
+
+  if hits.empty?
+    puts "#{ip} 未查到归属信息 (该 IP 不在已知数据范围内, 如回环/保留地址)"
+  else
+    puts "#{ip} 的归属信息:"
+    hits.each { |h| puts "  #{h}" }
+  end
+end

data/bin/geo-load

@@ -0,0 +1,190 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+# coding: utf-8
+#
+# geo-load — network-infra-utility 的 GeoLite2 CSV → JSON 转换命令行工具
+#
+# 把 GeoLite2 解压后的 CSV 目录转换为 geo-api 可用的 JSON 数据文件。
+# 一次调用依次执行 load_asn / load_city / load_country，生成 7 个 JSON:
+#   asn.json / city-IPv4.json / city-IPv6.json
+#   country-IPv4.json / country-IPv6.json / geo-city.json / geo-country.json
+#
+# 用法:
+#   geo-load                                            # 当前目录找 CSV, 输出到 ./geodb/
+#   geo-load /data/GeoLite2-CSV                         # 指定 CSV 目录
+#   geo-load /data/GeoLite2-CSV /var/lib/geodb          # 指定输入+输出
+#
+# 参数规则:
+#   geo-load [RAW_DIR] [DOC_DIR]
+#
+#   RAW_DIR (第一个参数, 可省):
+#     不写 → 在当前工作目录下查找 GeoLite2 CSV 目录, 找不到报错
+#     写了 → 用指定路径; 路径不存在报错
+#
+#   DOC_DIR (第二个参数, 可省):
+#     不写 → 默认在当前工作目录下生成 geodb/ 子目录
+#     写了 → 用指定路径; 路径不存在则自动创建, 创建不了报错
+#
+# 选项:
+#   --asn-only          仅转换 ASN 数据
+#   --city-only         仅转换 City 数据
+#   --country-only      仅转换 Country 数据
+#   (不指定时三类全转)
+#   -h, --help          显示帮助
+#   -v, --version       显示版本
+
+require "optparse"
+
+# ---- 命令行解析 -------------------------------------------------------------
+options = { only: nil }
+parser = OptionParser.new do |opts|
+  opts.banner = <<~BAN
+    geo-load — GeoLite2 CSV → JSON 转换工具 (network-infra-utility)
+
+    用法:
+      geo-load [RAW_DIR] [DOC_DIR]
+      geo-load                            # 当前目录找 CSV, 输出到 ./geodb/
+      geo-load /data/GeoLite2-CSV         # 指定 CSV 目录
+      geo-load /data/GeoLite2-CSV /opt/geodb
+
+    参数:
+      RAW_DIR  GeoLite2 CSV 目录 (可省, 省则在当前目录查找)
+      DOC_DIR  JSON 输出目录 (可省, 省则输出到 ./geodb/)
+
+    选项:
+  BAN
+  opts.on("--asn-only", "仅转换 ASN 数据")     { options[:only] = :asn }
+  opts.on("--city-only", "仅转换 City 数据")    { options[:only] = :city }
+  opts.on("--country-only", "仅转换 Country 数据") { options[:only] = :country }
+  opts.on("-v", "--version", "显示版本") do
+    require_relative "../version"
+    puts "geo-load #{NetworkInfraUtility::VERSION}"
+    exit 0
+  end
+  opts.on("-h", "--help", "显示帮助") do
+    puts opts
+    exit 0
+  end
+end
+parser.parse!
+
+# 取位置参数: 第一个非 -- 参数是 RAW_DIR, 第二个是 DOC_DIR
+positional = ARGV.reject { |a| a.start_with?("-") }
+raw_arg = positional[0]
+doc_arg = positional[1]
+
+# ---- 确定 RAW_DIR (输入目录) ------------------------------------------------
+# GeoLite2 解压后的 CSV 目录特征: 含 GeoLite2-ASN-CSV_* / GeoLite2-City-CSV_* /
+# GeoLite2-Country-CSV_* 子目录, 或本身就是这些子目录之一。
+GEO_PATTERN = /GeoLite2-(ASN|City|Country)-CSV/i
+
+if raw_arg
+  # 用户显式指定了 RAW_DIR
+  raw_dir = File.expand_path(raw_arg)
+  unless File.directory?(raw_dir)
+    abort "✗ 输入目录不存在: #{raw_dir}\n  请指定有效的 GeoLite2 CSV 解压目录。"
+  end
+else
+  # 没指定 → 在当前工作目录下查找
+  cwd = Dir.pwd
+  # 先看当前目录本身是否符合
+  candidates = []
+  if cwd =~ GEO_PATTERN || Dir.glob(File.join(cwd, "GeoLite2-*-CSV_*")).any?
+    candidates << cwd
+  end
+  # 再看子目录
+  Dir.glob(File.join(cwd, "*")).each do |p|
+    candidates << p if File.directory?(p) && (p =~ GEO_PATTERN || Dir.glob(File.join(p, "GeoLite2-*-CSV_*")).any?)
+  end
+  # 也看下一层 (常见: cwd/GeoLite2-CSV_xxx/GeoLite2-ASN-CSV_xxx/)
+  Dir.glob(File.join(cwd, "*", "*")).each do |p|
+    candidates << p if File.directory?(p) && p =~ GEO_PATTERN
+  end
+
+  if candidates.empty?
+    abort <<~MSG
+      ✗ 在当前目录下未找到 GeoLite2 CSV 数据:
+        #{cwd}
+      请将 GeoLite2 CSV 解压到当前目录, 或显式指定路径:
+        geo-load /path/to/GeoLite2-CSV [DOC_DIR]
+      期望的目录结构:
+        GeoLite2-ASN-CSV_*/  GeoLite2-City-CSV_*/  GeoLite2-Country-CSV_*/
+    MSG
+  end
+
+  # 去重, 取包含最多 CSV 类型的目录 (优先选父目录)
+  raw_dir = candidates.uniq.max_by { |d| Dir.glob(File.join(d, "**", "*.csv")).length }
+end
+
+# ---- 确定 DOC_DIR (输出目录) ------------------------------------------------
+if doc_arg
+  doc_dir = File.expand_path(doc_arg)
+  # 指定了路径: 不存在则创建, 创建不了报错
+  unless File.directory?(doc_dir)
+    begin
+      require "fileutils"
+      FileUtils.mkdir_p(doc_dir)
+    rescue SystemCallError => e
+      abort "✗ 无法创建输出目录 #{doc_dir}: #{e.message}"
+    end
+  end
+else
+  # 没指定 → 默认在当前工作目录下生成 geodb/
+  doc_dir = File.expand_path("geodb", Dir.pwd)
+  begin
+    require "fileutils"
+    FileUtils.mkdir_p(doc_dir)
+  rescue SystemCallError => e
+    abort "✗ 无法创建输出目录 #{doc_dir}: #{e.message}"
+  end
+end
+
+# ---- 加载 GeoDB 模块 --------------------------------------------------------
+# geodb.rb 依赖 cc + network + CC.use, 需要把 service/geodb 加入加载路径
+ROOT = File.expand_path("..", __dir__)
+$LOAD_PATH.unshift File.join(ROOT, "service", "geodb")
+
+require "cc"
+require "network"
+CC.use "file", "enum", "shell-tools"
+require "geodb"
+
+# ---- 执行转换 ---------------------------------------------------------------
+do_asn     = options[:only].nil? || options[:only] == :asn
+do_city    = options[:only].nil? || options[:only] == :city
+do_country = options[:only].nil? || options[:only] == :country
+
+puts "╭─ geo-load #{NetworkInfraUtility::VERSION} ───────────────────────────────"
+puts "│ 输入目录: #{raw_dir}"
+puts "│ 输出目录: #{doc_dir}"
+puts "│ 转换范围: #{options[:only] ? "仅 #{options[:only]}" : 'asn / city / country 全量'}"
+puts "╰─ 开始转换 ─────────────────────────────────────"
+$stdout.flush
+
+# GeoDB 的三个方法用 Dir[glob] 匹配 CSV, 路径模式相对于调用目录
+# 把 raw_dir + glob 拼成绝对路径模式
+asn_pattern    = File.join(raw_dir, "**", "GeoLite2-ASN-CSV_*", "*.csv")
+city_pattern   = File.join(raw_dir, "**", "GeoLite2-City-CSV_*", "*.csv")
+country_pattern = File.join(raw_dir, "**", "GeoLite2-Country-CSV_*", "*.csv")
+
+# out_path 需要以 / 结尾 (geodb.rb 里拼 "#{out_path}asn.json")
+out_path = doc_dir.end_with?("/", "\\") ? doc_dir : doc_dir + "/"
+
+if do_asn
+  puts "--- ASN ---"
+  GeoDB.load_asn(asn_pattern, out_path)
+end
+if do_city
+  puts "--- City ---"
+  GeoDB.load_city(city_pattern, out_path)
+end
+if do_country
+  puts "--- Country ---"
+  GeoDB.load_country(country_pattern, out_path)
+end
+
+puts "╭─ 完成 ───────────────────────────────────────"
+puts "│ JSON 文件已生成到: #{doc_dir}"
+puts "│ 文件列表:"
+Dir.glob(File.join(doc_dir, "*.json")).each { |f| puts "│   #{File.basename(f)}" }
+puts "╰─ 可用 geo-api -d #{doc_dir} 启动查询服务 ────"

data/network-infra-utility.gemspec

@@ -21,7 +21,14 @@ Gem::Specification.new do |spec|
   spec.files = Dir.chdir(File.expand_path(__dir__)) do
     `git ls-files -z`.split("\x0").reject { |f| f.match(%r{\A(?:test|spec|features|example)/}) }
   end
-  spec.bindir       = "exe"
-  spec.executables  = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
+  # bin/ 既放对外命令 (geo-api) 也放开发脚本 (console/setup)，
+  # executables 显式声明，避免把开发脚本当作系统命令安装到用户 PATH。
+  spec.bindir       = "bin"
+  spec.executables  = %w[geo-api geo-get geo-load]
   spec.require_paths = ["document", "service", "support", "tool", "."]
+
+  # geo-api 命令行服务依赖的运行时 gem
+  spec.add_runtime_dependency "roda", "~> 3.0"
+  spec.add_runtime_dependency "rackup", "~> 2.0"
+  spec.add_runtime_dependency "puma", "~> 6.0"
 end

data/network.rb

@@ -12,7 +12,7 @@ require_relative "version"
 # 加载顺序按依赖方向上游在前，避免循环依赖。
 module NetworkInfraUtility
   # 支撑层：无外部依赖的基础工具，最先加载
-  # require_relative "support/xxx"
+  require_relative "support/basic/ip"
 
   # 工具层：依赖 support
   # require_relative "tool/xxx"