From db3372eaaa165e3c6c43b5e6a00214692443afbf Mon Sep 17 00:00:00 2001
From: hahwul <hahwul@gmail.com>
Date: Thu, 18 Aug 2022 23:55:54 +0900
Subject: [PATCH] Update

---
 scripts/fetch_lang.rb | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 scripts/fetch_lang.rb

diff --git a/scripts/fetch_lang.rb b/scripts/fetch_lang.rb
new file mode 100644
index 0000000..6eb359e
--- /dev/null
+++ b/scripts/fetch_lang.rb
@@ -0,0 +1,30 @@
+# https://api.github.com/repos/hahwul/dalfox/languages
+#
+#
+require 'yaml'
+
+langs = []
+Dir.entries("./weapons").each do | name |
+    if name.strip != "."  || name != ".."
+      begin
+        data = YAML.load(File.open("./weapons/#{name}"))
+        if data['url'].include? "//github.com"
+            t = data['url'].split("/")
+            lang = `curl -s https://api.github.com/repos/#{t[3]}/#{t[4]}/languages | jq 'to_entries | max_by(.value) | .key'`
+            lang_str = lang.gsub("\"","").gsub("\n","")
+            if lang_str != "documentation_url"
+              puts 'hit'
+              data['lang'] = lang_str
+              yaml_data = YAML.dump(data)
+              File.write("./weapons/#{filename}", yaml_data)
+              langs.push lang_str
+            else
+              puts 'denied'
+            end
+            sleep(1.5.minutes)
+        end
+      rescue
+      end
+    end
+  end
+puts langs.uniq
\ No newline at end of file