chotto 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9c41d0ab7f3b4a69a3ddc7386d5246476f177dbb2494f35323829fc00156df00
-  data.tar.gz: e3a675154168eb084157975d87efc4ab52446b160d4e53a6dbb687eb2ada1054
+  metadata.gz: 372dec253b24d3abd183e75e361210b2ec2c601a70466816789361b0e3070ab6
+  data.tar.gz: ee395a74ab114ec52d3a4e019f1e35e42fa972aeb7a17f3cd59cb6e797f2688a
 SHA512:
-  metadata.gz: c4108b72fba8e744a832d8c8238d99951b3070afbc7a710ec7d96cbff84b9ae7a908972dacad76fee9be2703c8240f43c73c073b64c28d1258d45c788f9eded7
-  data.tar.gz: 91243ec3f65d27ec16f85d005bdea9b09d4fe04ca58b798cce5501f123a4099ed268d0c765007b5230059d6d91e8e9b5d78cc12d4a849dc93cc6c4af9b51dd7e
+  metadata.gz: f00b7a2a92cbb8ae9c1785cf0c11a447a269f1dba097fdf161cb4beb1976252a5aef4a7e4cadce95e68e22b6cf65a1dece24069624966c7a3d69d30456237fb6
+  data.tar.gz: fe0350716b91cf3e3fb9a75413f91861120d56db2983c105e7233e0d7ab72525c650fa506b2b253f21ed9954c93d420ba44546bda63c85bf327c3c91ab59ba04

data/lib/chotto/config.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Chotto
+  Config = Struct.new(:database_path, :db_class, :only_new)
+end

data/lib/chotto/database.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Chotto
+  class Database
+    attr_reader :db
+
+    def initialize(path:, db_class:)
+      @db = db_class.new(path, mode: Notmuch::MODE_READ_WRITE)
+    end
+
+    def search_messages(query)
+      db.query(query).search_messages
+    end
+
+    def close
+      db.close
+    end
+  end
+end

data/lib/chotto/helpers.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Chotto
+  module Helpers
+    def self.header_name_from_dsl(method_name)
+      method_name
+        .to_s
+        .split('_')
+        .map(&:downcase)
+        .join('-')
+    end
+  end
+end

data/lib/chotto/message.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+module Chotto
+  class Message
+    attr_reader :message, :messages, :db
+
+    # attr_accessor :tags
+
+    def initialize(msg:, db:)
+      @message = msg
+      @db = db
+      # @tags = MessageTags.new(message: message)
+    end
+
+    def method_missing(method_name, *_args)
+      handle_get_header(Chotto::Helpers.header_name_from_dsl(method_name))
+    end
+
+    def tags
+      @tags ||= message.tags
+    end
+
+    attr_writer :tags
+
+    def save!
+      message.remove_all_tags
+      tags.each do |tag|
+        message.add_tag(tag.to_s)
+      end
+    end
+
+    def thread
+      thread_id = message.thread_id
+      MessageThread.new(thread_id: thread_id, db: db)
+    end
+
+    private
+
+    def handle_get_header(header_name)
+      message.header(header_name)
+    end
+  end
+end

data/lib/chotto/message_thread.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module Chotto
+  class MessageThread
+    attr_reader :thread_id, :db
+
+    def initialize(thread_id:, db:)
+      @thread_id = thread_id
+      @db = db
+    end
+
+    def each
+      db.search_messages("thread:#{thread_id}").each do |msg|
+        yield(
+            Message.new(
+              msg: msg,
+              db: db
+            ))
+      end
+    end
+  end
+end

data/lib/chotto/messages.rb

@@ -0,0 +1,98 @@
+# frozen_string_literal: true
+
+module Chotto
+  AND_CONJUCTION = :and
+  OR_CONJUCTION = :or
+  DEFAULT_CONJUCTION = AND_CONJUCTION
+  OLDEST_FIRST_ORDER = 'oldest_first'
+  NEWEST_FIRST_ORDER = 'newest_first'
+  DEFAULT_ORDER = OLDEST_FIRST_ORDER
+
+  class Messages
+    attr_accessor :query, :current_conjuction, :token_count, :order, :only_new
+    attr_reader :db
+
+    include Enumerable
+
+    def initialize(db:, only_new:)
+      @db = db
+      @current_conjuction = AND_CONJUCTION
+      @token_count = 0
+      @order = DEFAULT_ORDER
+      @only_new = only_new
+      clear_filter!
+    end
+
+    def or
+      @current_conjuction = OR_CONJUCTION
+      self
+    end
+
+    def and
+      @current_conjuction = AND_CONJUCTION
+      self
+    end
+
+    def filter(params)
+      case params
+      when String
+        add_filter_from_string(params)
+      when Hash
+        add_filter_from_hash(params)
+      end
+
+      self
+    end
+
+    def clear_filter!
+      @query = []
+    end
+
+    def newest_first
+      @order = NEWEST_FIRST_ORDER
+
+      self
+    end
+
+    def oldest_first
+      @order = OLDEST_FIRST_ORDER
+
+      self
+    end
+
+    def each
+      db.search_messages(query_string).each do |msg|
+        yield(
+            Message.new(
+              msg: msg,
+              db: db
+            ))
+      end
+    end
+
+    def add_filter_from_string(string)
+      @query << Token.new(current_conjuction, :direct, string, self)
+    end
+
+    def add_filter_from_hash(hash)
+      group = TokenGroup.new(current_conjuction, [], self)
+      hash.each do |key, value|
+        group.push_token(key, value)
+      end
+      @query << group
+    end
+
+    def query_string
+      string_from_filters = query.map(&:to_query).join(' ')
+      if only_new
+        "tag:new AND (#{string_from_filters})"
+      else
+        string_from_filters
+      end
+    end
+
+    def next_token_id
+      @token_count += 1
+    end
+  end
+end

data/lib/chotto/ruleset.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Chotto
+  RuleSet = Struct.new(:name, :db, :only_new, :rule) do
+    def messages
+      Messages.new(db: db, only_new: only_new)
+    end
+
+    def run
+      instance_eval(&rule)
+    end
+  end
+end

data/lib/chotto/token.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Chotto
+  Token = Struct.new(:conjuction, :field, :value, :messages) do
+    def to_query
+      conjuction_to_use = messages.next_token_id > 1 ? conjuction : ''
+      return "#{conjuction_to_use} (#{value})" if field == :direct
+
+      "#{conjuction_to_use} (#{field}:#{value})"
+    end
+  end
+end

data/lib/chotto/token_group.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module Chotto
+  TokenGroup = Struct.new(:conjuction, :tokens, :messages) do
+    def initialize(*)
+      super
+    end
+
+    def to_query
+      tokens.flat_map(&:to_query).join(' ').to_s
+    end
+
+    def push_token(field, value)
+      if value.is_a? String
+        tokens.push(Token.new(conjuction, field, value, messages))
+      elsif value.is_a? Array
+        group = TokenGroup.new(OR_CONJUCTION, [], messages)
+        value.each do |val|
+          group.push_token(field, val)
+        end
+        tokens.push group
+
+      end
+    end
+
+    def next_token_id
+      @token_count += 1
+    end
+  end
+end

metadata

@@ -1,24 +1,47 @@
 --- !ruby/object:Gem::Specification
 name: chotto
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - mms
 autorequire:
-bindir: bin
+bindir: exe
 cert_chain: []
-date: 2024-11-27 00:00:00.000000000 Z
-dependencies: []
-description:
+date: 2024-11-28 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3'
+description: Initial tagging script for Notmuch
 email: chotto@sapka.me
 executables:
 - chotto
 extensions: []
 extra_rdoc_files: []
 files:
-- bin/chotto
+- exe/chotto
 - lib/chotto.rb
+- lib/chotto/config.rb
+- lib/chotto/database.rb
+- lib/chotto/helpers.rb
+- lib/chotto/message.rb
+- lib/chotto/message_thread.rb
+- lib/chotto/messages.rb
+- lib/chotto/ruleset.rb
+- lib/chotto/token.rb
+- lib/chotto/token_group.rb
 homepage: https://crys.site/projects/chotto
 licenses:
 - BSD-3-Clause
@@ -39,8 +62,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.16
+rubygems_version: 3.5.23
 signing_key:
 specification_version: 4
-summary: Initial tagging script for Notmuch
+summary: chotto-0.1.1
 test_files: []