From ff435950cc5c3ea7fb16ca3e0a5eee7e1da9d372 Mon Sep 17 00:00:00 2001
From: Denys Vuika <denis.vuyka@gmail.com>
Date: Fri, 20 May 2016 11:18:39 +0100
Subject: [PATCH] #82 replace columns demo for datatable

---
 .../datatable/datatable-demo.component.html          |  5 +++++
 .../components/datatable/datatable-demo.component.ts | 12 +++++++++++-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/demo-shell-ng2/app/components/datatable/datatable-demo.component.html b/demo-shell-ng2/app/components/datatable/datatable-demo.component.html
index 021301e109..b69f30cc77 100644
--- a/demo-shell-ng2/app/components/datatable/datatable-demo.component.html
+++ b/demo-shell-ng2/app/components/datatable/datatable-demo.component.html
@@ -12,4 +12,9 @@
         (click)="replaceRows()">
         Replace rows
     </button>
+    <button
+        class="mdl-button mdl-js-button mdl-button--raised mdl-js-ripple-effect"
+        (click)="replaceColumns()">
+        Replace columns
+    </button>
 </div>
diff --git a/demo-shell-ng2/app/components/datatable/datatable-demo.component.ts b/demo-shell-ng2/app/components/datatable/datatable-demo.component.ts
index bcc75927c5..31bc57133a 100644
--- a/demo-shell-ng2/app/components/datatable/datatable-demo.component.ts
+++ b/demo-shell-ng2/app/components/datatable/datatable-demo.component.ts
@@ -21,7 +21,8 @@ import {
     ALFRESCO_DATATABLE_DIRECTIVES,
     ObjectDataTableAdapter, 
     DataSorting,
-    ObjectDataRow
+    ObjectDataRow,
+    ObjectDataColumn
 } from 'ng2-alfresco-datatable/ng2-alfresco-datatable';
 
 declare let __moduleName: string;
@@ -83,4 +84,13 @@ export class DataTableDemoComponent {
         let rows = objects.map(obj => new ObjectDataRow(obj));
         this.data.setRows(rows);
     }
+
+    replaceColumns() {
+        let schema = [
+            { type: 'text', key: 'id', sortable: true },
+            { type: 'text', key: 'name', sortable: true }
+        ];
+        let columns = schema.map(col => new ObjectDataColumn(col));
+        this.data.setColumns(columns);
+    }
 }