src/tint/lang/wgsl/ls/signature_help.cc - tint - Git at Google

 // Copyright 2024 The Dawn & Tint Authors
 //
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions are met:
 //
 // 1. Redistributions of source code must retain the above copyright notice, this
 //    list of conditions and the following disclaimer.
 //
 // 2. Redistributions in binary form must reproduce the above copyright notice,
 //    this list of conditions and the following disclaimer in the documentation
 //    and/or other materials provided with the distribution.
 //
 // 3. Neither the name of the copyright holder nor the names of its
 //    contributors may be used to endorse or promote products derived from
 //    this software without specific prior written permission.
 //
 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 // DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
 // FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 // DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 // SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 // CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 // OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 #include "src/tint/lang/wgsl/ls/server.h"

 #include "langsvr/lsp/comparators.h"
 #include "src/tint/lang/core/intrinsic/table.h"
 #include "src/tint/lang/wgsl/intrinsic/dialect.h"
 #include "src/tint/lang/wgsl/ls/utils.h"
 #include "src/tint/lang/wgsl/sem/call.h"
 #include "src/tint/utils/rtti/switch.h"
 #include "src/tint/utils/text/string_stream.h"

 namespace lsp = langsvr::lsp;

 namespace tint::wgsl::ls {

 namespace {

 /// @returns the parameter information for all the parameters of the intrinsic overload @p overload.
 std::vector<lsp::ParameterInformation> Params(const core::intrinsic::OverloadInfo& overload) {
     auto& data = wgsl::intrinsic::Dialect::kData;

     std::vector<lsp::ParameterInformation> params;
     for (size_t i = 0; i < overload.num_parameters; i++) {
         lsp::ParameterInformation param_out;
         auto& param_in = data[overload.parameters + i];
         if (param_in.usage != core::ParameterUsage::kNone) {
             param_out.label = std::string(core::ToString(param_in.usage));
         } else {
             param_out.label = "param-" + std::to_string(i);
         }
         params.push_back(std::move(param_out));
     }
     return params;
 }

 /// @returns the zero-based index of the parameter at with the cursor at @p position, for a call
 /// with the source @p call_source.
 size_t CalcParamIndex(const Source& call_source, const Source::Location& position) {
     size_t index = 0;
     int depth = 0;

     auto range = Conv(call_source.range);
     auto start = range.start;
     auto end = std::min(range.end, Conv(position));
     auto& lines = call_source.file->content.lines;

     for (auto line_idx = start.line; line_idx <= end.line; line_idx++) {
         auto& line = lines[line_idx];
         auto start_character = (line_idx == start.line) ? start.character : 0;
         auto end_character = (line_idx == end.line) ? end.character : line.size();
         auto text = line.substr(start_character, end_character - start_character);
         for (char c : text) {
             switch (c) {
                 case '(':
                 case '[':
                     depth++;
                     break;
                 case ')':
                 case ']':
                     depth--;
                     break;
                 case ',':
                     if (depth == 1) {
                         index++;
                     }
             }
         }
     }
     return index;
 }

 /// PrintOverload() emits a description of the intrinsic overload @p overload of the function with
 /// name @p intrinsic_name to @p ss.
 void PrintOverload(StyledText& ss,
                    core::intrinsic::Context& context,
                    const core::intrinsic::OverloadInfo& overload,
                    std::string_view intrinsic_name) {
     // Restore old style before returning.
     auto prev_style = ss.Style();
     TINT_DEFER(ss << prev_style);

     core::intrinsic::TemplateState templates;

     auto earliest_eval_stage = core::EvaluationStage::kConstant;

     ss << style::Code << style::Function(intrinsic_name);

     if (overload.num_explicit_templates > 0) {
         ss << "<";
         for (size_t i = 0; i < overload.num_explicit_templates; i++) {
             const auto& tmpl = context.data[overload.templates + i];
             if (i > 0) {
                 ss << ", ";
             }
             ss << style::Type(tmpl.name) << " ";
         }
         ss << ">";
     }

     ss << "(";
     for (size_t i = 0; i < overload.num_parameters; i++) {
         const auto& parameter = context.data[overload.parameters + i];
         auto* matcher_indices = context.data[parameter.matcher_indices];

         if (i > 0) {
             ss << ", ";
         }

         if (parameter.usage != core::ParameterUsage::kNone) {
             ss << style::Variable(parameter.usage, ": ");
         }
         context.Match(templates, overload, matcher_indices, earliest_eval_stage).PrintType(ss);
     }
     ss << ")";
     if (overload.return_matcher_indices.IsValid()) {
         ss << " -> ";
         auto* matcher_indices = context.data[overload.return_matcher_indices];
         context.Match(templates, overload, matcher_indices, earliest_eval_stage).PrintType(ss);
     }

     bool first = true;
     auto separator = [&] {
         ss << style::Plain(first ? " where:\n     " : "\n     ");
         first = false;
     };

     for (size_t i = 0; i < overload.num_templates; i++) {
         auto& tmpl = context.data[overload.templates + i];
         if (auto* matcher_indices = context.data[tmpl.matcher_indices]) {
             separator();

             ss << style::Type(tmpl.name) << style::Plain(" is ");
             if (tmpl.kind == core::intrinsic::TemplateInfo::Kind::kType) {
                 context.Match(templates, overload, matcher_indices, earliest_eval_stage)
                     .PrintType(ss);
             } else {
                 context.Match(templates, overload, matcher_indices, earliest_eval_stage)
                     .PrintNum(ss);
             }
         }
     }
 }

 }  // namespace

 typename lsp::TextDocumentSignatureHelpRequest::ResultType  //
 Server::Handle(const lsp::TextDocumentSignatureHelpRequest& r) {
     auto file = files_.Get(r.text_document.uri);
     if (!file) {
         return lsp::Null{};
     }

     auto& program = (*file)->program;
     auto pos = Conv(r.position);

     auto call = (*file)->NodeAt<sem::Call>(pos);
     if (!call) {
         return lsp::Null{};
     }

     lsp::SignatureHelp help;
     help.active_parameter = CalcParamIndex(call->Declaration()->source, pos);
     Switch(call->Target(),  //
            [&](const sem::BuiltinFn* target) {
                auto& data = wgsl::intrinsic::Dialect::kData;
                auto& builtins = data.builtins;
                auto& intrinsic_info = builtins[static_cast<size_t>(target->Fn())];
                std::string name{wgsl::str(target->Fn())};

                for (size_t i = 0; i < intrinsic_info.num_overloads; i++) {
                    auto& overload = data[intrinsic_info.overloads + i];

                    auto params = Params(overload);

                    auto type_mgr = core::type::Manager::Wrap(program.Types());
                    auto symbols = SymbolTable::Wrap(program.Symbols());

                    StyledText ss;
                    core::intrinsic::Context ctx{data, type_mgr, symbols};
                    PrintOverload(ss, ctx, overload, name);

                    lsp::SignatureInformation sig;
                    sig.parameters = params;
                    sig.label = ss.Plain();
                    help.signatures.push_back(sig);

                    if (&overload == &target->Overload()) {
                        help.active_signature = i;
                    }
                }
            });

     return help;
 }

 }  // namespace tint::wgsl::ls
	// Copyright 2024 The Dawn & Tint Authors
	//
	// Redistribution and use in source and binary forms, with or without
	// modification, are permitted provided that the following conditions are met:
	//
	// 1. Redistributions of source code must retain the above copyright notice, this
	// list of conditions and the following disclaimer.
	//
	// 2. Redistributions in binary form must reproduce the above copyright notice,
	// this list of conditions and the following disclaimer in the documentation
	// and/or other materials provided with the distribution.
	//
	// 3. Neither the name of the copyright holder nor the names of its
	// contributors may be used to endorse or promote products derived from
	// this software without specific prior written permission.
	//
	// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
	// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
	// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
	// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
	// FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
	// DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
	// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
	// CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
	// OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
	// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

	#include "src/tint/lang/wgsl/ls/server.h"

	#include "langsvr/lsp/comparators.h"
	#include "src/tint/lang/core/intrinsic/table.h"
	#include "src/tint/lang/wgsl/intrinsic/dialect.h"
	#include "src/tint/lang/wgsl/ls/utils.h"
	#include "src/tint/lang/wgsl/sem/call.h"
	#include "src/tint/utils/rtti/switch.h"
	#include "src/tint/utils/text/string_stream.h"

	namespace lsp = langsvr::lsp;

	namespace tint::wgsl::ls {

	namespace {

	/// @returns the parameter information for all the parameters of the intrinsic overload @p overload.
	std::vector<lsp::ParameterInformation> Params(const core::intrinsic::OverloadInfo& overload) {
	auto& data = wgsl::intrinsic::Dialect::kData;

	std::vector<lsp::ParameterInformation> params;
	for (size_t i = 0; i < overload.num_parameters; i++) {
	lsp::ParameterInformation param_out;
	auto& param_in = data[overload.parameters + i];
	if (param_in.usage != core::ParameterUsage::kNone) {
	param_out.label = std::string(core::ToString(param_in.usage));
	} else {
	param_out.label = "param-" + std::to_string(i);
	}
	params.push_back(std::move(param_out));
	}
	return params;
	}

	/// @returns the zero-based index of the parameter at with the cursor at @p position, for a call
	/// with the source @p call_source.
	size_t CalcParamIndex(const Source& call_source, const Source::Location& position) {
	size_t index = 0;
	int depth = 0;

	auto range = Conv(call_source.range);
	auto start = range.start;
	auto end = std::min(range.end, Conv(position));
	auto& lines = call_source.file->content.lines;

	for (auto line_idx = start.line; line_idx <= end.line; line_idx++) {
	auto& line = lines[line_idx];
	auto start_character = (line_idx == start.line) ? start.character : 0;
	auto end_character = (line_idx == end.line) ? end.character : line.size();
	auto text = line.substr(start_character, end_character - start_character);
	for (char c : text) {
	switch (c) {
	case '(':
	case '[':
	depth++;
	break;
	case ')':
	case ']':
	depth--;
	break;
	case ',':
	if (depth == 1) {
	index++;
	}
	}
	}
	}
	return index;
	}

	/// PrintOverload() emits a description of the intrinsic overload @p overload of the function with
	/// name @p intrinsic_name to @p ss.
	void PrintOverload(StyledText& ss,
	core::intrinsic::Context& context,
	const core::intrinsic::OverloadInfo& overload,
	std::string_view intrinsic_name) {
	// Restore old style before returning.
	auto prev_style = ss.Style();
	TINT_DEFER(ss << prev_style);

	core::intrinsic::TemplateState templates;

	auto earliest_eval_stage = core::EvaluationStage::kConstant;

	ss << style::Code << style::Function(intrinsic_name);

	if (overload.num_explicit_templates > 0) {
	ss << "<";
	for (size_t i = 0; i < overload.num_explicit_templates; i++) {
	const auto& tmpl = context.data[overload.templates + i];
	if (i > 0) {
	ss << ", ";
	}
	ss << style::Type(tmpl.name) << " ";
	}
	ss << ">";
	}

	ss << "(";
	for (size_t i = 0; i < overload.num_parameters; i++) {
	const auto& parameter = context.data[overload.parameters + i];
	auto* matcher_indices = context.data[parameter.matcher_indices];

	if (i > 0) {
	ss << ", ";
	}

	if (parameter.usage != core::ParameterUsage::kNone) {
	ss << style::Variable(parameter.usage, ": ");
	}
	context.Match(templates, overload, matcher_indices, earliest_eval_stage).PrintType(ss);
	}
	ss << ")";
	if (overload.return_matcher_indices.IsValid()) {
	ss << " -> ";
	auto* matcher_indices = context.data[overload.return_matcher_indices];
	context.Match(templates, overload, matcher_indices, earliest_eval_stage).PrintType(ss);
	}

	bool first = true;
	auto separator = [&] {
	ss << style::Plain(first ? " where:\n " : "\n ");
	first = false;
	};

	for (size_t i = 0; i < overload.num_templates; i++) {
	auto& tmpl = context.data[overload.templates + i];
	if (auto* matcher_indices = context.data[tmpl.matcher_indices]) {
	separator();

	ss << style::Type(tmpl.name) << style::Plain(" is ");
	if (tmpl.kind == core::intrinsic::TemplateInfo::Kind::kType) {
	context.Match(templates, overload, matcher_indices, earliest_eval_stage)
	.PrintType(ss);
	} else {
	context.Match(templates, overload, matcher_indices, earliest_eval_stage)
	.PrintNum(ss);
	}
	}
	}
	}

	} // namespace

	typename lsp::TextDocumentSignatureHelpRequest::ResultType //
	Server::Handle(const lsp::TextDocumentSignatureHelpRequest& r) {
	auto file = files_.Get(r.text_document.uri);
	if (!file) {
	return lsp::Null{};
	}

	auto& program = (*file)->program;
	auto pos = Conv(r.position);

	auto call = (*file)->NodeAt<sem::Call>(pos);
	if (!call) {
	return lsp::Null{};
	}

	lsp::SignatureHelp help;
	help.active_parameter = CalcParamIndex(call->Declaration()->source, pos);
	Switch(call->Target(), //
	[&](const sem::BuiltinFn* target) {
	auto& data = wgsl::intrinsic::Dialect::kData;
	auto& builtins = data.builtins;
	auto& intrinsic_info = builtins[static_cast<size_t>(target->Fn())];
	std::string name{wgsl::str(target->Fn())};

	for (size_t i = 0; i < intrinsic_info.num_overloads; i++) {
	auto& overload = data[intrinsic_info.overloads + i];

	auto params = Params(overload);

	auto type_mgr = core::type::Manager::Wrap(program.Types());
	auto symbols = SymbolTable::Wrap(program.Symbols());

	StyledText ss;
	core::intrinsic::Context ctx{data, type_mgr, symbols};
	PrintOverload(ss, ctx, overload, name);

	lsp::SignatureInformation sig;
	sig.parameters = params;
	sig.label = ss.Plain();
	help.signatures.push_back(sig);

	if (&overload == &target->Overload()) {
	help.active_signature = i;
	}
	}
	});

	return help;
	}

	} // namespace tint::wgsl::ls