// Copyright (c) 2022 Sentry. All Rights Reserved.

#pragma once

#include "HAL/CriticalSection.h"

#include "Interface/SentrySubsystemInterface.h"

class SentryScopeDesktop;
class SentryCrashReporter;

#if USE_SENTRY_NATIVE

class SentrySubsystemDesktop : public ISentrySubsystem
{
public:
	SentrySubsystemDesktop();

	virtual void InitWithSettings(const USentrySettings* settings, USentryBeforeSendHandler* beforeSendHandler, USentryTraceSampler* traceSampler) override;
	virtual void Close() override;
	virtual bool IsEnabled() override;
	virtual ESentryCrashedLastRun IsCrashedLastRun() override;
	virtual void AddBreadcrumb(TSharedPtr<ISentryBreadcrumb> breadcrumb) override;
	virtual void AddBreadcrumbWithParams(const FString& Message, const FString& Category, const FString& Type, const TMap<FString, FString>& Data, ESentryLevel Level) override;
	virtual void ClearBreadcrumbs() override;
	virtual TSharedPtr<ISentryId> CaptureMessage(const FString& message, ESentryLevel level) override;
	virtual TSharedPtr<ISentryId> CaptureMessageWithScope(const FString& message, const FSentryScopeDelegate& onScopeConfigure, ESentryLevel level) override;
	virtual TSharedPtr<ISentryId> CaptureEvent(TSharedPtr<ISentryEvent> event) override;
	virtual TSharedPtr<ISentryId> CaptureEventWithScope(TSharedPtr<ISentryEvent> event, const FSentryScopeDelegate& onScopeConfigure) override;
	virtual TSharedPtr<ISentryId> CaptureException(const FString& type, const FString& message, int32 framesToSkip) override;
	virtual TSharedPtr<ISentryId> CaptureAssertion(const FString& type, const FString& message) override;
	virtual TSharedPtr<ISentryId> CaptureEnsure(const FString& type, const FString& message) override;
	virtual void CaptureUserFeedback(TSharedPtr<ISentryUserFeedback> userFeedback) override;
	virtual void SetUser(TSharedPtr<ISentryUser> user) override;
	virtual void RemoveUser() override;
	virtual void ConfigureScope(const FSentryScopeDelegate& onConfigureScope) override;
	virtual void SetContext(const FString& key, const TMap<FString, FString>& values) override;
	virtual void SetTag(const FString& key, const FString& value) override;
	virtual void RemoveTag(const FString& key) override;
	virtual void SetLevel(ESentryLevel level) override;
	virtual void StartSession() override;
	virtual void EndSession() override;
	virtual TSharedPtr<ISentryTransaction> StartTransaction(const FString& name, const FString& operation) override;
	virtual TSharedPtr<ISentryTransaction> StartTransactionWithContext(TSharedPtr<ISentryTransactionContext> context) override;
	virtual TSharedPtr<ISentryTransaction> StartTransactionWithContextAndTimestamp(TSharedPtr<ISentryTransactionContext> context, int64 timestamp) override;
	virtual TSharedPtr<ISentryTransaction> StartTransactionWithContextAndOptions(TSharedPtr<ISentryTransactionContext> context, const TMap<FString, FString>& options) override;
	virtual TSharedPtr<ISentryTransactionContext> ContinueTrace(const FString& sentryTrace, const TArray<FString>& baggageHeaders) override;

	USentryBeforeSendHandler* GetBeforeSendHandler();

	void TryCaptureScreenshot() const;

	FString GetGpuDumpBackupPath() const;

	TSharedPtr<SentryScopeDesktop> GetCurrentScope();

private:
	FString GetHandlerPath() const;
	FString GetDatabasePath() const;
	FString GetScreenshotPath() const;

	USentryBeforeSendHandler* beforeSend;

	TSharedPtr<SentryCrashReporter> crashReporter;

	TArray<TSharedPtr<SentryScopeDesktop>> scopeStack;

	bool isEnabled;

	bool isStackTraceEnabled;
	bool isScreenshotAttachmentEnabled;

	FCriticalSection CriticalSection;

	FString databaseParentPath;
};

#endif