From 5d190f0f2a2b5bec2c72f771cd088271c9ed76ef Mon Sep 17 00:00:00 2001
From: Radu Ioan Fericean <radu@fericean.ro>
Date: Tue, 28 Jan 2014 13:00:04 +0200
Subject: [PATCH] add actions for enabling/disabling user

---
 engine/action.go | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/engine/action.go b/engine/action.go
index 9360ba7a9..459c6c30d 100644
--- a/engine/action.go
+++ b/engine/action.go
@@ -23,13 +23,14 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
-	"github.com/cgrates/cgrates/config"
-	"github.com/cgrates/cgrates/utils"
 	"net/http"
 	"net/smtp"
 	"sort"
 	"strings"
 	"time"
+
+	"github.com/cgrates/cgrates/config"
+	"github.com/cgrates/cgrates/utils"
 )
 
 /*
@@ -58,6 +59,8 @@ const (
 	DEBIT          = "*debit"
 	RESET_COUNTER  = "*reset_counter"
 	RESET_COUNTERS = "*reset_counters"
+	ENABLE_USER    = "*enable_user"
+	DISABLE_USER   = "*disable_user"
 	CALL_URL       = "*call_url"
 	CALL_URL_ASYNC = "*call_url_async"
 	MAIL_ASYNC     = "*mail_async"
@@ -90,6 +93,10 @@ func getActionFunc(typ string) (actionTypeFunc, bool) {
 		return resetCounterAction, true
 	case RESET_COUNTERS:
 		return resetCountersAction, true
+	case ENABLE_USER:
+		return enableUserAction, true
+	case DISABLE_USER:
+		return disableUserAction, true
 	case CALL_URL:
 		return callUrl, true
 	case CALL_URL_ASYNC:
@@ -181,6 +188,16 @@ func genericDebit(ub *UserBalance, a *Action) (err error) {
 	return
 }
 
+func enableUserAction(ub *UserBalance, a *Action) (err error) {
+	ub.Disabled = false
+	return
+}
+
+func disableUserAction(ub *UserBalance, a *Action) (err error) {
+	ub.Disabled = true
+	return
+}
+
 func genericReset(ub *UserBalance) {
 	for k, _ := range ub.BalanceMap {
 		ub.BalanceMap[k] = BalanceChain{&Balance{Value: 0}}