1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00

vdoc: remove duplicated code

This commit is contained in:
Alexey 2020-02-27 00:44:29 +03:00 committed by GitHub
parent ca9fa6407f
commit 394d64bfd0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -19,6 +19,8 @@ mut:
stmts []ast.Stmt // all module statements from all files
}
type FilterFn fn(node ast.FnDecl) bool
pub fn doc(mod string, table &table.Table) string {
mut d := Doc{
out: strings.new_builder(1000)
@ -64,39 +66,42 @@ fn (d &Doc) get_fn_node(f ast.FnDecl) string {
}
fn (d mut Doc) print_fns() {
mut fn_names := []string
for stmt in d.stmts {
match stmt {
ast.FnDecl {
if it.is_pub && !it.is_method {
name := d.get_fn_node(it)
fn_names << name
fn_signatures := d.get_fn_signatures(is_pub_function)
d.write_fn_signatures(fn_signatures)
}
fn (d mut Doc) print_methods() {
fn_signatures := d.get_fn_signatures(is_pub_method)
d.write_fn_signatures(fn_signatures)
}
else {}
}
}
fn_names.sort()
for s in fn_names {
[inline]
fn (d mut Doc) write_fn_signatures(fn_signatures []string) {
for s in fn_signatures {
d.out.writeln(s)
}
}
fn (d mut Doc) print_methods() {
mut fn_names := []string
fn (d Doc) get_fn_signatures(filter_fn FilterFn) []string {
mut fn_signatures := []string
for stmt in d.stmts {
match stmt {
ast.FnDecl {
if it.is_pub && it.is_method {
name := d.get_fn_node(it)
fn_names << name
if filter_fn(it) {
fn_signatures << d.get_fn_node(it)
}
}
else {}
}
}
fn_names.sort()
for s in fn_names {
d.out.writeln(s)
fn_signatures.sort()
return fn_signatures
}
fn is_pub_method(node ast.FnDecl) bool {
return node.is_pub && node.is_method
}
fn is_pub_function(node ast.FnDecl) bool {
return node.is_pub && !node.is_method
}