lhj-tools 0.1.30 → 0.1.31

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d687a5f900925541da6435425f42e505ff3d80abcab177e1c5a619082da05b6a
-  data.tar.gz: 593fcf17a101528199b43cc9905865e94d185ba687e81f2733623a99c512981c
+  metadata.gz: 3b6137bd64ebed91f50de3ca6d35f4c65adcfa6ab2a7e57d5553e9fd49643aa5
+  data.tar.gz: e8f239f0d6893d8654a10f188413bd1b0f2ea6ee7dff4bd276e5727ca89d10cb
 SHA512:
-  metadata.gz: b4d2b8d1da73e9a0906501f26a419a7f7bde4195d428bd30b2c2e119d2c18882430e69015dacc9e88e1dc1a5c50ae986201cec584213230bc5c298bbe0331423
-  data.tar.gz: 1f1d936c41971b3ab3214c8eea90c49cbac6931e8c4b8174a11ae083d09cab9ad4ee0b3c05f718545d9763ffd07bb4b15c67f2c750bfa8a4dd2b8d53c1610ae0
+  metadata.gz: 4f84252797b129d329cfceddc18244253d5f3efedbcca4b9cdf6c432c532cedd4a583f93e7e658a5a7382a3c1350a3a7bad69e2b355f1e47558a45a0709fa654
+  data.tar.gz: 67e0baddcd638984097e5b37646d50533771130ea67313654c84fdd7a4a905646e3ff8c31b37db61203261927cd87b94802d9f0b2e9b3e101db5c484bf524a9b

data/lib/lhj/helper/vika_config.rb

@@ -16,20 +16,13 @@ module Lhj
       @yaml ||= YAML.load_file(config_file)
     end
 
-    def self.datasheet_id
-      config['datasheet_id']
-    end
-
-    def self.view_id
-      config['view_id']
+    def self.authorization
+      config['authorization']
     end
 
-    def self.record_id
-      config['record_id']
+    def self.display_fields
+      config['display_fields']
     end
 
-    def self.authorization
-      config['authorization']
-    end
   end
 end

data/lib/lhj/helper/vika_helper.rb

@@ -6,24 +6,271 @@ module Lhj
   # pgyer upload
   class VikaHelper
 
-    RECORDS_API_URL = 'https://api.vika.cn/fusion/v1/datasheets/dstmDXBooTe1Wj0QqE/views'.freeze
+    QUERY_SPACES_API_URL = 'https://api.vika.cn/fusion/v1/spaces'.freeze
 
-    def self.records
-      url = URI(RECORDS_API_URL)
+    QUERY_NODES_API_URL = 'https://api.vika.cn/fusion/v1/spaces/spcid/nodes'.freeze
+
+    QUERY_NODE_DETAIL_API_URL = 'https://api.vika.cn/fusion/v1/spaces/spcid/nodes/fodid'.freeze
+
+    QUERY_VIEWS_API_URL = 'https://api.vika.cn/fusion/v1/datasheets/dstid/views'.freeze
+
+    QUERY_RECORDS_API_URL = 'https://api.vika.cn/fusion/v1/datasheets/dstid/records?viewId=viwid'.freeze
+
+    QUERY_RECORDS_FIELD_KEY_API_URL = 'https://api.vika.cn/fusion/v1/datasheets/dstid/records?viewId=viwid&fieldKey=id'.freeze
+
+    RECORD_LINK_URL = 'https://vika.cn/workbench/dstid/viwid/recid'.freeze
+
+    def self.trans_vika(note)
+      str = note.dup
+      if /#.+#/ =~ str
+        note.scan(/#[^#]+#/) do |m|
+          ma = m.match(/rec\w+/)
+          next unless ma && ma[0] && ma[0].length.positive?
+
+          rec_id = ma[0]
+          rec_value = query_with_rec_id(rec_id)
+          str = str.gsub(m, rec_value) if rec_value && rec_value.length.positive?
+        end
+      end
+      str
+    end
+
+    def self.query_with_rec_id(rec_id)
+      file = vika_file
+      if File.exist?(file)
+        map = YAML.load_file(file)
+        key = map.keys.find { |k| k =~ /#{rec_id}/ }
+        if key && key.length.positive?
+          map[key]
+        else
+          query_network_with_rec_id(rec_id)
+        end
+      else
+        query_network_with_rec_id(rec_id)
+      end
+    end
+    
+    def self.query_network_with_rec_id(rec_id)
+      sync
+      key = data_source.keys.find { |k| k =~ /#{rec_id}/ }
+      data_source[key]
+    end
+
+    def self.sync
+      request_spaces
+      save_data_source
+    end
+
+    def self.request_spaces
+      response = fetch_result(:spaces)
+      handle_spaces_response(response)
+    end
+
+    def self.handle_spaces_response(res)
+      return unless res['code'].to_i == 200
+
+      res['data']['spaces'].each do |spc|
+        spc_id = spc['id']
+        request_nodes(spc_id)
+      end
+    end
+
+    def self.request_nodes(spc_id)
+      response = fetch_result(:nodes, spc_id)
+      handle_nodes_response(spc_id, response)
+    end
+
+    def self.handle_nodes_response(spc_id, res)
+      return unless res['code'].to_i == 200
+
+      res['data']['nodes'].each do |node|
+        fod_id = node['id']
+        request_node_detail(spc_id, fod_id)
+      end
+    end
+
+    def self.request_node_detail(spc_id, node_id)
+      response = fetch_result(:nodes_detail, spc_id, node_id)
+      handle_node_detail_response(spc_id, response)
+    end
+
+    def self.handle_node_detail_response(spc_id, res)
+      return unless res['code'].to_i == 200
+
+      res['data']['children'].each do |node|
+        fod_dst_id = node['id']
+        sym = type_from_id(fod_dst_id)
+        case sym
+        when :fod
+          request_node_detail(spc_id, fod_dst_id)
+        when :dst
+          request_views(fod_dst_id)
+        end
+      end
+    end
+
+    # 区分开，使用缓存dst来操作
+    def self.request_views(dst_id)
+      response = fetch_result(:views, dst_id)
+      handle_views_response(dst_id, response)
+    end
+
+    def self.handle_views_response(dst_id, res)
+      return unless res['code'].to_i == 200
+
+      res['data']['views'].each do |viw|
+        viw_id = viw['id']
+        request_records(dst_id, viw_id)
+      end
+    end
+
+    def self.request_records(dst_id, viw_id)
+      response = fetch_result(:records_key, dst_id, viw_id)
+      handle_records_response(dst_id, viw_id, response)
+    end
+
+    def self.handle_records_response(dst_id, viw_id, res)
+      return unless res['code'].to_i == 200
+
+      res['data']['records'].each do |rec|
+        # 处理查询记录显示
+        next unless rec['fields']
+
+        rec_id = rec['recordId']
+        rec['fields'].each do |k, v|
+          display_fields = Lhj::VikaConfig.display_fields
+          next unless display_fields.any? { |f| f.eql?(k) }
+
+          ds_key = "#{dst_id}_#{viw_id}_#{rec_id}"
+          ds_value = "#{v}[#{link_url(dst_id, viw_id, rec_id)}]"
+          data_source[ds_key] = ds_value
+        end
+      end
+    end
+
+    def self.link_url(dst_id, viw_id, rec_id)
+      url_with_kind(:link, dst_id, viw_id, rec_id)
+    end
+
+    def self.data_source
+      @data_source ||= {}
+    end
+
+    def self.save_data_source
+      File.write(vika_file, data_source.to_yaml)
+    end
+
+    def self.vika_file
+      file_with_name(vika_file_name)
+    end
+
+    def self.vika_file_name
+      'vika_data_source.yml'
+    end
+
+    def self.file_with_name(file_name)
+      target_folder = File.join(Lhj::Config.instance.home_dir, 'vika')
+      FileUtils.mkdir_p(target_folder) unless File.exist?(target_folder)
+      File.join(target_folder, file_name)
+    end
+
+    def self.fetch_result(kind, *params)
+      url = url_with_kind(kind, *params)
+      res = nil
+      res = get_with_url(url) if url.length.positive?
+      res
+    end
+
+    def self.url_with_kind(kind, *params)
+      url = ''
+      case kind
+      when :spaces
+        url = url_with_param(QUERY_SPACES_API_URL, *params)
+      when :nodes
+        url = url_with_param(QUERY_NODES_API_URL, *params)
+      when :nodes_detail
+        url = url_with_param(QUERY_NODE_DETAIL_API_URL, *params)
+      when :views
+        url = url_with_param(QUERY_VIEWS_API_URL, *params)
+      when :records
+        url = url_with_param(QUERY_RECORDS_API_URL, *params)
+      when :records_key
+        url = url_with_param(QUERY_RECORDS_FIELD_KEY_API_URL, *params)
+      when :link
+        url = url_with_param(RECORD_LINK_URL, *params)
+      end
+      url
+    end
+
+    def self.url_with_param(url, *params)
+      handle_url = url.dup
+      params.each do |p|
+        sym = type_from_id(p)
+        case sym
+        when :spc
+          handle_url.gsub!('spcid', p)
+        when :fod
+          handle_url.gsub!('fodid', p)
+        when :dst
+          handle_url.gsub!('dstid', p)
+        when :viw
+          handle_url.gsub!('viwid', p)
+        when :rec
+          handle_url.gsub!('recid', p)
+        end
+      end
+      handle_url
+    end
+
+    def self.get_with_url(api_url)
+      url = URI(api_url)
 
       https = Net::HTTP.new(url.host, url.port)
       https.use_ssl = true
 
       request = Net::HTTP::Get.new(url)
-      request["Authorization"] = "Bearer usk3cvEicoXG84y32J61BRg"
+      auth_str = "Bearer #{Lhj::VikaConfig.authorization}"
+      request['Authorization'] = auth_str
 
       response = https.request(request)
-      File.write(File.join(Lhj::Config.instance.home_dir, 'vika_all_records.json'), JSON.parse(response.read_body))
+      JSON.parse(response.read_body)
+    end
+
+    def self.type_from_id(id_str)
+      t = :spc
+      if spc_id? id_str
+        t = :spc
+      elsif fod_id? id_str
+        t = :fod
+      elsif dst_id? id_str
+        t = :dst
+      elsif viw_id? id_str
+        t = :viw
+      elsif rec_id? id_str
+        t = :rec
+      end
+      t
     end
 
-    def self.all_records
-      map = {'v6.2.0' => '月底发版分支'}
-      puts map.to_yaml
+    def self.spc_id?(id_str)
+      /^spc/ =~ id_str
     end
+
+    def self.fod_id?(id_str)
+      /^fod/ =~ id_str
+    end
+
+    def self.dst_id?(id_str)
+      /^dst/ =~ id_str
+    end
+
+    def self.viw_id?(id_str)
+      /^viw/ =~ id_str
+    end
+
+    def self.rec_id?(id_str)
+      /^rec/ =~ id_str
+    end
+
   end
 end

data/lib/lhj/tools/version.rb

@@ -2,6 +2,6 @@
 
 module Lhj
   module Tools
-    VERSION = "0.1.30"
+    VERSION = "0.1.31"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lhj-tools
 version: !ruby/object:Gem::Version
-  version: 0.1.30
+  version: 0.1.31
 platform: ruby
 authors:
 - lihaijian
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-04-15 00:00:00.000000000 Z
+date: 2022-04-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: claide