Fix import from wcag2.1de

lib/tasks/import.rake

@@ -1,56 +1,65 @@
 # frozen_string_literal: true
+URL = "https://outline-rocks.github.io/wcag/translations/WCAG21-de/"
+
+def import_wcag21de
+  doc = Nokogiri::HTML5(URI.open(URL))
+
+  standard = Standard.find_by(name_de: "WCAG 2.1")
+
+  doc.css("section.principle").each do |principle_node|
+    _principle_id = principle_node.attributes["id"].value
+    principle_title = principle_node.css("h2").first.content.scan(/[\d\.]+ ([a-zA-ZöäüÖÄÜ]+)/)
+    puts principle_title
+    principle = Principle.find_by!(name_de: principle_title)
+
+    principle_node.css("section").each do |guideline_node|
+      next unless guideline_node.css("h3").first
+
+      guideline_title = guideline_node.css("h3").first.content
+      guideline_text = guideline_node.css("p").first
+
+      guideline_number = guideline_title.scan(/Richtlinie \d+\.(\d+).*/).first.last
+      # guideline = principle.guidelines.find_or_create_by(number: guideline_number)
+      # guideline.title = guideline_title.scan(/Richtlinie \d+\.\d+(.*)/).first.last
+      # guideline.text = guideline_text
+
+      # guideline.save!
+
+      guideline_node.css("section").each do |sc|
+        sc_title = sc.css("h4").first.content.scan(/Erfolgskriterium \d+\.\d+\.\d+(.*)/).first.last
+        sc_number = sc.css("h4").first.content.scan(/Erfolgskriterium (\d+\.\d+\.\d+).*/).first.last
+        sc_level = sc.css("p").first.content.scan(/\(Stufe (A+)\)/).first.last
+        sc_url = sc.css("h4 a.self-link").first.attr("href")
+        sc_conformity_notice = sc.css("div.note p").first&.content
+        text = "<div>#{guideline_text}<br>#{sc.css('p')[1]}"
+        text += "<p>#{sc.css('dl').first&.to_s}</p>" if sc.css("dl").first
+
+
+        # _sc_number = sc_title.scan(/Erfolgskriterium \d+\.\d+\.(\d+).*/).first.last
+
+        check = Check.find_or_initialize_by(external_number: sc_number)
+        check.name_de = sc_title
+        check.principle = principle
+        check.standards << standard unless check.standards.include?(standard)
+        check.applicable_to_app = check.applicable_to_web = true
+        check.external_number = sc_number
+        check.external_url = "#{URL}#{sc_url}"
+        check.conformity_level = sc_level.to_sym
+        check.conformity_notice_de = sc_conformity_notice
+        check.criterion_de = "#{text}</div>".gsub('href="#', %(href="#{URL}#))
+        check.save!
+      end
+    end
+  end
+  @output
+end
 
 namespace :import do
   desc "Import WCAG2.1 in german"
   task wcag21de: :environment do
-    URL = "https://outline-rocks.github.io/wcag/translations/WCAG21-de/"
-
-    doc = Nokogiri::HTML5(URI.open(URL))
-
-    # standard = Standard.find_or_create_by(url: URL)
-    # standard.name = 'WCAG 2.1'
-    # standard.description = doc.css('#abstract p').map(&:to_s).join
-    # standard.save!
-
-    doc.css("section.principle").each do |principle_node|
-      _principle_id = principle_node.attributes["id"].value
-      _principle_title = principle_node.css("h2").first.content
-      # principle = standard.principles.find_or_create_by(code: principle_id)
-      # principle.name = principle_title.scan(/\d+\. (.*)/).first.last
-      # principle.number = principle_title.scan(/(\d+)\..*/).first.last
-      # principle.save!
-
-      principle_node.css("section").each do |guideline_node|
-        next unless guideline_node.css("h3").first
-
-        guideline_title = guideline_node.css("h3").first.content
-        guideline_text = guideline_node.css("p").first
-
-        _guideline_number = guideline_title.scan(/Richtlinie \d+\.(\d+).*/).first.last
-        # guideline = principle.guidelines.find_or_create_by(number: guideline_number)
-        # guideline.title = guideline_title.scan(/Richtlinie \d+\.\d+(.*)/).first.last
-        # guideline.text = guideline_text
-
-        # guideline.save!
-
-        guideline_node.css("section").each do |sc|
-          sc_title = sc.css("h4").first.content.scan(/Erfolgskriterium \d+\.\d+\.\d+(.*)/).first.last
-          sc_level = sc.css("p").first.content.scan(/\(Stufe (A+)\)/).first.last
-          # sc_hints = sc.css('div p').map { ReverseMarkdown.convert(_1.to_s) }
-
-          # _sc_number = sc_title.scan(/Erfolgskriterium \d+\.\d+\.(\d+).*/).first.last
-
-          success_criterion = Check.find_or_create_by(name: sc_title)
-          success_criterion.name = sc_title
-          text = "<div>#{guideline_text}<br>#{sc.css('p')[1]}"
-          text += "<p>#{sc.css('dl').first&.to_s}</p>" if sc.css("dl").first
-
-          success_criterion.success_criterion_html = "#{text}</div>".gsub('href="#', %(href="#{URL}#))
-          success_criterion.level = sc_level
-          success_criterion.save!
-        end
-      end
+    Check.transaction do
+      import_wcag21de
+      # raise ActiveRecord::Rollback
     end
-    @output
   end
 end