Skip to content

保管庫のファイル更新を検知して自動再読込するようにした #13

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
soukouki merged 3 commits into from
Jul 10, 2026
Merged

保管庫のファイル更新を検知して自動再読込するようにした #13

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
571218d
保管庫のファイルが更新されていたときに再読込するようにした
soukouki Jul 10, 2026
929123b
フレーキーなテストを解消
soukouki Jul 10, 2026
f06c87f
YAMLのパースエラーのメッセージが鬱陶しかったので標準出力にstubを入れるようにした
soukouki Jul 10, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
34 changes: 8 additions & 26 deletions lib/exe.rb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -78,14 +78,13 @@
end
end

# Obsidian Vaultの初期化
$vault = ObsidianFetch::Vault.new(vault_paths)
STDERR.puts "Found #{$vault.notes.size} notes"
STDERR.puts "Found #{$vault.links_by_file_name.size} links and #{$vault.links_by_file_path.size} files linked by notes"
$manager = ObsidianFetch::VaultManager.new(vault_paths)
STDERR.puts "Found #{$manager.vault.notes.size} notes"
STDERR.puts "Found #{$manager.vault.links_by_file_name.size} links and #{$manager.vault.links_by_file_path.size} files linked by notes"
STDERR.puts "Transport type: #{transport_type}"
STDERR.puts "Port: #{port}" if transport_type == 'streamable-http'

# readツールの定義
# MCPツール定義
class ReadTool < MCP::Tool
description "Read a note from Obsidian vault. If multiple notes with the same name are found, all will be shown."
input_schema(
Expand All @@ -94,18 +93,13 @@ class ReadTool < MCP::Tool
},
required: ["name"]
)

def self.call(name:)
# 名前が文字列でない場合
return MCP::Tool::Response.new([{ type: "text", text: "Name must be a string" }], error: true) unless name.is_a?(String)

# Vaultからノートを読み取る
result = $vault.tool_read(name)
result = ObsidianFetch.read_note($manager.vault, name)
MCP::Tool::Response.new([{ type: "text", text: result.text }], error: result.error)
end
end

# listツールの定義
class ListTool < MCP::Tool
description "Search for files with matching names partially."
input_schema(
Expand All @@ -116,19 +110,16 @@ class ListTool < MCP::Tool
)

def self.call(name:)
# 名前が文字列でない場合
return MCP::Tool::Response.new([{ type: "text", text: "Name must be a string" }], error: true) unless name.is_a?(String)

# Vaultからノートを検索
result = $vault.tool_list(name)
result = ObsidianFetch.list_notes($manager.vault, name)
MCP::Tool::Response.new([{ type: "text", text: result.text }], error: result.error)
end
end

# MCPサーバーの初期化
server = MCP::Server.new(
name: "obsidian-fetch",
version: "0.1.0",
version: ObsidianFetch::VERSION,
tools: [ReadTool, ListTool],
)

Expand All @@ -137,31 +128,22 @@ def self.call(name:)
when 'streamable-http'
require 'puma'
require 'rack'

# Streamable HTTP トランスポートを使用してサーバーを起動
transport = MCP::Server::Transports::StreamableHTTPTransport.new(server, stateless: true)

# PumaでRack appとして起動
$stdout.sync = true
$stderr.sync = true

server_obj = Puma::Server.new(transport) do
max_threads 16
min_threads 1
end
server_obj.add_tcp_listener "0.0.0.0", port

STDERR.puts "Streamable HTTP transport started on http://localhost:#{port}"
STDERR.puts "Press Ctrl+C to stop"

# サーバーを起動してメインプロセスをブロックする
server_thread = server_obj.run
server_thread.join
when 'stdio'
# Stdioトランスポートを使用してサーバーを起動
transport = MCP::Server::Transports::StdioTransport.new(server)
transport.open
else
STDERR.puts "Unknown transport type: #{transport_type}. Supported: stdio, sse"
exit 1
end
end
250 changes: 3 additions & 247 deletions lib/obsidian_fetch.rb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,254 +4,10 @@
require 'date'

require_relative "obsidian_fetch/version"
require_relative "obsidian_fetch/vault"
require_relative "obsidian_fetch/vault_manager"
require_relative "obsidian_fetch/logic"

module ObsidianFetch
class Error < StandardError; end

# ツール呼び出しの結果を保持する
Result = Struct.new(:text, :error, keyword_init: true) do
def initialize(text:, error: false)
super(text: text, error: error)
end
end

class Vault
attr_reader :notes, :links_by_file_path, :links_by_file_name

def initialize vault_pathes
@vault_pathes = vault_pathes

# key: note_name, value: [file_pathes]
@notes = {}

# key: file_path, value: [file_pathes]
@links_by_file_path = {}

# ノートが見つからなかった場合に被リンクを表示するためのハッシュ
# key: note_name, value: [file_pathes]
@links_by_file_name = {}

collect_pathes()
collect_notes()
collect_links()
end

private

def collect_pathes
@all_pathes = @vault_pathes.flat_map do |vault_path|
Dir.glob("#{vault_path}/**/*.md")
end
end

# ノート名を正規化する
# ファイル名に使えない文字については除去しない。aliasで設定されている場合もあるため。
def self.normalize_note_name(note_name)
# ノート名を小文字に変換し、リンクに使えない文字を除去する
note_name.gsub(/[\[\]#\^\|]/, '')
end

def collect_notes
# ノート名を整理するための処理
@all_pathes.each do |file_path|
file_name = File.basename(file_path, '.md')
file_name = Vault.normalize_note_name(file_name)
@notes[file_name] ||= []
@notes[file_name] << file_path
content = open(file_path) { |f| f.read.force_encoding('UTF-8') }
# もしaliasが設定されていれば、それも追加する
frontmatter_str = content.match(/\A---(.*?\n)---/m)&.[](1)
if frontmatter_str
begin
frontmatter = YAML.safe_load(frontmatter_str, symbolize_names: true, aliases: true, permitted_classes: [Date])
rescue Psych::SyntaxError => e
puts "YAML syntax error in #{file_path}: #{e.message}"
next
end
Array(frontmatter&.[](:aliases) || []).each do |alias_name|
alias_name = Vault.normalize_note_name(alias_name)
@notes[alias_name] ||= []
@notes[alias_name] << file_path
end
end
end
@notes.each{|_name, file_pathes| file_pathes.uniq!}
end

def collect_links
# リンクを整理するための処理
# リンクを整理するためにノート名のリストを使用するため、ここで処理する
@all_pathes.each do |file_path|
file_name = File.basename(file_path, '.md')
file_name = Vault.normalize_note_name(file_name)
content = open(file_path) { |f| f.read.force_encoding('UTF-8') }
# もしリンクが設定されていれば、linksに追加する
# [[link]]と[[link|displayname]]の場合
# linkに.mdが付いている場合、付いていない場合両方を考慮する
# TODO : linkにディレクトリ名が着くことがあるので、それを除去する
content.scan(/\[\[(.*?)(?:\|.*)?\]\]/) do |match|
link_name = match[0]
link_name = link_name.sub(/\.md$/, '') # .mdを削除
# #、^以降の文字列を削除
link_name = link_name.sub(/#.*$/, '')
link_name = Vault.normalize_note_name(link_name)
next if link_name == file_name # 自分自身をリンクしている場合は無視する
@links_by_file_name[link_name] ||= []
@links_by_file_name[link_name] << file_path
# 同名のノートがある場合に、どこにリンクしているかを処理するのは結構難しい
# ここでは簡単のため、同名のノートがある場合はすべてのファイルをリンク先として追加する
# ただし、aliasは除外する
linked_candidates = (@notes[link_name] || []).filter{|path| path.include?(link_name + '.md') }
linked_candidates.each do |linked_candidate|
@links_by_file_path[linked_candidate] ||= []
@links_by_file_path[linked_candidate] << file_path
end
end
# [displayname](path)の場合
# pathの末尾に.mdが付いている場合、付いていない場合両方を考慮する
content.scan(/\[(.*?)\]\((?!\[\[)(.*?)(?<!\]\])\)/) do |match|
path = match[1]
# `(.+)://`から始まる場合は除外する
next if path =~ /^[a-z]+:\/\// # 外部リンク
# #、^以降の文字列を削除
path = path.sub(/#.*$/, '')
path = path.sub(/\.md$/, '') # .mdを削除
link_name = File.basename(path)
link_name = Vault.normalize_note_name(link_name)
@links_by_file_name[link_name] ||= []
@links_by_file_name[link_name] << file_path
# Obsidianはかなり賢くて、リンク先のファイルが無い場合には、その配下のファイルを探してくれる
# 今回の実装ではそこまで考慮せず、pathを信用する
path_from_vault = File.join(File.dirname(file_path), path) + '.md'
@links_by_file_path[path_from_vault] ||= []
@links_by_file_path[path_from_vault] << file_path
end
end
@links_by_file_name.each do |_file_name, file_pathes|
file_pathes.uniq!
end
@links_by_file_path.each do |_file_path, file_pathes|
file_pathes.uniq!
end
end

public

def tool_read name
name = Vault.normalize_note_name(name)
file_pathes = @notes[name]

# 名前のノートが見つからないが、nameがパスっぽい場合は、パスを修正したうえでもう一度試す
preface = ""
if file_pathes.nil? && name.include?('/')
fixed_name = File.basename(name, '.md')
file_pathes = @notes[fixed_name]
if file_pathes.nil?
# もしも名前で見つからなければ、リンクにも存在しないか確認する
link_pathes = @links_by_file_name[fixed_name]
if link_pathes.nil?
return Result.new(text: note_not_found(name), error: true)
else
# リンク先のノートが見つかった場合は、prefaceを追加する
preface = <<~EOS
Presumably a path was specified. The process was automatically renamed and processed.
EOS
return Result.new(text: list_links(fixed_name, preface), error: true)
end
else
# ノート名が見つかった場合は、prefaceを追加する
preface = <<~EOS
Presumably a path was specified. The process was automatically renamed and processed.
EOS
return Result.new(text: open_file(fixed_name, file_pathes, preface))
end
end

# 名前のノートが存在しない場合
if file_pathes.nil?
return Result.new(text: note_not_found(name), error: true) if @links_by_file_name[name].nil?
return Result.new(text: list_links(name, preface), error: true)
end

Result.new(text: open_file(name, file_pathes, preface))
end

private def note_not_found name
return <<~EOS
Note not found: #{name}
EOS
end

private def list_links name, preface
return <<~EOS
#{preface}
Note not found: #{name}
However, I found other notes linked to this note.
#{@links_by_file_name[name].shuffle.map { |file_path| "- #{File.basename(file_path, '.md')}" }.join("\n")}
EOS
end

private def open_file name, file_pathes, preface
# 複数のファイルがある場合は、---とファイル名で区切って返す
result = file_pathes.map do |file_path|
content = open(file_path) { |f| f.read.force_encoding('UTF-8') }
link_notes = if @links_by_file_path[file_path].nil?
""
else
<<~EOS
This note is linked by the following notes:
#{(@links_by_file_path[file_path] || []).shuffle.map { |file_path| "- #{File.basename(file_path, '.md')}" }.join("\n")}
EOS
end
metadata = <<~EOS
The contents of the note '#{name}' is as follows.
#{link_notes}
---
EOS
metadata + content
end.join("\n\n---\n\n")
preface + result
end

MAX_LIST_SIZE = 20
def tool_list name
name = Vault.normalize_note_name(name)
split_name = name.split(/[\s ]+/)
# 空白文字で検索された場合は失敗した旨を返す
# 仮に全ノートからランダムなMAX_LIST_SIZEを返してしまうと、LLMが誤って呼び出した場合に偽の関連性を持ってしまうため
if split_name.empty?
return Result.new(text: <<~EOS, error: true)
It cannot be listed in a blank string.
EOS
end
matched_notes = @notes.select do |note_name, _file_pathes|
split_name.map {|name_part| note_name.include?(name_part) }.all?
end
# 名前で検索したが見つからない場合
if matched_notes.empty?
has_found_link = @links_by_file_name[name] && !@links_by_file_name[name].empty?
return Result.new(text: <<~EOS, error: true) unless has_found_link
Note not found: #{name}
Search again with a substring or a string with a different notation.
EOS
return Result.new(text: <<~EOS, error: true)
Note not found: #{name}
However, I found other notes linked to this note.
#{@links_by_file_name[name].shuffle.map { |file_path| "- #{File.basename(file_path, '.md')}" }.join("\n")}
EOS
end
# マッチした名前の数が多すぎる場合は、ランダムにMAX_LIST_SIZE個選ぶ
preface = "Notes matching '#{name}' are as follows.\n"
if matched_notes.size > MAX_LIST_SIZE
matched_notes = matched_notes.to_a.sample(MAX_LIST_SIZE).to_h
preface = "Too many notes matched. I will show you only #{MAX_LIST_SIZE} of them.\n" + preface
end
# マッチした名前のリストで返す
list = matched_notes.keys.shuffle.map do |note_name|
"- #{note_name}"
end.join("\n")
Result.new(text: preface + list)
end
end

end
Loading
Loading