lucabook 0.2.8 → 0.2.17

data/lib/luca_book/list.rb

@@ -0,0 +1,138 @@
+# frozen_string_literal: true
+
+require 'pathname'
+require 'date'
+require 'luca_support'
+require 'luca_record'
+require 'luca_record/dict'
+require 'luca_book'
+
+# Journal List on specified term
+#
+module LucaBook
+  class List < LucaBook::Journal
+    @dirname = 'journals'
+
+    def initialize(data, start_date, code = nil)
+      @data = data
+      @code = code
+      @start = start_date
+      @dict = LucaRecord::Dict.load('base.tsv')
+    end
+
+    def self.term(from_year, from_month, to_year = from_year, to_month = from_month, code: nil, basedir: @dirname)
+      data = LucaBook::Journal.term(from_year, from_month, to_year, to_month, code).select do |dat|
+        if code.nil?
+          true
+        else
+          [:debit, :credit].map { |key| serialize_on_key(dat[key], :code) }.flatten.include?(code)
+        end
+      end
+      new data, Date.new(from_year.to_i, from_month.to_i, 1), code
+    end
+
+    def list_on_code
+      calc_code
+      convert_label
+      @data = [code_header] + @data.map do |dat|
+        date, txid = LucaSupport::Code.decode_id(dat[:id])
+        {}.tap do |res|
+          res['code'] = dat[:code]
+          res['date'] = date
+          res['no'] = txid
+          res['id'] = dat[:id]
+          res['diff'] = dat[:diff]
+          res['balance'] = dat[:balance]
+          res['counter_code'] = dat[:counter_code].length == 1 ? dat[:counter_code].first : dat[:counter_code]
+          res['note'] = dat[:note]
+        end
+      end
+      self
+    end
+
+    def list_journals
+      convert_label
+      @data = @data.map do |dat|
+        date, txid = LucaSupport::Code.decode_id(dat[:id])
+        {}.tap do |res|
+          res['date'] = date
+          res['no'] = txid
+          res['id'] = dat[:id]
+          res['debit_code'] = dat[:debit].length == 1 ? dat[:debit][0][:code] : dat[:debit].map { |d| d[:code] }
+          res['debit_amount'] =  dat[:debit].inject(0) { |sum, d| sum + d[:amount] }
+          res['credit_code'] = dat[:credit].length == 1 ? dat[:credit][0][:code] : dat[:credit].map { |d| d[:code] }
+          res['credit_amount'] = dat[:credit].inject(0) { |sum, d| sum + d[:amount] }
+          res['note'] = dat[:note]
+        end
+      end
+      self
+    end
+
+    def accumulate_code
+      @data.inject(BigDecimal('0')) do |sum, dat|
+        sum + Util.diff_by_code(dat[:debit], @code) - Util.diff_by_code(dat[:credit], @code)
+      end
+    end
+
+    def to_yaml
+      YAML.dump(LucaSupport::Code.readable(@data)).tap { |data| puts data }
+    end
+
+    private
+
+    def set_balance
+      return BigDecimal('0') if @code.nil? || /^[A-H]/.match(@code)
+
+      balance_dict = Dict.latest_balance
+      start_balance = BigDecimal(balance_dict.dig(@code.to_s, :balance) || '0')
+      start = Dict.issue_date(balance_dict)&.next_month
+      last = @start.prev_month
+      if last.year >= start.year && last.month >= start.month
+        start_balance + self.class.term(start.year, start.month, last.year, last.month, code: @code).accumulate_code
+      else
+        start_balance
+      end
+    end
+
+    def calc_code
+      @balance = set_balance
+      if @code
+        balance = @balance
+        @data.each do |dat|
+          dat[:diff] = Util.diff_by_code(dat[:debit], @code) - Util.diff_by_code(dat[:credit], @code)
+          balance += dat[:diff]
+          dat[:balance] = balance
+          dat[:code] = @code
+          counter = dat[:diff] * Util.pn_debit(@code) > 0 ? :credit : :debit
+          dat[:counter_code] = dat[counter].map { |d| d[:code] }
+        end
+      end
+      self
+    end
+
+    def convert_label
+      @data.each do |dat|
+        if @code
+          dat[:code] = "#{dat[:code]} #{@dict.dig(dat[:code], :label)}"
+          dat[:counter_code] = dat[:counter_code].map { |counter| "#{counter} #{@dict.dig(counter, :label)}" }
+        else
+          dat[:debit].each { |debit| debit[:code] = "#{debit[:code]} #{@dict.dig(debit[:code], :label)}" }
+          dat[:credit].each { |credit| credit[:code] = "#{credit[:code]} #{@dict.dig(credit[:code], :label)}" }
+        end
+      end
+      self
+    end
+
+    def dict
+      LucaBook::Dict::Data
+    end
+
+    def code_header
+      {}.tap do |h|
+        %w[code date no id diff balance counter_code note].each do |k|
+          h[k] = k == 'balance' ? @balance : ''
+        end
+      end
+    end
+  end
+end

data/lib/luca_book/setup.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require 'luca_book'
+require 'fileutils'
+
+module LucaBook
+  class Setup
+    # create project skeleton under specified directory
+    def self.create_project(country = nil, dir = LucaSupport::Config::Pjdir)
+      FileUtils.mkdir_p(dir) unless Dir.exist?(dir)
+      Dir.chdir(dir) do
+        %w[data/journals data/balance dict].each do |subdir|
+          FileUtils.mkdir_p(subdir) unless Dir.exist?(subdir)
+        end
+        dict = if File.exist?("#{__dir__}/templates/dict-#{country}.tsv")
+                 "dict-#{country}.tsv"
+               else
+                 'dict-en.tsv'
+               end
+        FileUtils.cp("#{__dir__}/templates/#{dict}", 'dict/base.tsv') unless File.exist?('dict/base.tsv')
+        prepare_starttsv(dict) unless File.exist? 'data/balance/start.tsv'
+      end
+    end
+
+    # Generate initial balance template.
+    # Codes are same as base dictionary.
+    # The previous month of start date is better for _date.
+    #
+    def self.prepare_starttsv(dict)
+      CSV.open('data/balance/start.tsv', 'w', col_sep: "\t", encoding: 'UTF-8') do |csv|
+        csv << ['code', 'label', 'balance']
+        csv << ['_date', '2020-1-1']
+        CSV.open("#{__dir__}/templates/#{dict}", 'r', col_sep: "\t", encoding: 'UTF-8').each do |row|
+          csv << row if /^[1-9]/.match(row[0])
+        end
+      end
+    end
+  end
+end

data/lib/luca_book/state.rb

@@ -1,3 +1,4 @@
+# frozen_string_literal: true
 
 require 'csv'
 require 'pathname'
@@ -7,7 +8,6 @@ require 'luca_record'
 require 'luca_record/dict'
 require 'luca_book'
 
-#
 # Statement on specified term
 #
 module LucaBook
@@ -17,9 +17,11 @@ module LucaBook
 
     attr_reader :statement
 
-    def initialize(data)
+    def initialize(data, count = nil)
       @data = data
+      @count = count
       @dict = LucaRecord::Dict.load('base.tsv')
+      @start_balance = set_balance
     end
 
     # TODO: not compatible with LucaRecord::Base.open_records
@@ -41,66 +43,42 @@ module LucaBook
       last_date = Date.new(to_year.to_i, to_month.to_i, -1)
       raise 'invalid term specified' if date > last_date
 
+      counts = []
       reports = [].tap do |r|
         while date <= last_date do
-          r << accumulate_month(date.year, date.month)
+          diff, count = accumulate_month(date.year, date.month)
+          r << diff
+          counts << count
           date = date.next_month
         end
       end
-      new reports
+      new reports, counts
     end
 
     def by_code(code, year=nil, month=nil)
-      raise "not supported year range yet" if ! year.nil? && month.nil?
+      raise 'not supported year range yet' if ! year.nil? && month.nil?
 
-      bl = @book.load_start.dig(code) || 0
-      full_term = scan_terms(LucaSupport::Config::Pjdir)
+      balance = @book.load_start.dig(code) || 0
+      full_term = self.class.scan_terms
       if ! month.nil?
-        pre_term = full_term.select{|y,m| y <= year.to_i && m < month.to_i }
-        bl += pre_term.map{|y,m| self.class.net(y, m)}.inject(0){|sum, h| sum + h[code]}
-        [{ code: code, balance: bl, note: "#{code} #{dict.dig(code, :label)}" }] + records_with_balance(year, month, code, bl)
+        pre_term = full_term.select { |y, m| y <= year.to_i && m < month.to_i }
+        balance += pre_term.map { |y, m| self.class.net(y, m)}.inject(0){|sum, h| sum + h[code] }
+        [{ code: code, balance: balance, note: "#{code} #{@dict.dig(code, :label)}" }] + records_with_balance(year, month, code, balance)
       else
-        start = { code: code, balance: bl, note: "#{code} #{dict.dig(code, :label)}" }
-        full_term.map {|y, m| y }.uniq.map {|y|
-          records_with_balance(y, nil, code, bl)
+        start = { code: code, balance: balance, note: "#{code} #{@dict.dig(code, :label)}" }
+        full_term.map { |y, m| y }.uniq.map { |y|
+          records_with_balance(y, nil, code, balance)
         }.flatten.prepend(start)
       end
     end
 
     def records_with_balance(year, month, code, balance)
       @book.search(year, month, nil, code).each do |h|
-        balance += self.class.calc_diff(amount_by_code(h[:debit], code), code) - @book.calc_diff(amount_by_code(h[:credit], code), code)
+        balance += Util.calc_diff(Util.amount_by_code(h[:debit], code), code) - Util.calc_diff(Util.amount_by_code(h[:credit], code), code)
         h[:balance] = balance
       end
     end
 
-    #
-    # TODO: useless method. consider to remove
-    #
-    def accumulate_all
-      current = @book.load_start
-      target = []
-      Dir.chdir(@book.pjdir) do
-        net_records = scan_terms(@book.pjdir).map do |year, month|
-          target << [year, month]
-          accumulate_month(year, month)
-        end
-        all_keys = net_records.map{|h| h.keys}.flatten.uniq
-        net_records.each.with_index(0) do |diff, i|
-          all_keys.each {|key| diff[key] = 0 unless diff.has_key?(key)}
-          diff.each do |k,v|
-            if current[k]
-              current[k] += v
-            else
-              current[k] = v
-            end
-          end
-          f = { target: "#{target[i][0]}-#{target[i][1]}", diff: diff.sort, current: current.sort }
-          yield f
-        end
-      end
-    end
-
     def to_yaml
       YAML.dump(code2label).tap { |data| puts data }
     end
@@ -114,33 +92,105 @@ module LucaBook
       end
     end
 
-    def bs
-      @statement = @data.map do |data|
-        data.select { |k, v| /^[0-9].+/.match(k) }
+    def stats(level = nil)
+      keys = @count.map(&:keys).flatten.uniq.sort
+      @count.map! do |data|
+        keys.each do |k|
+          data[k] ||= 0
+          next if level.nil? || k.length <= level
+
+          if data[k[0, level]]
+            data[k[0, level]] += data[k]
+          else
+            data[k[0, level]] = data[k]
+          end
+        end
+        data.select! { |k, _v| k.length <= level } if level
+        data.sort.to_h
       end
-      self
+      keys.map! { |k| k[0, level] }.uniq.select! { |k| k.length <= level } if level
+      @count.prepend({}.tap { |header| keys.each { |k| header[k] = @dict.dig(k, :label) }})
+      puts YAML.dump(@count)
+      @count
     end
 
-    def pl
-      @statement = @data.map do |data|
-        data.select { |k, v| /^[A-F].+/.match(k) }
+    def bs(level = 3, legal: false)
+      @data.map! { |data| data.select { |k, _v| k.length <= level } }
+      @data.map! { |data| code_sum(data).merge(data) } if legal
+      base = accumulate_balance(@data)
+      length = [base[:debit].length, base[:credit].length].max
+      @statement = [].tap do |a|
+        length.times do |i|
+          {}.tap do |res|
+            res['debit_label'] = base[:debit][i] ? @dict.dig(base[:debit][i].keys[0], :label) : ''
+            res['debit_balance'] = base[:debit][i] ? @start_balance.dig(base[:debit][i].keys[0]) + base[:debit][i].values[0] : ''
+            res['debit_diff'] = base[:debit][i] ? base[:debit][i].values[0] : ''
+            res['credit_label'] = base[:credit][i] ? @dict.dig(base[:credit][i].keys[0], :label) : ''
+            res['credit_balance'] = base[:credit][i] ? @start_balance.dig(base[:credit][i].keys[0]) + base[:credit][i].values[0] : ''
+            res['credit_diff'] = base[:credit][i] ? base[:credit][i].values[0] : ''
+            a << res
+          end
+        end
       end
+      puts YAML.dump(@statement)
       self
     end
 
-    def self.accumulate_month(year, month)
-      monthly_record = net(year, month)
-      total_subaccount(monthly_record)
+    def accumulate_balance(monthly_diffs)
+      data = monthly_diffs.each_with_object({}) do |month, h|
+        month.each do |k, v|
+          h[k] = h[k].nil? ? v : h[k] + v
+        end
+      end.sort.to_h
+      { debit: [], credit: [] }.tap do |res|
+        data.each do |k, v|
+          case k
+          when /^[0-4].*/
+            res[:debit] << { k => v }
+          when /^[5-9H].*/
+            res[:credit] << { k => v }
+          end
+        end
+      end
     end
 
-    def amount_by_code(items, code)
-      items
-        .select{|item| item.dig(:code) == code }
-        .inject(0){|sum, item| sum + item[:amount] }
+    def pl
+      @statement = @data.map { |data| data.select { |k, _v| /^[A-H].+/.match(k) } }
+      @statement << @statement.each_with_object({}) { |item, h| item.each { |k, v| h[k].nil? ? h[k] = v : h[k] += v } }
+      self
     end
 
+    def self.accumulate_month(year, month)
+      monthly_record, count = net(year, month)
+      [total_subaccount(monthly_record), count]
+    end
+
+    # Accumulate Level 2, 3 account.
+    #
     def self.total_subaccount(report)
-      report.dup.tap do |res|
+      {}.tap do |res|
+        res['10'] = sum_matched(report, /^[123][0-9A-Z]{2,}/)
+        res['40'] = sum_matched(report, /^[4][0-9A-Z]{2,}/)
+        res['50'] = sum_matched(report, /^[56][0-9A-Z]{2,}/)
+        res['70'] = sum_matched(report, /^[78][0-9A-Z]{2,}/)
+        res['8ZZ'] = res['50'] + res['70']
+        res['9ZZ'] = sum_matched(report, /^[9][0-9A-Z]{2,}/)
+        res['A0'] = sum_matched(report, /^[A][0-9A-Z]{2,}/)
+        res['B0'] = sum_matched(report, /^[B][0-9A-Z]{2,}/)
+        res['BA'] = res['A0'] - res['B0']
+        res['C0'] = sum_matched(report, /^[C][0-9A-Z]{2,}/)
+        res['CA'] = res['BA'] - res['C0']
+        res['D0'] = sum_matched(report, /^[D][0-9A-Z]{2,}/)
+        res['E0'] = sum_matched(report, /^[E][0-9A-Z]{2,}/)
+        res['EA'] = res['CA'] + res['D0'] - res['E0']
+        res['F0'] = sum_matched(report, /^[F][0-9A-Z]{2,}/)
+        res['G0'] = sum_matched(report, /^[G][0-9A-Z]{2,}/)
+        res['GA'] = res['EA'] + res['F0'] - res['G0']
+        res['HA'] = res['GA'] - sum_matched(report, /^[H][0-9A-Z]{2,}/)
+
+        res['1'] = res['10'] + res['40']
+        res['5'] = res['8ZZ'] + res['9ZZ'] + res['HA']
+
         report.each do |k, v|
           if k.length >= 4
             if res[k[0, 3]]
@@ -150,26 +200,23 @@ module LucaBook
             end
           end
         end
-        res['10'] = sum_matched(report, /^[123].[^0]/)
-        res['40'] = sum_matched(report, /^[4].[^0]}/)
-        res['50'] = sum_matched(report, /^[56].[^0]/)
-        res['70'] = sum_matched(report, /^[78].[^0]/)
-        res['90'] = sum_matched(report, /^[9].[^0]/)
-        res['A0'] = sum_matched(report, /^[A].[^0]/)
-        res['B0'] = sum_matched(report, /^[B].[^0]/)
-        res['BA'] = res['A0'] - res['B0']
-        res['C0'] = sum_matched(report, /^[C].[^0]/)
-        res['CA'] = res['BA'] - res['C0']
-        res['D0'] = sum_matched(report, /^[D].[^0]/)
-        res['E0'] = sum_matched(report, /^[E].[^0]/)
-        res['EA'] = res['CA'] + res['D0'] - res['E0']
-        res['F0'] = sum_matched(report, /^[F].[^0]/)
-        res['G0'] = sum_matched(report, /^[G].[^0]/)
-        res['GA'] = res['EA'] + res['F0'] - res['G0']
-        res['HA'] = res['GA'] - sum_matched(report, /^[H].[^0]/)
+        res.sort.to_h
       end
     end
 
+    def code_sum(report)
+      legal_items.each.with_object({}) do |k, h|
+        h[k] = self.class.sum_matched(report, /^#{k}.*/)
+      end
+    end
+
+    def set_balance
+      base = Dict.latest_balance.each_with_object({}) do |(k, v), h|
+        h[k] = v[:balance].to_i if v[:balance]
+      end
+      code_sum(base).merge(self.class.total_subaccount(base))
+    end
+
     def self.sum_matched(report, reg)
       report.select { |k, v| reg.match(k)}.values.sum
     end
@@ -181,7 +228,7 @@ module LucaBook
         # TODO: date based range search
       end
 
-      sum = { debit: {}, credit: {} }
+      sum = { debit: {}, credit: {}, debit_count: {}, credit_count: {} }
       idx_memo = []
       asof(year, month) do |f, _path|
         CSV.new(f, headers: false, col_sep: "\t", encoding: 'UTF-8')
@@ -190,14 +237,26 @@ module LucaBook
           case i
           when 0
             idx_memo = row.map(&:to_s)
-            idx_memo.each { |r| sum[:debit][r] ||= 0 }
+            idx_memo.each do |r|
+              sum[:debit][r] ||= 0
+              sum[:debit_count][r] ||= 0
+            end
           when 1
-            row.each_with_index { |r, i| sum[:debit][idx_memo[i]] += r.to_i } # TODO: bigdecimal support
+            row.each_with_index do |r, j|
+              sum[:debit][idx_memo[j]] += r.to_i # TODO: bigdecimal support
+              sum[:debit_count][idx_memo[j]] += 1
+            end
           when 2
             idx_memo = row.map(&:to_s)
-            idx_memo.each { |r| sum[:credit][r] ||= 0 }
+            idx_memo.each do |r|
+              sum[:credit][r] ||= 0
+              sum[:credit_count][r] ||= 0
+            end
           when 3
-            row.each_with_index { |r, i| sum[:credit][idx_memo[i]] += r.to_i } # TODO: bigdecimal support
+            row.each_with_index do |r, j|
+              sum[:credit][idx_memo[j]] += r.to_i # TODO: bigdecimal support
+              sum[:credit_count][idx_memo[j]] += 1
+            end
           else
             puts row # for debug
           end
@@ -210,17 +269,20 @@ module LucaBook
     def self.net(year, month = nil, code = nil, date_range = nil)
       g = gross(year, month, code, date_range)
       idx = (g[:debit].keys + g[:credit].keys).uniq.sort
-      {}.tap do |sum|
+      count = {}
+      diff = {}.tap do |sum|
         idx.each do |code|
-          sum[code] = g.dig(:debit, code).nil? ? 0 : calc_diff(g[:debit][code], code)
-          sum[code] -= g.dig(:credit, code).nil? ? 0 : calc_diff(g[:credit][code], code)
+          sum[code] = g.dig(:debit, code).nil? ? 0 : Util.calc_diff(g[:debit][code], code)
+          sum[code] -= g.dig(:credit, code).nil? ? 0 : Util.calc_diff(g[:credit][code], code)
+          count[code] = (g.dig(:debit_count, code) || 0) + (g.dig(:credit_count, code) || 0)
         end
       end
+      [diff, count]
     end
 
     # TODO: replace load_tsv -> generic load_tsv_dict
     def load_start
-      file = LucaSupport::Config::Pjdir + 'start.tsv'
+      file = Pathname(LucaSupport::Config::Pjdir) / 'data' / 'balance' / 'start.tsv'
       {}.tap do |dic|
         load_tsv(file) do |row|
           dic[row[0]] = row[2].to_i if ! row[2].nil?
@@ -235,24 +297,15 @@ module LucaBook
       data.each { |row| yield row }
     end
 
-    def self.calc_diff(num, code)
-      amount = /\./.match(num.to_s) ? BigDecimal(num) : num.to_i
-      amount * pn_debit(code.to_s)
-    end
+    private
 
-    def self.pn_debit(code)
-      case code
-      when /^[0-4BCEGH]/
-        1
-      when /^[5-9ADF]/
-        -1
-      else
-        nil
-      end
-    end
+    def legal_items
+      return [] unless LucaSupport::Config::COUNTRY
 
-    def dict
-      LucaBook::Dict::Data
+      case LucaSupport::Config::COUNTRY
+      when 'jp'
+        ['91', '911', '912', '913', '9131', '9132', '914', '9141', '9142', '915', '916', '92', '93']
+      end
     end
   end
 end