Implement correct DataSourceID lookup
[mono.git] / mcs / class / System.Web / System.Web.UI.WebControls / HierarchicalDataBoundControl.cs
1 //
2 // System.Web.UI.WebControls.HierarchicalDataBoundControl.cs
3 //
4 // Authors:
5 //      Lluis Sanchez Gual (lluis@novell.com)
6 //
7 // (C) 2004 Novell, Inc (http://www.novell.com)
8 //
9
10 //
11 // Permission is hereby granted, free of charge, to any person obtaining
12 // a copy of this software and associated documentation files (the
13 // "Software"), to deal in the Software without restriction, including
14 // without limitation the rights to use, copy, modify, merge, publish,
15 // distribute, sublicense, and/or sell copies of the Software, and to
16 // permit persons to whom the Software is furnished to do so, subject to
17 // the following conditions:
18 // 
19 // The above copyright notice and this permission notice shall be
20 // included in all copies or substantial portions of the Software.
21 // 
22 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
23 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
24 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
25 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
26 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
27 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
28 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
29 //
30
31 #if NET_2_0
32 using System.Collections;
33 using System.ComponentModel;
34
35 namespace System.Web.UI.WebControls
36 {
37         [DesignerAttribute ("System.Web.UI.Design.WebControls.HierarchicalDataBoundControlDesigner, " + Consts.AssemblySystem_Design, "System.ComponentModel.Design.IDesigner")]
38         public abstract class HierarchicalDataBoundControl : BaseDataBoundControl
39         {
40                 [IDReferencePropertyAttribute (typeof(HierarchicalDataSourceControl))]
41                 public override string DataSourceID {
42                         get {
43                                 object o = ViewState ["DataSourceID"];
44                                 if (o != null)
45                                         return (string)o;
46                                 
47                                 return String.Empty;
48                         }
49                         set {
50                                 if (Initialized)
51                                         RequiresDataBinding = true;
52                                 
53                                 ViewState ["DataSourceID"] = value;
54                         }
55                 }
56                 
57                 protected virtual HierarchicalDataSourceView GetData (string viewPath)
58                 {
59                         if (DataSource != null && DataSourceID != "")
60                                 throw new HttpException ();
61                         
62                         IHierarchicalDataSource ds = GetDataSource ();
63                         if (ds != null)
64                                 return ds.GetHierarchicalView (viewPath);
65                         else
66                                 return null; 
67                 }
68                 
69                 protected virtual IHierarchicalDataSource GetDataSource ()
70                 {
71                         if (DataSourceID != "")
72                                 return BaseDataBoundControl.FindDataSource (this, DataSourceID);
73                         
74                         return DataSource as IHierarchicalDataSource;
75                 }
76
77                 bool IsDataBound {
78                         get {
79                                 return ViewState.GetBool ("DataBound", false);
80                         }
81                         set {
82                                 ViewState ["DataBound"] = value;
83                         }
84                 }
85
86                 protected void MarkAsDataBound ()
87                 {
88                         IsDataBound = true;
89                 }
90                 
91                 protected override void OnDataPropertyChanged ()
92                 {
93                         RequiresDataBinding = true;
94                 }
95                 
96                 protected virtual void OnDataSourceChanged (object sender, EventArgs e)
97                 {
98                         RequiresDataBinding = true;
99                 }
100
101                 protected internal override void OnLoad (EventArgs e)
102                 {
103                         if (!Initialized) {
104                                 Initialize ();
105                                 ConfirmInitState ();
106                         }
107                         
108                         base.OnLoad(e);
109                 }
110
111                 private void Initialize ()
112                 {
113                         if (!Page.IsPostBack || (IsViewStateEnabled && !IsDataBound))
114                                 RequiresDataBinding = true;
115
116                         IHierarchicalDataSource ds = GetDataSource ();
117                         if (ds != null && DataSourceID != "")
118                                 ds.DataSourceChanged += new EventHandler (OnDataSourceChanged);
119                 }
120
121                 protected override void OnPagePreLoad (object sender, EventArgs e)
122                 {
123                         base.OnPagePreLoad (sender, e);
124                         
125                         Initialize ();
126                 }
127                 
128                 protected internal virtual void PerformDataBinding ()
129                 {
130                 }
131                 
132                 protected override void PerformSelect ()
133                 {
134                         OnDataBinding (EventArgs.Empty);
135                         PerformDataBinding ();
136                         // The PerformDataBinding method has completed.
137                         RequiresDataBinding = false;
138                         MarkAsDataBound ();
139                         OnDataBound (EventArgs.Empty);
140                 }
141                 
142                 protected override void ValidateDataSource (object dataSource)
143                 {
144                         if (dataSource == null || dataSource is IHierarchicalDataSource || dataSource is IHierarchicalEnumerable)
145                                 return;
146                         throw new InvalidOperationException ("Invalid data source");
147                 }
148         }
149 }
150 #endif
151
152