From 78868f9a85ee0452d5c1e9efefe0296b77ff5973 Mon Sep 17 00:00:00 2001
From: poetwang <poetwang@gmail.com>
Date: Sun, 26 Aug 2012 20:44:47 +0800
Subject: [PATCH] [api] add url for setpassword

---
 api/urls.py  | 10 +++++-----
 api/views.py | 51 +++++++++++++++++++++++++++++++++++++++++++--------
 2 files changed, 48 insertions(+), 13 deletions(-)

diff --git a/api/urls.py b/api/urls.py
index d53a9fe9e9..fcc5e2ba6c 100644
--- a/api/urls.py
+++ b/api/urls.py
@@ -8,11 +8,11 @@ urlpatterns = patterns('',
     url(r'login/$', api_login),
     url(r'^$', ReposView.as_view()),
     url(r'^repo/list/$', ReposView.as_view(), name='repos'),
-    url(r'^repo/(?P<repo_id>[^/]+)/$', RepoView.as_view(), name='repo'),
+    url(r'^repo/(?P<repo_id>[^/]+)/$', csrf_exempt(RepoView.as_view()), name='repo'),
 
-    url(r'^dir/(?P<repo_id>[^/]+)/$', RepoDirPathView.as_view(), name='repo-dir-path'),
-    url(r'^dir/(?P<repo_id>[^/]+)/(?P<dir_id>[^/]+)/$', RepoDirIdView.as_view(), name='repo-dir-id'),
-    url(r'^file/(?P<repo_id>[^/]+)/$', RepoFilePathView.as_view(), name='repo-file-path'),
-    url(r'^file/(?P<repo_id>[^/]+)/(?P<file_id>[^/]+)/$', RepoFileIdView.as_view(), name='repo-file-id'),
+    url(r'^dir/(?P<repo_id>[^/]+)/$', csrf_exempt(RepoDirPathView.as_view()), name='repo-dir-path'),
+    url(r'^dir/(?P<repo_id>[^/]+)/(?P<dir_id>[^/]+)/$', csrf_exempt(RepoDirIdView.as_view()), name='repo-dir-id'),
+    url(r'^file/(?P<repo_id>[^/]+)/$', csrf_exempt(RepoFilePathView.as_view()), name='repo-file-path'),
+    url(r'^file/(?P<repo_id>[^/]+)/(?P<file_id>[^/]+)/$', csrf_exempt(RepoFileIdView.as_view()), name='repo-file-id'),
 )
 
diff --git a/api/views.py b/api/views.py
index 1809d20a02..a8b05ccbd8 100644
--- a/api/views.py
+++ b/api/views.py
@@ -124,6 +124,23 @@ def get_dir_entrys_by_id(request, dir_id):
     response["oid"] = dir_id
     return response
 
+def set_repo_password(request, repo):
+    if not password:
+        return api_error(request, '400', 'password should not be empty')
+
+    try:
+        seafserv_threaded_rpc.set_passwd(repo_id, request.user.username, password)
+    except SearpcError, e:
+        if e.msg == 'Bad arguments':
+            return api_error(request, '400', e.msg)
+        elif e.msg == 'Repo is not encrypted':
+            return api_error(request, '400', e.msg)
+        elif e.msg == 'Incorrect password':
+            return api_error(request, '400', 'Wrong password')
+        elif e.msg == 'Internal server error':
+            return api_error(request, '500', e.msg)
+        else:
+            return api_error(request, '400', e.msg)
 
 def check_repo_access_permission(request, repo):
     if not repo:
@@ -141,8 +158,12 @@ def check_repo_access_permission(request, repo):
         except SearpcError, e:
             return api_error(request, '403', e.msg)
 
-    if repo.encrypted and not password_set:
-        return api_error(request, '403', "password needed")
+        if not password_set:
+            password = request.REQUEST['password']
+            if not password:
+                return api_error(request, '403', "password needed")
+
+            return set_repo_password(request, password)
 
 
 @csrf_exempt
@@ -226,9 +247,11 @@ class RepoView(ResponseMixin, View):
     def get_repo_info(request, repo_id):
         # check whether user can view repo
         repo = get_repo(repo_id)
-        resp = check_repo_access_permission(request, repo)
-        if resp:
-            return resp
+        if not repo:
+            return api_error(request, '404', "repo not found")
+
+        if not can_access_repo(request, repo.id):
+            return api_error(request, '403', "can not access repo")
 
         # check whether use is repo owner
         if validate_owner(request, repo_id):
@@ -242,7 +265,7 @@ class RepoView(ResponseMixin, View):
             repo.latest_modify = None
 
         # query repo infomation
-        repo_size = seafserv_threaded_rpc.server_repo_size(repo_id)
+        repo.size = seafserv_threaded_rpc.server_repo_size(repo_id)
         current_commit = get_commits(repo_id, 0, 1)[0]
         repo_json = {
             "type":"repo",
@@ -251,14 +274,26 @@ class RepoView(ResponseMixin, View):
             "name":repo.name,
             "desc":repo.desc,
             "mtime":repo.lastest_modify,
-            "password_need":password_need,
-            "size":repo_size,
+            "password_need":repo.password_need,
+            "size":repo.size,
             "root":current_commit.root_id,
             }
 
         response = Response(200, repo_json)
         return self.render(response)
 
+    @api_login_required
+    def post(self, request, repo_id):
+        resp = check_repo_access_permission(request, get_repo(repo_id))
+        if resp:
+            return resp
+        op = request.GET.get('op', 'setpassword')
+        if op == 'setpassword':
+            return HttpResponse(json.dumps("success"), status=200,
+                                content_type=json_content_type)
+
+        return HttpResponse(json.dumps("unsupported operation"), status=200,
+                            content_type=json_content_type)
 
 class RepoDirPathView(ResponseMixin, View):
     renderers = (JSONRenderer,)