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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
|
# File 'lib/aptible/cli/subcommands/logs.rb', line 9
def self.included(thor)
thor.class_eval do
include Helpers::Operation
include Helpers::AppOrDatabase
include Helpers::DateHelpers
desc 'logs [--app APP | --database DATABASE]',
'Follows logs from a running app or database'
app_or_database_options
def logs
resource = ensure_app_or_database(options)
unless resource.status == 'provisioned'
raise Thor::Error, 'Unable to retrieve logs. ' \
"Have you deployed #{resource.handle} yet?"
end
op = resource.create_operation!(type: 'logs', status: 'succeeded')
ENV['ACCESS_TOKEN'] = fetch_token
exit_with_ssh_portal(op, '-o', 'SendEnv=ACCESS_TOKEN', '-T')
end
desc 'logs_from_archive --bucket NAME --region REGION ' \
'--stack NAME [ --decryption-keys ONE [OR MORE] ] ' \
'[ --download-location LOCATION ] ' \
'[ [ --string-matches ONE [OR MORE] ] ' \
'| [ --app-id ID | --database-id ID | --endpoint-id ID | ' \
'--container-id ID ] ' \
'[ --start-date YYYY-MM-DD --end-date YYYY-MM-DD ] ]',
'Retrieves container logs from an S3 archive in your own ' \
'AWS account. You must provide your AWS credentials via ' \
'the environment variables AWS_ACCESS_KEY_ID and ' \
'AWS_SECRET_ACCESS_KEY'
option :region,
desc: 'The AWS region your S3 bucket resides in',
type: :string, required: true
option :bucket,
desc: 'The name of your S3 bucket',
type: :string, required: true
option :stack,
desc: 'The name of the Stack to download logs from',
type: :string, required: true
option :decryption_keys,
desc: 'The Aptible-provided keys for decryption. ' \
'(Space separated if multiple)',
type: :array
option :string_matches,
desc: 'The strings to match in log file names.' \
'(Space separated if multiple)',
type: :array
option :app_id,
desc: 'The Application ID to download logs for.',
type: :numeric
option :database_id,
desc: 'The Database ID to download logs for.',
type: :numeric
option :endpoint_id,
desc: 'The Endpoint ID to download logs for.',
type: :numeric
option :container_id,
desc: 'The container ID to download logs for'
option :start_date,
desc: 'Get logs starting from this (UTC) date ' \
'(format: YYYY-MM-DD)',
type: :string
option :end_date,
desc: 'Get logs before this (UTC) date (format: YYYY-MM-DD)',
type: :string
option :download_location,
desc: 'The local path place downloaded log files. ' \
'If you do not set this option, the file names ' \
'will be shown, but not downloaded.',
type: :string
def logs_from_archive
ensure_aws_creds
validate_log_search_options(options)
id_options = [
options[:app_id],
options[:database_id],
options[:endpoint_id],
options[:container_id]
]
date_options = [options[:start_date], options[:end_date]]
r_type = 'apps' if options[:app_id]
r_type = 'databases' if options[:database_id]
r_type = 'proxy' if options[:endpoint_id]
if date_options.any?
start_date = utc_date(options[:start_date])
end_date = utc_date(options[:end_date])
if end_date < start_date
raise Thor::Error, 'End date must be after start date.'
end
time_range = [start_date, end_date]
CLI.logger.info "Searching from #{start_date} to #{end_date}"
else
time_range = nil
end
if options[:string_matches]
files = find_s3_files_by_string_match(
options[:region],
options[:bucket],
options[:stack],
options[:string_matches]
)
elsif id_options.any?
if options[:container_id]
search_attrs = { container_id: options[:container_id] }
else
search_attrs = { type: r_type, id: id_options.compact.first }
end
files = find_s3_files_by_attrs(
options[:region],
options[:bucket],
options[:stack],
search_attrs,
time_range
)
end
unless files.any?
raise Thor::Error, 'No files found that matched all criteria'
end
CLI.logger.info "Found #{files.count} matching files..."
if options[:download_location]
files.each do |file|
shasum = info_from_path(file)[:shasum]
decrypt_and_translate_s3_file(
file,
encryption_key(shasum, options[:decryption_keys]),
options[:region],
options[:bucket],
options[:download_location]
)
end
else
files.each do |file|
CLI.logger.info file.split('/').drop(4).join('/')
end
m = 'No files were downloaded. Please provide a location ' \
'with --download-location to download the files.'
CLI.logger.warn m
end
end
end
end
|