//! Completion for visibility specifiers.

use hir::ScopeDef;

use crate::{
    context::{CompletionContext, PathCompletionCtx, PathKind, Qualified},
    Completions,
};

pub(crate) fn complete_vis_path(
    acc: &mut Completions,
    ctx: &CompletionContext,
    path_ctx: &PathCompletionCtx,
) {
    let (qualified, &has_in_token) = match path_ctx {
        PathCompletionCtx { kind: PathKind::Vis { has_in_token }, qualified, .. } => {
            (qualified, has_in_token)
        }
        _ => return,
    };

    match qualified {
        Qualified::With {
            resolution: Some(hir::PathResolution::Def(hir::ModuleDef::Module(module))),
            is_super_chain,
            ..
        } => {
            // Try completing next child module of the path that is still a parent of the current module
            let next_towards_current =
                ctx.module.path_to_root(ctx.db).into_iter().take_while(|it| it != module).last();
            if let Some(next) = next_towards_current {
                if let Some(name) = next.name(ctx.db) {
                    cov_mark::hit!(visibility_qualified);
                    acc.add_resolution(ctx, name, ScopeDef::ModuleDef(next.into()));
                }
            }

            if *is_super_chain {
                acc.add_keyword(ctx, "super::");
            }
        }
        Qualified::Absolute | Qualified::Infer | Qualified::With { .. } => {}
        Qualified::No => {
            if !has_in_token {
                cov_mark::hit!(kw_completion_in);
                acc.add_keyword(ctx, "in");
            }
            acc.add_nameref_keywords(ctx);
        }
    }
}