/
Workspace_Registration.cs
87 lines (73 loc) · 3.15 KB
/
Workspace_Registration.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
// Copyright (c) Microsoft. All Rights Reserved. Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using System;
using System.Runtime.CompilerServices;
using Microsoft.CodeAnalysis.Text;
namespace Microsoft.CodeAnalysis
{
/* This is the static API on Workspace that lets you associate text containers with workspace instances */
public abstract partial class Workspace
{
private static readonly ConditionalWeakTable<SourceTextContainer, WorkspaceRegistration> s_bufferToWorkspaceRegistrationMap =
new ConditionalWeakTable<SourceTextContainer, WorkspaceRegistration>();
/// <summary>
/// Gets the workspace associated with the specific text container.
/// </summary>
public static bool TryGetWorkspace(SourceTextContainer textContainer, out Workspace workspace)
{
if (textContainer == null)
{
throw new ArgumentNullException(nameof(textContainer));
}
var registration = GetWorkspaceRegistration(textContainer);
workspace = registration.Workspace;
return workspace != null;
}
/// <summary>
/// Register a correspondence between a text container and a workspace.
/// </summary>
protected void RegisterText(SourceTextContainer textContainer)
{
if (textContainer == null)
{
throw new ArgumentNullException(nameof(textContainer));
}
var registration = GetWorkspaceRegistration(textContainer);
registration.SetWorkspace(this);
this.ScheduleTask(() =>
{
registration.RaiseEvents();
}, "Workspace.RegisterText");
}
/// <summary>
/// Unregister a correspondence between a text container and a workspace.
/// </summary>
protected void UnregisterText(SourceTextContainer textContainer)
{
if (textContainer == null)
{
throw new ArgumentNullException(nameof(textContainer));
}
var registration = GetWorkspaceRegistration(textContainer);
if (registration.Workspace == this)
{
registration.SetWorkspaceAndRaiseEvents(null);
}
}
private static WorkspaceRegistration CreateRegistration(SourceTextContainer container)
{
return new WorkspaceRegistration();
}
private static readonly ConditionalWeakTable<SourceTextContainer, WorkspaceRegistration>.CreateValueCallback s_createRegistration = CreateRegistration;
/// <summary>
/// Returns a <see cref="WorkspaceRegistration" /> for a given text container.
/// </summary>
public static WorkspaceRegistration GetWorkspaceRegistration(SourceTextContainer textContainer)
{
if (textContainer == null)
{
throw new ArgumentNullException(nameof(textContainer));
}
return s_bufferToWorkspaceRegistrationMap.GetValue(textContainer, s_createRegistration);
}
}
}