From 2c81988ed3c0d077ba00f4aac6445e9f648f21d2 Mon Sep 17 00:00:00 2001
From: robk-tahoe <robk-tahoe@allmydata.com>
Date: Wed, 5 Mar 2008 16:34:18 -0700
Subject: [PATCH] confwiz: add nickname determination

the confwiz now uses socket.gethostname() if a 'nickname' file doesn't already
exist, and passes that nickname into the 'record_install' method on the backend,
so that the moniker can be recorded in the system table.
---
 src/allmydata/gui/confwiz.py | 37 +++++++++++++++++++++++++++++++++---
 1 file changed, 34 insertions(+), 3 deletions(-)

diff --git a/src/allmydata/gui/confwiz.py b/src/allmydata/gui/confwiz.py
index 75fac188..ec36caaa 100644
--- a/src/allmydata/gui/confwiz.py
+++ b/src/allmydata/gui/confwiz.py
@@ -9,6 +9,7 @@ WINFUSESVC_NAME = 'Allmydata SMB'
 
 import os
 import re
+import socket
 import sys
 #import time
 import traceback
@@ -63,12 +64,13 @@ def create_account(url, user, passwd, subscribe):
     result_code = post(url, args)
     return result_code
 
-def record_install(url, user, passwd, nodeid):
+def record_install(url, user, passwd, nodeid, nickname):
     args = {
         'action': 'record_install',
         'email': unicode_to_utf8(user),
         'passwd': unicode_to_utf8(passwd),
         'nodeid': nodeid,
+        'moniker': nickname,
         }
     result_code = post(url, args)
     return result_code
@@ -116,6 +118,31 @@ def get_nodeid():
     tub = foolscap.Tub(certFile=certfile)
     return tub.getTubID()
 
+def get_nickname():
+    nick = None
+    nnfile = os.path.join(get_basedir(), 'nickname')
+    if os.path.exists(nnfile):
+        try:
+            fh = file(nnfile, 'rb')
+            nick = fh.read().strip()
+            fh.close()
+        except:
+            DisplayTraceback('Failed to read existing nickname file %s' % (nnfile,))
+    if not nick:
+        nick = socket.gethostname()
+    return nick
+
+def maybe_write_nickname(nickname):
+    nnfile = os.path.join(get_basedir(), 'nickname')
+    try:
+        if not os.path.exists(nnfile):
+            fh = file(nnfile, 'wb')
+            fh.write(nickname)
+            fh.write('\n')
+            fh.close()
+    except:
+        DisplayTraceback('Failed to write nickname file %s' % (nnfile,))
+
 def configure(backend, user, passwd):
     _config_re = re.compile('^([^:]*): (.*)$')
     config = get_config(backend, user, passwd)
@@ -314,12 +341,14 @@ class LoginPanel(wx.Panel):
             return
 
         nodeid = get_nodeid()
-        ret = record_install(backend, user, passwd, nodeid)
+        nickname = get_nickname()
+        ret = record_install(backend, user, passwd, nodeid, nickname)
         if ret != 'ok':
             wx.MessageBox('Error "%s" recording this system (%s)' % (ret, nodeid), 'Error')
 
         configure(backend, user, passwd)
         maybe_start_services()
+        maybe_write_nickname(nickname)
 
         self.app.open_welcome_page()
 
@@ -459,12 +488,14 @@ class RegisterPanel(wx.Panel):
             return
 
         nodeid = get_nodeid()
-        ret = record_install(backend, user, passwd, nodeid)
+        nickname = get_nickname()
+        ret = record_install(backend, user, passwd, nodeid, nickname)
         if ret != 'ok':
             wx.MessageBox('Error "%s" recording this system (%s)' % (ret, nodeid), 'Error')
 
         configure(backend, user, passwd)
         maybe_start_services()
+        maybe_write_nickname(nickname)
 
         self.app.open_welcome_page()
 
-- 
2.45.2