1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
# frozen_string_literal: true
require 'babosa'
require_relative '../logic/helpers/hash'
require_relative './crypto'
module NanoBot
module Components
class Storage
def self.end_user(cartridge, environment)
user = ENV.fetch('NANO_BOTS_END_USER', nil)
if cartridge[:provider][:id] == 'openai' &&
!cartridge[:provider][:settings][:user].nil? &&
!cartridge[:provider][:settings][:user].to_s.strip.empty?
user = cartridge[:provider][:settings][:user]
end
candidate = environment && (
environment['NANO_BOTS_END_USER'] ||
environment[:NANO_BOTS_END_USER]
)
user = candidate if !candidate.nil? && !candidate.to_s.strip.empty?
user = if user.nil? || user.to_s.strip.empty?
'unknown'
else
user.to_s.strip
end
Crypto.encrypt(user, soft: true)
end
def self.build_path_and_ensure_state_file!(key, cartridge, environment: {})
path = [
Logic::Helpers::Hash.fetch(cartridge, %i[state directory]),
ENV.fetch('NANO_BOTS_STATE_DIRECTORY', nil)
].find do |candidate|
!candidate.nil? && !candidate.empty?
end
path = "#{user_home!.sub(%r{/$}, '')}/.local/state/nano-bots" if path.nil?
path = "#{path.sub(%r{/$}, '')}/ruby-nano-bots"
path = "#{path}/#{cartridge[:meta][:author].to_slug.normalize}"
path = "#{path}/#{cartridge[:meta][:name].to_slug.normalize}"
path = "#{path}/#{cartridge[:meta][:version].to_s.gsub('.', '-').to_slug.normalize}"
path = "#{path}/#{end_user(cartridge, environment)}"
path = "#{path}/#{Crypto.encrypt(key, soft: true)}"
path = "#{path}/state.json"
FileUtils.mkdir_p(File.dirname(path))
unless File.exist?(path)
File.write(
path,
Crypto.encrypt(JSON.generate({ key:, history: [] }))
)
end
path
end
def self.cartridges_path
[
ENV.fetch('NANO_BOTS_CARTRIDGES_DIRECTORY', nil),
"#{user_home!.sub(%r{/$}, '')}/.local/share/nano-bots/cartridges"
].compact.uniq.filter { |path| File.directory?(path) }.compact.first
end
def self.cartridge_path(path)
partial = File.join(File.dirname(path), File.basename(path, File.extname(path)))
candidates = [
path,
"#{partial}.yml",
"#{partial}.yaml"
]
unless ENV.fetch('NANO_BOTS_CARTRIDGES_DIRECTORY', nil).nil?
directory = ENV.fetch('NANO_BOTS_CARTRIDGES_DIRECTORY').sub(%r{/$}, '')
partial = File.join(File.dirname(partial), File.basename(partial, File.extname(partial)))
partial = partial.sub(%r{^\.?/}, '')
candidates << "#{directory}/#{partial}"
candidates << "#{directory}/#{partial}.yml"
candidates << "#{directory}/#{partial}.yaml"
end
directory = "#{user_home!.sub(%r{/$}, '')}/.local/share/nano-bots/cartridges"
partial = File.join(File.dirname(partial), File.basename(partial, File.extname(partial)))
partial = partial.sub(%r{^\.?/}, '')
candidates << "#{directory}/#{partial}"
candidates << "#{directory}/#{partial}.yml"
candidates << "#{directory}/#{partial}.yaml"
candidates = candidates.uniq
candidates.find do |candidate|
File.exist?(candidate) && File.file?(candidate)
end
end
def self.user_home!
[Dir.home, `echo ~`.strip, '~'].find do |candidate|
!candidate.nil? && !candidate.empty?
end
end
end
end
end
|
