blob: ab2fb687ffa4a46531416b48073ae32174c6249c [file] [log] [blame]
// 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.
#ifndef SRC_TINT_LANG_WGSL_LS_SERVER_H_
#define SRC_TINT_LANG_WGSL_LS_SERVER_H_
#include <memory>
#include <string>
#include <utility>
#include "langsvr/lsp/lsp.h"
#include "langsvr/session.h"
#include "src/tint/lang/wgsl/ls/file.h"
#include "src/tint/utils/containers/hashmap.h"
#include "src/tint/utils/text/string_stream.h"
namespace tint::wgsl::ls {
/// The language server state object.
class Server {
public:
/// Constructor
/// @param session the LSP session.
explicit Server(langsvr::Session& session);
/// Destructor
~Server();
/// @returns true if the server has been requested to shut down.
bool ShuttingDown() const { return shutting_down_; }
private:
////////////////////////////////////////////////////////////////////////////
// Requests
////////////////////////////////////////////////////////////////////////////
/// Handler for langsvr::lsp::TextDocumentCompletionRequest
typename langsvr::lsp::TextDocumentCompletionRequest::ResultType //
Handle(const langsvr::lsp::TextDocumentCompletionRequest&);
/// Handler for langsvr::lsp::TextDocumentDefinitionRequest
typename langsvr::lsp::TextDocumentDefinitionRequest::ResultType //
Handle(const langsvr::lsp::TextDocumentDefinitionRequest&);
/// Handler for langsvr::lsp::TextDocumentDocumentSymbolRequest
typename langsvr::lsp::TextDocumentDocumentSymbolRequest::ResultType //
Handle(const langsvr::lsp::TextDocumentDocumentSymbolRequest& r);
/// Handler for langsvr::lsp::TextDocumentHoverRequest
typename langsvr::lsp::TextDocumentHoverRequest::ResultType //
Handle(const langsvr::lsp::TextDocumentHoverRequest&);
/// Handler for langsvr::lsp::TextDocumentInlayHintRequest
typename langsvr::lsp::TextDocumentInlayHintRequest::ResultType //
Handle(const langsvr::lsp::TextDocumentInlayHintRequest&);
/// Handler for langsvr::lsp::TextDocumentPrepareRenameRequest
typename langsvr::lsp::TextDocumentPrepareRenameRequest::ResultType //
Handle(const langsvr::lsp::TextDocumentPrepareRenameRequest&);
/// Handler for langsvr::lsp::TextDocumentReferencesRequest
typename langsvr::lsp::TextDocumentReferencesRequest::ResultType //
Handle(const langsvr::lsp::TextDocumentReferencesRequest&);
/// Handler for langsvr::lsp::TextDocumentRenameRequest
typename langsvr::lsp::TextDocumentRenameRequest::ResultType //
Handle(const langsvr::lsp::TextDocumentRenameRequest&);
////////////////////////////////////////////////////////////////////////////
// Notifications
////////////////////////////////////////////////////////////////////////////
/// Handler for langsvr::lsp::CancelRequestNotification
langsvr::Result<langsvr::SuccessType> //
Handle(const langsvr::lsp::CancelRequestNotification&);
/// Handler for langsvr::lsp::InitializedNotification
langsvr::Result<langsvr::SuccessType> //
Handle(const langsvr::lsp::InitializedNotification&);
/// Handler for langsvr::lsp::SetTraceNotification
langsvr::Result<langsvr::SuccessType> //
Handle(const langsvr::lsp::SetTraceNotification&);
/// Handler for langsvr::lsp::TextDocumentSignatureHelpRequest
typename langsvr::lsp::TextDocumentSignatureHelpRequest::ResultType //
Handle(const langsvr::lsp::TextDocumentSignatureHelpRequest&);
/// Handler for langsvr::lsp::TextDocumentDidOpenNotification
langsvr::Result<langsvr::SuccessType> //
Handle(const langsvr::lsp::TextDocumentDidOpenNotification&);
/// Handler for langsvr::lsp::TextDocumentDidCloseNotification
langsvr::Result<langsvr::SuccessType> //
Handle(const langsvr::lsp::TextDocumentDidCloseNotification&);
/// Handler for langsvr::lsp::TextDocumentDidChangeNotification
langsvr::Result<langsvr::SuccessType> //
Handle(const langsvr::lsp::TextDocumentDidChangeNotification&);
/// Handler for langsvr::lsp::TextDocumentSemanticTokensFullRequest
typename langsvr::lsp::TextDocumentSemanticTokensFullRequest::ResultType //
Handle(const langsvr::lsp::TextDocumentSemanticTokensFullRequest&);
/// Handler for langsvr::lsp::WorkspaceDidChangeConfigurationNotification
langsvr::Result<langsvr::SuccessType> //
Handle(const langsvr::lsp::WorkspaceDidChangeConfigurationNotification&);
/// Handler for langsvr::lsp::WorkspaceDidChangeWatchedFilesNotification
langsvr::Result<langsvr::SuccessType> //
Handle(const langsvr::lsp::WorkspaceDidChangeWatchedFilesNotification&);
/// Publishes the tint::Program diagnostics to the server via a
/// TextDocumentPublishDiagnosticsNotification.
langsvr::Result<langsvr::SuccessType> //
PublishDiagnostics(File& file);
/// Logger is a string-stream like utility for logging to the client.
/// Append message content with '<<'. The message is sent when the logger is destructed.
struct Logger {
~Logger();
/// @brief Appends @p value to the log message
/// @return this logger
template <typename T>
Logger& operator<<(T&& value) {
msg << std::forward<T>(value);
return *this;
}
langsvr::Session& session;
langsvr::lsp::MessageType type;
StringStream msg{};
};
/// Log constructs a new Logger to send a log message to the client.
Logger Log() { return Logger{session_, langsvr::lsp::MessageType::kLog}; }
/// Error constructs a new Logger to send a log message to the client.
Logger Error() { return Logger{session_, langsvr::lsp::MessageType::kError}; }
/// The LSP session.
langsvr::Session& session_;
/// Map of URI to File.
Hashmap<std::string, std::shared_ptr<File>, 8> files_;
/// True if the server has been asked to shutdown.
bool shutting_down_ = false;
};
} // namespace tint::wgsl::ls
#endif // SRC_TINT_LANG_WGSL_LS_SERVER_H_