// slang-artifact-helper.h
#ifndef SLANG_ARTIFACT_HELPER_H
#define SLANG_ARTIFACT_HELPER_H

#include "../core/slang-com-object.h"
#include "slang-artifact-representation.h"
#include "slang-artifact.h"

namespace Slang
{

class IArtifactHelper : public ICastable
{
    SLANG_COM_INTERFACE(0x882b25d7, 0xe300, 0x4b20, {0xbe, 0xb, 0x26, 0xd2, 0x52, 0x3e, 0x70, 0x20})

    /// Create an artifact
    virtual SLANG_NO_THROW SlangResult SLANG_MCALL
    createArtifact(const ArtifactDesc& desc, const char* name, IArtifact** outArtifact) = 0;

    /// Get the parent to a kind
    virtual SLANG_NO_THROW ArtifactKind SLANG_MCALL getKindParent(ArtifactKind kind) = 0;
    /// Get the name of a kind
    virtual SLANG_NO_THROW UnownedStringSlice SLANG_MCALL getKindName(ArtifactKind kind) = 0;
    /// Returns true if kind is derived from base
    virtual SLANG_NO_THROW bool SLANG_MCALL
    isKindDerivedFrom(ArtifactKind kind, ArtifactKind base) = 0;

    /// Get the parent payload for payload
    virtual SLANG_NO_THROW ArtifactPayload SLANG_MCALL
    getPayloadParent(ArtifactPayload payload) = 0;
    /// Get the payload name text
    virtual SLANG_NO_THROW UnownedStringSlice SLANG_MCALL
    getPayloadName(ArtifactPayload payload) = 0;
    /// Returns true if payload is derived from base
    virtual SLANG_NO_THROW bool SLANG_MCALL
    isPayloadDerivedFrom(ArtifactPayload payload, ArtifactPayload base) = 0;

    /// Get the parent type of a style
    virtual SLANG_NO_THROW ArtifactStyle SLANG_MCALL getStyleParent(ArtifactStyle style) = 0;
    /// Get text name for a style
    virtual SLANG_NO_THROW UnownedStringSlice SLANG_MCALL getStyleName(ArtifactStyle style) = 0;
    /// Returns true if style is derived from base
    virtual SLANG_NO_THROW bool SLANG_MCALL
    isStyleDerivedFrom(ArtifactStyle style, ArtifactStyle base) = 0;

    /// Create a lock file, the path of which can be used to generate other temporary files
    virtual SLANG_NO_THROW SlangResult SLANG_MCALL
    createLockFile(const CharSlice& nameBase, IOSFileArtifactRepresentation** outLockFile) = 0;

    /// Given a desc and a basePath returns a suitable name
    virtual SLANG_NO_THROW SlangResult SLANG_MCALL
    calcArtifactDescPath(const ArtifactDesc& desc, const char* basePath, ISlangBlob** outPath) = 0;

    /// Given an artifact and a basePath returns a suitable name
    virtual SLANG_NO_THROW SlangResult SLANG_MCALL
    calcArtifactPath(IArtifact*, const char* basePath, ISlangBlob** outPath) = 0;

    /// Given a compile target return the equivalent desc
    virtual SLANG_NO_THROW ArtifactDesc SLANG_MCALL
    makeDescForCompileTarget(SlangCompileTarget target) = 0;

    /// Given an interface returns as a castable interface. This might just cast unk into ICastable,
    /// or wrap it such that it uses the castable interface
    virtual SLANG_NO_THROW void SLANG_MCALL
    getCastable(ISlangUnknown* unk, ICastable** outCastable) = 0;

    /// Create a file rep
    virtual SLANG_NO_THROW SlangResult SLANG_MCALL createOSFileArtifactRepresentation(
        IOSFileArtifactRepresentation::Kind kind,
        const CharSlice& path,
        IOSFileArtifactRepresentation* lockFile,
        IOSFileArtifactRepresentation** outRep) = 0;

    virtual SLANG_NO_THROW SlangResult SLANG_MCALL createExtFileArtifactRepresentation(
        const CharSlice& path,
        ISlangFileSystemExt* system,
        IExtFileArtifactRepresentation** outRep) = 0;

    virtual SLANG_NO_THROW SlangResult SLANG_MCALL createOSFileArtifact(
        const ArtifactDesc& desc,
        const CharSlice& slice,
        IArtifact** outArtifact) = 0;
};

class DefaultArtifactHelper : public IArtifactHelper
{
public:
    // ISlangUnknown
    SLANG_NO_THROW uint32_t SLANG_MCALL addRef() SLANG_OVERRIDE { return 1; }
    SLANG_NO_THROW uint32_t SLANG_MCALL release() SLANG_OVERRIDE { return 1; }
    SLANG_NO_THROW SlangResult SLANG_MCALL queryInterface(SlangUUID const& uuid, void** outObject)
        SLANG_OVERRIDE;

    // ICastable
    SLANG_NO_THROW void* SLANG_MCALL castAs(const Guid& guid) SLANG_OVERRIDE;

    // IArtifactHelper
    virtual SLANG_NO_THROW SlangResult SLANG_MCALL createArtifact(
        const ArtifactDesc& desc,
        const char* name,
        IArtifact** outArtifact) SLANG_OVERRIDE;

    virtual SLANG_NO_THROW ArtifactKind SLANG_MCALL getKindParent(ArtifactKind kind) SLANG_OVERRIDE;
    virtual SLANG_NO_THROW UnownedStringSlice SLANG_MCALL getKindName(ArtifactKind kind)
        SLANG_OVERRIDE;
    virtual SLANG_NO_THROW bool SLANG_MCALL isKindDerivedFrom(ArtifactKind kind, ArtifactKind base)
        SLANG_OVERRIDE;

    virtual SLANG_NO_THROW ArtifactPayload SLANG_MCALL getPayloadParent(ArtifactPayload payload)
        SLANG_OVERRIDE;
    virtual SLANG_NO_THROW UnownedStringSlice SLANG_MCALL getPayloadName(ArtifactPayload payload)
        SLANG_OVERRIDE;
    virtual SLANG_NO_THROW bool SLANG_MCALL
    isPayloadDerivedFrom(ArtifactPayload payload, ArtifactPayload base) SLANG_OVERRIDE;

    virtual SLANG_NO_THROW ArtifactStyle SLANG_MCALL getStyleParent(ArtifactStyle style)
        SLANG_OVERRIDE;
    virtual SLANG_NO_THROW UnownedStringSlice SLANG_MCALL getStyleName(ArtifactStyle style)
        SLANG_OVERRIDE;
    virtual SLANG_NO_THROW bool SLANG_MCALL
    isStyleDerivedFrom(ArtifactStyle style, ArtifactStyle base) SLANG_OVERRIDE;

    virtual SLANG_NO_THROW SlangResult SLANG_MCALL createLockFile(
        const CharSlice& nameBase,
        IOSFileArtifactRepresentation** outLockFile) SLANG_OVERRIDE;

    virtual SLANG_NO_THROW SlangResult SLANG_MCALL calcArtifactDescPath(
        const ArtifactDesc& desc,
        const char* basePath,
        ISlangBlob** outPath) SLANG_OVERRIDE;

    virtual SLANG_NO_THROW SlangResult SLANG_MCALL
    calcArtifactPath(IArtifact*, const char* basePath, ISlangBlob** outPath) SLANG_OVERRIDE;

    virtual SLANG_NO_THROW ArtifactDesc SLANG_MCALL
    makeDescForCompileTarget(SlangCompileTarget target) SLANG_OVERRIDE;

    virtual SLANG_NO_THROW void SLANG_MCALL getCastable(ISlangUnknown* unk, ICastable** outCastable)
        SLANG_OVERRIDE;

    virtual SLANG_NO_THROW SlangResult SLANG_MCALL createOSFileArtifactRepresentation(
        IOSFileArtifactRepresentation::Kind kind,
        const CharSlice& path,
        IOSFileArtifactRepresentation* lockFile,
        IOSFileArtifactRepresentation** outRep) SLANG_OVERRIDE;

    virtual SLANG_NO_THROW SlangResult SLANG_MCALL createExtFileArtifactRepresentation(
        const CharSlice& path,
        ISlangFileSystemExt* system,
        IExtFileArtifactRepresentation** outRep) SLANG_OVERRIDE;

    virtual SLANG_NO_THROW SlangResult SLANG_MCALL createOSFileArtifact(
        const ArtifactDesc& desc,
        const CharSlice& slice,
        IArtifact** outArtifact) SLANG_OVERRIDE;

    static IArtifactHelper* getSingleton() { return &g_singleton; }

protected:
    void* getInterface(const Guid& guid);
    void* getObject(const Guid& guid);

    static DefaultArtifactHelper g_singleton;
};

} // namespace Slang

#endif