atco 1.0.5 → 1.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7415f66d4eafb4428737b275baca8471a43e0f03863ca2705cbebe2124312129
-  data.tar.gz: 27043e632b025a4084d6666c145787bb6acf53b878675ce525b2946bbd5c8d30
+  metadata.gz: d9b77a9ff8bc2044b0985cdc1cf6b9f9f997e6884c72ce88e713cc807c04aa40
+  data.tar.gz: 8c09fd9b592ce9c1c8eae5b2bf4b36e9b1f1e8a98aff075773285f4a2da28e92
 SHA512:
-  metadata.gz: 210d0f52a29cc65760d3445b829430c9018ce18d429658bc5311caa8e1279af96f9fc03b0a369f8c5ea251373e2fdeea9152fc6ee964631a4ef1085bdcff1b7e
-  data.tar.gz: 8dad485652657476d6d00a5c1203d599cc8560091737712aef3a0728da39ffa585df94a580c7b80f5b6aaf5fb5cb5e92e157928ee5d9e415a6f33ccfacad4ab1
+  metadata.gz: 7962c7bd938e9ec5d406b72994148b01368651b49ae4b023088dda40e8317895999805f1d6d8e5cc58d23caf5eb7d05da71fc64251cab7e666c30ba38302929b
+  data.tar.gz: 4b1469dce23b861431b42cbdc1aeaf0f731388c648fb5bd4951b72e1fe889ce567cf0ff451c633737c6879a6cf950752409f79675e83234dee47c60dc03494bf

data/lib/atco/header.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module Atco
+  # Atco::Header is a class to abstract ATCO-CIF Header data.
+  class Header
+    attr_accessor :file_type,
+                  :version,
+                  :file_originator,
+                  :source_product,
+                  :production_datetime
+
+    # Public: Parse a header line from an ATCO-CIF file.
+    # EXAMPLE:
+    # "ATCO-CIF0500Electronic Registration         MIA 4.20.18     20090915113809\r\n"
+    def self.parse(line)
+      data = {
+        file_type: line[0, 8],
+        version: "#{line[8, 2].to_i}.#{line[10, 2].to_i}",
+        file_originator: line[12, 32].strip!,
+        source_product: line[44, 16].strip!,
+        production_datetime: line[60, 14]
+      }
+      new(data)
+    end
+
+    def initialize(data)
+      @file_type = data[:file_type]
+      @version = data[:version]
+      @file_originator = data[:file_originator]
+      @source_product = data[:source_product]
+      @production_datetime = data[:production_datetime]
+    end
+
+    def attributes
+      {
+        file_type: @file_type,
+        version: @version,
+        file_originator: @file_originator,
+        source_product: @source_product,
+        production_datetime: @production_datetime
+      }
+    end
+
+    def to_json(*attrs)
+      attributes.to_json(*attrs)
+    end
+  end
+end

data/lib/atco/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Atco
-  VERSION = "1.0.5"
+  VERSION = "1.0.8"
 end

data/lib/atco.rb

@@ -2,6 +2,7 @@
 
 require "open3"
 require "tempfile"
+require_relative "atco/header"
 require_relative "atco/location"
 require_relative "atco/journey"
 require_relative "atco/stop"
@@ -38,8 +39,8 @@ module Atco # rubocop:disable Metrics/ModuleLength
       unparsed = []
 
       data.each_with_index do |line, line_number| # rubocop:disable Metrics/BlockLength
-        if line == data.first
-          header = parse_header(line)
+        if line_number.zero?
+          header = Header.parse(line)
           next
         end
 
@@ -52,13 +53,13 @@ module Atco # rubocop:disable Metrics/ModuleLength
           object = send("parse_#{method}", line)
           next unless object[:record_identity] && object[:record_identity] == identifier
 
-          current_journey = object if object[:record_identity] && object[:record_identity] == METHODS[:journey_header]
-          if object[:record_identity] && (object[:record_identity] == METHODS[:location] || object[:record_identity] == METHODS[:additional_location_info]) # rubocop:disable Layout/LineLength
-            if object[:record_identity] == METHODS[:location]
-              current_location = object
-            else
-              locations << Location.new(current_location, object)
-            end
+          case method
+          when :journey_header
+            current_journey = object
+          when :location
+            current_location = object
+          when :additional_location_info
+            locations << Location.new(current_location, object)
           end
 
           if current_journey
@@ -73,20 +74,11 @@ module Atco # rubocop:disable Metrics/ModuleLength
           unparsed << { line: line, line_number: line_number }
           next
         end
+        objects << object
       end
       { header: header, locations: locations, journeys: journeys, unparsed: unparsed }
     end
 
-    def parse_header(string)
-      {
-        file_type: string[0, 8],
-        version: "#{string[8, 2].to_i}.#{string[10, 2].to_i}",
-        file_originator: string[12, 32].strip!,
-        source_product: string[44, 16].strip!,
-        production_datetime: string[60, 14]
-      }
-    end
-
     def parse_bank_holiday(string)
       {
         record_identity: string[0, 2],

data/spec/atco_spec.rb

@@ -7,44 +7,6 @@ RSpec.describe Atco do # rubocop:disable Metrics/BlockLength
     expect(Atco::VERSION).not_to be nil
   end
 
-  it "should parse header from fixture" do
-    result = Atco.parse("spec/fixtures/example.cif")
-    expect(result[:header]).to eq(
-      {
-        file_type: "ATCO-CIF",
-        file_originator: "Electronic Registration",
-        source_product: "MIA 4.20.18",
-        version: "5.0",
-        production_datetime: "20090915113809"
-      }
-    )
-  end
-
-  it "should parse locations from fixture" do
-    result = Atco.parse("spec/fixtures/example.cif")
-    expect(result[:header]).to eq(
-      {
-        file_type: "ATCO-CIF",
-        file_originator: "Electronic Registration",
-        source_product: "MIA 4.20.18",
-        version: "5.0",
-        production_datetime: "20090915113809"
-      }
-    )
-  end
-
-  it "should parse header" do
-    expect(Atco.parse_header("ATCO-CIF0500Electronic Registration         MIA 4.20.18     20090915113809\r\n")).to eq(
-      {
-        file_type: "ATCO-CIF",
-        version: "5.0",
-        file_originator: "Electronic Registration",
-        source_product: "MIA 4.20.18",
-        production_datetime: "20090915113809"
-      }
-    )
-  end
-
   it "should parse bank holiday" do
     expect(Atco.parse_bank_holiday("QHN20061225")).to eq(
       {
@@ -159,62 +121,4 @@ RSpec.describe Atco do # rubocop:disable Metrics/BlockLength
       }
     )
   end
-
-  describe "with example.cif" do # rubocop:disable Metrics/BlockLength
-    before(:all) do
-      @atco = Atco.parse("spec/fixtures/example.cif")
-    end
-
-    it "should parse 1 journey" do
-      expect(@atco[:journeys].size).to eq(1)
-    end
-
-    it "should parse journeys into Atco::Joruney objects" do
-      expect(@atco[:journeys]["139748"]).to be_a_kind_of(Atco::Journey)
-    end
-
-    it "should parse 6 stops for journey 139748" do
-      expect(@atco[:journeys]["139748"].stops.size).to eq(6)
-    end
-
-    it "should parse 2 locations" do
-      expect(@atco[:locations].size).to eq(2)
-    end
-
-    it "should output file as JSON" do
-      output = File.join(File.dirname(__FILE__), "artefacts", "test.json")
-      File.open(output, "w+") do |f|
-        f.flush
-        f.write(JSON.pretty_generate(@atco))
-      end
-
-      expect(File.exist?(output)).to be true
-
-      data = File.read(output)
-      json = JSON.parse(data)
-      expect(json).to be_a(Hash)
-    end
-
-    it "should return 17 unparsed lines" do
-      expect(@atco[:unparsed].size).to eq(17)
-    end
-
-    it "should not parse GS records" do
-      expect(@atco[:unparsed][0]).to eq(
-        {
-          line: "GS00001433 N                    Belfast Metro Ops                                 7000\n",
-          line_number: 3
-        }
-      )
-    end
-
-    it "should not parse GR records" do
-      expect(@atco[:unparsed][1]).to eq(
-        {
-          line: "GR00001433Donegall Square East                                                                                7000\n", # rubocop:disable Layout/LineLength
-          line_number: 4
-        }
-      )
-    end
-  end
 end

data/spec/integration/translink_spec.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+RSpec.describe "with example.cif" do # rubocop:disable Metrics/BlockLength
+  before(:all) do
+    @atco = Atco.parse("spec/fixtures/translink-example.cif")
+  end
+
+  it "should parse header from fixture" do
+    expect(@atco[:header].attributes).to eq(
+      {
+        file_type: "ATCO-CIF",
+        file_originator: "Electronic Registration",
+        source_product: "MIA 4.20.18",
+        version: "5.0",
+        production_datetime: "20090915113809"
+      }
+    )
+  end
+
+  it "should parse 1 journey" do
+    expect(@atco[:journeys].size).to eq(1)
+  end
+
+  it "should parse journeys into Atco::Joruney objects" do
+    expect(@atco[:journeys]["139748"]).to be_a_kind_of(Atco::Journey)
+  end
+
+  it "should parse 6 stops for journey 139748" do
+    expect(@atco[:journeys]["139748"].stops.size).to eq(6)
+  end
+
+  it "should parse 2 locations" do
+    expect(@atco[:locations].size).to eq(2)
+  end
+
+  it "should output file as JSON" do
+    output = File.join(File.dirname(__FILE__), "..", "artefacts", "test.json")
+    File.open(output, "w+") do |f|
+      f.flush
+      f.write(JSON.pretty_generate(@atco))
+    end
+
+    expect(File.exist?(output)).to be true
+
+    data = File.read(output)
+    json = JSON.parse(data)
+    expect(json).to be_a(Hash)
+  end
+
+  it "should return 17 unparsed lines" do
+    expect(@atco[:unparsed].size).to eq(17)
+  end
+
+  it "should not parse GS records" do
+    expect(@atco[:unparsed][0]).to eq(
+      {
+        line: "GS00001433 N                    Belfast Metro Ops                                 7000\n",
+        line_number: 3
+      }
+    )
+  end
+
+  it "should not parse GR records" do
+    expect(@atco[:unparsed][1]).to eq(
+      {
+        line: "GR00001433Donegall Square East                                                                                7000\n", # rubocop:disable Layout/LineLength
+        line_number: 4
+      }
+    )
+  end
+end

data/spec/unit/header_spec.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+RSpec.describe Atco::Header do
+  before(:all) do
+    line = "ATCO-CIF0500Electronic Registration         MIA 4.20.18     20090915113809\r\n"
+    @header = Atco::Header.parse(line)
+  end
+
+  it "should be a Header object" do
+    expect(@header).to be_a_kind_of(Atco::Header)
+  end
+
+  it "should parse header to attributes" do
+    expect(@header.attributes).to eq(
+      {
+        file_type: "ATCO-CIF",
+        version: "5.0",
+        file_originator: "Electronic Registration",
+        source_product: "MIA 4.20.18",
+        production_datetime: "20090915113809"
+      }
+    )
+  end
+
+  it "should parse header to json" do
+    expect(@header.to_json).to eq(
+      "{\"file_type\":\"ATCO-CIF\",\"version\":\"5.0\",\"file_originator\":\"Electronic Registration\",\"source_product\":\"MIA 4.20.18\",\"production_datetime\":\"20090915113809\"}" # rubocop:disable Metrics/LineLength
+    )
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: atco
 version: !ruby/object:Gem::Version
-  version: 1.0.5
+  version: 1.0.8
 platform: ruby
 authors:
 - David Rice
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-04-12 00:00:00.000000000 Z
+date: 2024-04-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -49,14 +49,17 @@ files:
 - README.md
 - Rakefile
 - lib/atco.rb
+- lib/atco/header.rb
 - lib/atco/journey.rb
 - lib/atco/location.rb
 - lib/atco/stop.rb
 - lib/atco/version.rb
 - spec/atco_spec.rb
-- spec/fixtures/example.cif
 - spec/fixtures/example.json
+- spec/fixtures/translink-example.cif
+- spec/integration/translink_spec.rb
 - spec/spec_helper.rb
+- spec/unit/header_spec.rb
 homepage: http://github.com/davidjrice/atco
 licenses:
 - MIT