notion_rb 0.1.0

data/lib/notion_rb/api/base.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+module NotionRb
+  module Api
+    class Base
+      BASE_URL = 'https://www.notion.so/'
+      API_BASE_URL = BASE_URL + 'api/v3/'
+      SIGNED_URL_PREFIX = 'https://www.notion.so/signed/'
+      S3_URL_PREFIX = 'https://s3-us-west-2.amazonaws.com/secure.notion-static.com/'
+      S3_URL_PREFIX_ENCODED = 'https://s3.us-west-2.amazonaws.com/secure.notion-static.com/'
+      REDEFINE_EXCEPTION = 'Define methods on subclass'
+
+      def initialize(params)
+        @notion_id = params[:notion_id]
+        @token_v2 = NotionRb.config[:token_v2]
+        @converter = NotionRb::Utils::Converter.new
+      end
+
+      def call
+        response
+      end
+
+      def success?
+        response.code == '200'
+      rescue Mechanize::ResponseCodeError
+        false
+      end
+
+      private
+
+      def agent
+        return @agent if @agent
+
+        @agent = Mechanize.new
+        cookie = Mechanize::Cookie.new(domain: 'www.notion.so',
+                                       name: 'token_v2',
+                                       value: @token_v2,
+                                       path: '/',
+                                       expires: (Date.today + 1).to_s)
+        @agent.tap { |agent| agent.cookie_jar << cookie }
+      end
+
+      def response
+        @response ||= agent.post("#{API_BASE_URL}#{url}", params.to_json, 'Content-Type' => 'application/json')
+      end
+
+      def url
+        raise REDEFINE_EXCEPTION
+      end
+
+      def params
+        raise REDEFINE_EXCEPTION
+      end
+    end
+  end
+end

data/lib/notion_rb/api/create.rb

@@ -0,0 +1,74 @@
+# frozen_string_literal: true
+
+module NotionRb
+  module Api
+    class Create < Base
+      def initialize(params)
+        super
+
+        @parent_id = params[:parent_id]
+        @notion_id = SecureRandom.uuid
+        @created_at = (Time.now.to_i / 100) * 100_000
+      end
+
+      def block_uuid
+        success? && @notion_id
+      end
+
+      private
+
+      def url
+        'submitTransaction'
+      end
+
+      def params
+        {
+          requestId: SecureRandom.uuid,
+          transactions: [{
+            id: SecureRandom.uuid,
+            operations: [{
+              id: @notion_id,
+              table: 'block',
+              path: [],
+              command: 'set',
+              args: { type: 'text', id: @notion_id, version: 1 }
+            }, {
+              id: @notion_id,
+              table: 'block',
+              path: [],
+              command: 'update',
+              args: { parent_id: @parent_id, parent_table: 'block', alive: true }
+            }, {
+              id: @parent_id,
+              table: 'block',
+              path: ['content'],
+              command: 'listAfter',
+              args: { id: @notion_id }
+            }, {
+              id: @notion_id,
+              table: 'block',
+              path: %w[
+                properties
+                title
+              ],
+              command: 'set',
+              args: []
+            }, {
+              id: @notion_id,
+              table: 'block',
+              path: ['created_time'],
+              command: 'set',
+              args: @created_at
+            }, {
+              id: @notion_id,
+              table: 'block',
+              path: ['last_edited_time'],
+              command: 'set',
+              args: @created_at
+            }]
+          }]
+        }
+      end
+    end
+  end
+end

data/lib/notion_rb/api/destroy.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module NotionRb
+  module Api
+    class Destroy < Base
+      def initialize(params)
+        super
+
+        @parent_id = params[:parent_id]
+      end
+
+      private
+
+      def url
+        'submitTransaction'
+      end
+
+      def params
+        {
+          requestId: SecureRandom.uuid,
+          transactions: [{
+            id: SecureRandom.uuid,
+            operations: [{
+              id: @notion_id,
+              table: 'block',
+              path: [],
+              command: 'update',
+              args: { alive: false }
+            }, {
+              id: @parent_id,
+              table: 'block',
+              path: ['content'],
+              command: 'listRemove',
+              args: { id: @notion_id }
+            }]
+          }]
+        }
+      end
+    end
+  end
+end

data/lib/notion_rb/api/get.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+module NotionRb
+  module Api
+    class Get < Base
+      def initialize(_params)
+        super
+
+        @blocks = []
+        @data = {}
+      end
+
+      def blocks
+        return @blocks if @blocks.any?
+
+        call
+        @blocks
+      end
+
+      def call
+        body = JSON.parse(response.body)
+        @collection_rows = body.dig('result', 'blockIds')
+        data = body.dig('recordMap', 'block').merge(body.dig('recordMap', 'collection') || {})
+        data.values.each do |value|
+          convert_values(value.dig('value'))
+        end
+      end
+
+      private
+
+      def url
+        'loadPageChunk'
+      end
+
+      def params
+        {
+          chunkNumber: 0,
+          cursor: { stack: [] },
+          limit: 50,
+          pageId: @notion_id,
+          verticalColumns: false
+        }
+      end
+
+      def convert_values(value)
+        add_collection_data(value)
+        @blocks << @converter.convert(value)
+      end
+
+      def add_collection_data(value)
+        return unless value.key? 'schema'
+
+        value['content'] = @collection_rows if @collection_rows
+        value['type'] = 'collection'
+      end
+    end
+  end
+end

data/lib/notion_rb/api/query_collection.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module NotionRb
+  module Api
+    class QueryCollection < Get
+      def initialize(params)
+        super
+
+        @collection_id = params[:collection_id]
+        @view_id = params[:view_id]
+        @schema = nil
+      end
+
+      private
+
+      def url
+        'queryCollection'
+      end
+
+      def params
+        {
+          collectionId: @collection_id,
+          collectionViewId: @view_id,
+          query: { agregrate: [{}], filter: [], sort: [], filter_operation: 'and' },
+          loader: { type: 'table', limit: 70, userTimeZone: 'America/Santiago', userLocale: 'en', loadContentCover: true }
+        }
+      end
+    end
+  end
+end

data/lib/notion_rb/api/restore.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+module NotionRb
+  module Api
+    class Restore < Base
+      def initialize(params)
+        super
+
+        @parent_id = params[:parent_id]
+      end
+
+
+      private
+
+      def url
+        'submitTransaction'
+      end
+
+      def params
+        {
+          requestId: SecureRandom.uuid,
+          transactions: [{
+            id: SecureRandom.uuid,
+            operations: [{
+              id: @notion_id,
+              table: 'block',
+              path: [],
+              command: 'update',
+              args: { parent_id: @parent_id, parent_table: 'block', alive: true }
+            }, {
+              id: @parent_id,
+              table: 'block',
+              path: ['content'],
+              command: 'listAfter',
+              args: { id: @notion_id }
+            }]
+          }]
+        }
+      end
+    end
+  end
+end

data/lib/notion_rb/api/update.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module NotionRb
+  module Api
+    class Update < Base
+      def initialize(params)
+        super
+
+        @title = params[:title]
+        @block_type = params[:block_type]
+      end
+
+      private
+
+      def url
+        'submitTransaction'
+      end
+
+      def params
+        {
+          requestId: SecureRandom.uuid,
+          transactions: [{
+            id: SecureRandom.uuid,
+            operations: [{
+              id: @notion_id,
+              table: 'block',
+              path: %w[properties title],
+              command: 'set',
+              args: [[@title]]
+            }]
+          }, {
+            id: SecureRandom.uuid,
+            operations: [{
+              id: @notion_id,
+              table: 'block',
+              path: [],
+              command: 'update',
+              args: { type: @block_type }
+            }]
+          }]
+        }
+      end
+    end
+  end
+end

data/lib/notion_rb/block.rb

@@ -0,0 +1,97 @@
+# frozen_string_literal: true
+
+module NotionRb
+  class Block
+    include NotionRb::Utils::UuidValidator
+    include NotionRb::Utils::Types
+
+    attr_accessor :uuid
+
+    def initialize(url_or_uuid)
+      @block_container = NotionRb::Utils::BlockCache.instance
+      @uuid = parse_uuid url_or_uuid
+      get_resource
+    end
+
+    def title
+      @block[:title]
+    end
+
+    def title=(title)
+      @block[:title] = title
+      save
+    end
+
+    def type
+      @block[:block_type]
+    end
+
+    def type=(type)
+      return unless valid_block_type?(type)
+
+      @block[:block_type] = type
+      save
+    end
+
+    def metadata
+      metadata = @block[:metadata].except(:properties)
+      add_parent_metadata(metadata)
+    end
+
+    def parent
+      @parent ||= self.class.new(@block[:parent_id])
+    end
+
+    def children
+      if type == 'collection_view_page'
+        @children ||= [self.class.new(metadata[:collection_id])]
+      else
+        @children ||= @block_container.children(@uuid).map { |child_uuid| self.class.new(child_uuid) }
+      end
+    end
+
+    def create_child
+      creator = NotionRb::Api::Create.new(parent_id: @uuid)
+      return nil unless creator.success?
+
+      @block[:children] << creator.block_uuid
+      self.class.new(creator.block_uuid)
+    end
+
+    def save
+      # TODO: add validations if needed
+      post_resource
+    end
+
+    def destroy
+      NotionRb::Api::Destroy.new(notion_id: @uuid, parent_id: @block[:parent_id]).success?
+    end
+
+    def restore
+      NotionRb::Api::Restore.new(notion_id: @uuid, parent_id: @block[:parent_id]).success?
+    end
+
+    private
+
+    def get_resource
+      @block = @block_container.find(@uuid)
+      @block_container.add_collection_view_blocks(@block)
+    end
+
+    def post_resource
+      # TODO: detect changes if any before post
+      NotionRb::Api::Update.new(notion_id: @uuid, title: @block[:title], block_type: @block[:block_type]).success?
+    end
+
+    def add_parent_metadata(metadata)
+      return metadata unless parent.type == 'collection'
+
+      parent_schema = parent.metadata[:schema]
+      @block.dig(:metadata, :properties).each do |key, value|
+        schema = parent_schema[key]
+        metadata.merge!(schema['name'] => value[0][0])
+      end
+      metadata
+    end
+  end
+end