Skip to content
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

Jump to bottom

Support YARD notation #724

Merged
merged 10 commits into from
Jan 24, 2020
Merged

Support YARD notation #724

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
84e0f49
start to work on format on yard
tvallois Jan 14, 2020
8857ace
add json and jsonb to ruby classes
tvallois Jan 15, 2020
bb4d4f0
map_col_type_to_ruby_classes should not be private
tvallois Jan 17, 2020
32c239b
add YARD Boolean type instead of FalseClass and TrueClass + specs
tvallois Jan 18, 2020
5720c4e
adding array support for yard
tvallois Jan 18, 2020
e3b9cfe
make the ci pass
tvallois Jan 18, 2020
55ce931
Merge branch 'develop' of git://github.com/ctran/annotate_models into…
tvallois Jan 18, 2020
cf303fd
set map_col_type_to_ruby_classes to private
tvallois Jan 22, 2020
3df1e1c
Merge branch 'develop' into develop
tvallois Jan 23, 2020
4f489b8
remove whitespace
tvallois Jan 23, 2020
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
1 change: 1 addition & 0 deletions Gemfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@ end

group :development, :test do
gem 'byebug'
gem 'pry-byebug'
gem 'guard-rspec', require: false
gem 'rspec', require: false

Expand Down
10 changes: 3 additions & 7 deletions lib/annotate/annotate_models.rb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@ def get_schema_info(klass, header, options = {})
if col_type == 'decimal'
col_type << "(#{col.precision}, #{col.scale})"
elsif !%w[spatial geometry geography].include?(col_type)
if col.limit
if col.limit && !options[:format_yard]
if col.limit.is_a? Array
attrs << "(#{col.limit.join(', ')})"
else
Expand Down Expand Up @@ -306,12 +306,7 @@ def get_schema_info(klass, header, options = {})
info << sprintf("# %-#{max_size}.#{max_size}s<tt>%s</tt>", "*#{col_name}*::", attrs.unshift(col_type).join(", ")).rstrip + "\n"
elsif options[:format_yard]
info << sprintf("# @!attribute #{col_name}") + "\n"
if col_type == 'boolean'
info << sprintf("# @return [TrueClass]") + "\n"
info << sprintf("# @return [FalseClass]") + "\n"
else
info << sprintf("# @return [#{map_col_type_to_ruby_classes(col_type)}]") + "\n"
end
info << sprintf("# @return [#{map_col_type_to_ruby_classes(col_type)}]") + "\n"
elsif options[:format_markdown]
name_remainder = max_size - col_name.length - non_ascii_length(col_name)
type_remainder = (md_type_allowance - 2) - col_type.length
Expand Down Expand Up @@ -913,6 +908,7 @@ def map_col_type_to_ruby_classes(col_type)
when 'date' then Date.to_s
when 'text', 'string', 'binary', 'inet', 'uuid' then String.to_s
when 'json', 'jsonb' then Hash.to_s
when 'boolean' then 'Boolean'
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Does this change any current behavior?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Seems not. Boolean is a class that YARD understand quite well. It's better than TrueClass and FalseClass

Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

If you are referring to just the boolean case, Boolean is the string for what YARD expects a boolean type to be in the comments.

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Gotcha

end
end
private
Expand Down
20 changes: 20 additions & 0 deletions spec/lib/annotate/annotate_models_spec.rb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -555,6 +555,26 @@ def mock_column(name, type, options = {})
EOS
end

it 'should get schema info as YARD' do
klass = mock_class(:users,
:id,
[
mock_column(:id, :integer),
mock_column(:name, :string, limit: 50),
])
expect(AnnotateModels.get_schema_info(klass, AnnotateModels::PREFIX, format_yard: true)).to eql(<<-EOS)
# #{AnnotateModels::PREFIX}
#
# Table name: users
#
# @!attribute id
# @return [Integer]
# @!attribute name
# @return [String]
#
EOS
end

it 'should get schema info as Markdown' do
klass = mock_class(:users,
:id,
Expand Down