Keep user assignment if possible when importing games.

1 files changed, 8 insertions, 2 deletions

diff --git a/tools/import-game.js b/tools/import-game.js
index 8413b2f..87b93ab 100755
--- a/tools/import-game.js
+++ b/tools/import-game.js
@@ -39,16 +39,22 @@ game.setup.moves = game.snaps && game.snaps.length > 0 ? game.snaps.length - 1 :
 let db = new sqlite3("db")
 
 let insert_game = db.prepare("insert into games(status,owner_id,title_id,scenario,options,player_count,active,moves,notice) values (1,1,:title_id,:scenario,:options,:player_count,:active,:moves,:notice) returning game_id").pluck()
-let insert_player = db.prepare("insert into players(game_id,role,user_id) values (?,?,1)")
+let insert_player = db.prepare("insert into players(game_id,role,user_id) values (?,?,?)")
 let insert_state = db.prepare("insert into game_state(game_id,state) values (?,?)")
 
+let select_user = db.prepare("select user_id from users where name=?").pluck()
+
 db.exec("begin")
 
 game.setup.options = JSON.stringify(game.setup.options)
 
+function find_user(name) {
+	return select_user.get(name) || 1
+}
+
 let game_id = insert_game.get(game.setup)
 for (let p of game.players)
-	insert_player.run(game_id, p.role)
+	insert_player.run(game_id, p.role, find_user(p.name))
 insert_state.run(game_id, JSON.stringify(game.state))
 
 if (game.replay) {