1 # This file is part of NIT ( http://www.nitlanguage.org ).
3 # Copyright 2014 Alexis Laferrière <alexis.laf@xymus.net>
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at
9 # http://www.apache.org/licenses/LICENSE-2.0
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
17 # Data and DB model of Tnitter
23 # The Tnitter database
33 # Create the needed tables
36 assert create_table
("IF NOT EXISTS users (user TEXT PRIMARY KEY, pass TEXT)") else
37 print error
or else "?"
40 assert create_table
("IF NOT EXISTS posts (user TEXT, text TEXT, posted DATETIME DEFAULT CURRENT_TIMESTAMP)") else
41 print error
or else "?"
45 # Check if the login credentials are valid
47 # If valid, returns the username at database creation time. Otherwise returns `null`.
48 fun check_login
(user
, pass
: String): nullable String
50 var stmt
= select
("user FROM users WHERE lower({user.to_sql_string}) = lower(user) " +
51 "AND {pass.tnitter_hash.to_sql_string} = pass")
52 assert stmt
!= null else print error
or else "?"
54 for row
in stmt
do return row
[0].to_s
58 # Try to sign up a new user, return `true` on success
59 fun sign_up
(user
, pass
: String): Bool
61 # Check if already in user
62 var stmt
= select
("user FROM users WHERE lower({user.to_sql_string}) = lower(user)")
63 assert stmt
!= null else print error
or else "?"
65 if not stmt
.iterator
.to_a
.is_empty
then return false
68 assert insert
("INTO users(user, pass) VALUES ({user.to_sql_string}, {pass.tnitter_hash.to_sql_string})") else
69 print error
or else "?"
76 fun post
(user
, text
: String)
78 assert insert
("INTO posts(user, text) VALUES ({user.to_sql_string}, {text.to_sql_string})") else
79 print error
or else "?"
83 # Get the latest tnits
84 fun latest_posts
(count
: Int): Array[Post]
86 var stmt
= select
("user, text FROM posts ORDER BY datetime(posted) DESC LIMIT {count}")
87 assert stmt
!= null else print error
or else "?"
89 var posts
= new Array[Post]
90 for row
in stmt
do posts
.add
new Post(row
[0].to_s
, row
[1].to_s
)
96 # A single post (or Tnit)
106 # Hash passwords for Tnitter
107 fun tnitter_hash
: String do return (self+salt
).md5
110 # Salt used on passwords
112 # Can be redefed by user modules
113 fun salt
: String do return "tnitter is cool"