Compare commits
3 Commits
cb11018eba
...
25647e3e95
Author | SHA1 | Date |
---|---|---|
Yessiest | 25647e3e95 | |
Yessiest | 3e656163f9 | |
Yessiest | c1e2162ce2 |
|
@ -0,0 +1,3 @@
|
|||
[submodule "hyde"]
|
||||
path = hyde
|
||||
url = /hyde
|
|
@ -0,0 +1,14 @@
|
|||
# The behavior of RuboCop can be controlled via the .rubocop.yml
|
||||
# configuration file. It makes it possible to enable/disable
|
||||
# certain cops (checks) and to alter their behavior if they accept
|
||||
# any parameters. The file can be placed either in your home
|
||||
# directory or in some project directory.
|
||||
#
|
||||
# RuboCop will start looking for the configuration file in the directory
|
||||
# where the inspected file is and continue its way up to the root directory.
|
||||
#
|
||||
# See https://docs.rubocop.org/rubocop/configuration
|
||||
require: standard
|
||||
|
||||
inherit_gem:
|
||||
standard: config/base.yml
|
|
@ -0,0 +1,22 @@
|
|||
---
|
||||
include:
|
||||
- "**/*.rb"
|
||||
exclude:
|
||||
- spec/**/*
|
||||
- test/**/*
|
||||
- vendor/**/*
|
||||
- ".bundle/**/*"
|
||||
require: []
|
||||
domains: []
|
||||
reporters:
|
||||
- rubocop
|
||||
- require_not_found
|
||||
formatter:
|
||||
rubocop:
|
||||
cops: safe
|
||||
except: []
|
||||
only: []
|
||||
extra_args: []
|
||||
require_paths: []
|
||||
plugins: []
|
||||
max_files: 5000
|
|
@ -0,0 +1 @@
|
|||
Subproject commit fd76422b760e6c045581bcb11842ee42221d06e0
|
178
proto.rb
178
proto.rb
|
@ -1,101 +1,83 @@
|
|||
require "date"
|
||||
def validate(*a)
|
||||
# Check arguments against either a list of acceptable classes or just a class
|
||||
raise ArgumentError, "expected an even number of arguments" if a.count%2 != 0
|
||||
for i in (0..a.count-1).step(2) do
|
||||
if a[i+1].kind_of?(Class) then
|
||||
raise TypeError, "expected argument #{i/2} of type #{a[i+1].to_s}" unless a[i].kind_of?(a[i+1])
|
||||
elsif a[i+1].kind_of?(Array) then
|
||||
raise TypeError, "expected argument #{i/2} of any of the types #{a[i+1].to_s}" unless a[i+1].include?(a[i].class)
|
||||
UIDS = {}
|
||||
|
||||
module Heimdall
|
||||
class UID
|
||||
def initialize
|
||||
@UID_prefix = "abstract" if not @UID_prefix
|
||||
id = (1..32).map { |x| (rand()*10).floor }.join
|
||||
while UIDS.has_key? id do
|
||||
id = (1..32).map { |x| (rand()*10).floor }.join
|
||||
end
|
||||
UIDS[@UID_prefix+id] = self
|
||||
@UID = id
|
||||
end
|
||||
attr_reader :UID
|
||||
end
|
||||
|
||||
class UserCache
|
||||
def initialize
|
||||
@users = {}
|
||||
end
|
||||
def add(user)
|
||||
@users[user.protoid] = user
|
||||
end
|
||||
def get(protoid)
|
||||
return @users[user.protoid]
|
||||
end
|
||||
end
|
||||
|
||||
class User < UID
|
||||
def initialize(username, protoid)
|
||||
@username = username
|
||||
@protoid = protoid
|
||||
@UID_prefix = "user"
|
||||
super()
|
||||
end
|
||||
attr_reader :username
|
||||
attr_reader :protoid
|
||||
end
|
||||
|
||||
class Server < UID
|
||||
def initialize
|
||||
@UID_prefix = "server"
|
||||
super()
|
||||
end
|
||||
end
|
||||
|
||||
class Channel < UID
|
||||
def initialize
|
||||
@UID_prefix = "channel"
|
||||
super()
|
||||
end
|
||||
|
||||
end
|
||||
|
||||
class DirectChannel < Channel
|
||||
def initialize
|
||||
@UID_prefix = "dchannel"
|
||||
super()
|
||||
end
|
||||
|
||||
end
|
||||
|
||||
class ServerChannel < Channel
|
||||
def initialize
|
||||
@UID_prefix = "schannel"
|
||||
super()
|
||||
end
|
||||
end
|
||||
|
||||
class Message < UID
|
||||
def initialize
|
||||
@UID_prefix = "message"
|
||||
super()
|
||||
end
|
||||
end
|
||||
|
||||
class MsgStack < UID
|
||||
def initialize
|
||||
@UID_prefix = "msgstack"
|
||||
super()
|
||||
end
|
||||
end
|
||||
end
|
||||
module Heimdall
|
||||
# Core protocol
|
||||
class NetEntity
|
||||
# Abstraction that stores basic object properties
|
||||
@createdAt
|
||||
def initialize()
|
||||
@createdAt = DateTime.new
|
||||
end
|
||||
attr_reader :createdAt
|
||||
end
|
||||
|
||||
class User < NetEntity
|
||||
# User abstraction (not bound to a group chat)
|
||||
@username
|
||||
@nickname
|
||||
def initialize(username,nickname = nil)
|
||||
validate(
|
||||
username, String,
|
||||
nickname, [String,NilClass]
|
||||
)
|
||||
super()
|
||||
nickname = username if not nickname
|
||||
@username = username
|
||||
@nickname = nickname
|
||||
end
|
||||
attr_reader :username
|
||||
attr_accessor :nickname
|
||||
end
|
||||
|
||||
class Channel < NetEntity
|
||||
# Channel abstraction (not bound to a group)
|
||||
# Practically acts as a message stack
|
||||
# Read access to all elements
|
||||
# Write access only to top
|
||||
@messages
|
||||
def initialize()
|
||||
validate(
|
||||
id, String
|
||||
)
|
||||
super()
|
||||
@messages = []
|
||||
end
|
||||
def [](index)
|
||||
@messages[id]
|
||||
end
|
||||
def <<(message)
|
||||
@messages.append(message)
|
||||
end
|
||||
def top(count = 1)
|
||||
return @messages[-1] if count == 1
|
||||
@messages[(-1*count)..-1]
|
||||
end
|
||||
def filter(&block)
|
||||
@messages.filter(block)
|
||||
end
|
||||
def snapshot()
|
||||
@messages.clone
|
||||
end
|
||||
end
|
||||
|
||||
class Message < NetEntity
|
||||
# Message abstraction with edits and content history (not bound to a group)
|
||||
@content
|
||||
@author
|
||||
@editedAt
|
||||
@contentHist
|
||||
def initialize(content,author,channel)
|
||||
validate(
|
||||
content, String,
|
||||
author, Heimdall::User,
|
||||
channel, Heimdall::Channel
|
||||
)
|
||||
super()
|
||||
@editedAt = DateTime.new
|
||||
@contentHist = [content]
|
||||
@content = content
|
||||
@author = author
|
||||
end
|
||||
def edit(content)
|
||||
@content = content
|
||||
@editedAt = DateTime.new
|
||||
@contentHist.append(content)
|
||||
end
|
||||
attr_reader :content
|
||||
attr_reader :author
|
||||
attr_reader :contentHist
|
||||
attr_reader :editedAt
|
||||
end
|
||||
end
|
||||
|
|
128
server.rb
128
server.rb
|
@ -1,98 +1,44 @@
|
|||
# Prototype of the heimdall server
|
||||
$LOAD_PATH << "."
|
||||
$VERSION = "0.1"
|
||||
$PORT = 9128
|
||||
require "proto"
|
||||
require "socket"
|
||||
require_relative "proto"
|
||||
require_relative "hyde/hyde"
|
||||
require "json"
|
||||
|
||||
PROTO_CODES = {
|
||||
:SUCCESS => 0,
|
||||
:BADJSON => 1,
|
||||
:NOMETHOD => 2,
|
||||
:INVPARAM => 3,
|
||||
:NOTIMPL => 4
|
||||
}
|
||||
Users = Heimdall::UserCache.new
|
||||
|
||||
PROTO_MESSAGES = {
|
||||
:SUCCESS => "Done",
|
||||
:BADJSON => "Invalid JSON object",
|
||||
:NOMETHOD => "Invalid method",
|
||||
:INVPARAM => "Invalid parameters",
|
||||
:NOTIMPL => "Not implemented"
|
||||
}
|
||||
|
||||
class Server
|
||||
@channelPool
|
||||
@channelListeners
|
||||
@methods
|
||||
def initialize()
|
||||
@channelPool = {}
|
||||
@channelListeners = {}
|
||||
@methods = [
|
||||
"join",
|
||||
"useradd",
|
||||
"userdel",
|
||||
"userinfo",
|
||||
"lusers",
|
||||
"lchannels",
|
||||
"chantop",
|
||||
"chansend",
|
||||
"chanedit"
|
||||
]
|
||||
end
|
||||
|
||||
def send(client,message)
|
||||
client.puts JSON.dump(message)
|
||||
end
|
||||
|
||||
def err(client,code)
|
||||
client.puts JSON.dump({
|
||||
:error=>PROTO_MESSAGES[code],
|
||||
:code=>PROTO_CODES[code]
|
||||
})
|
||||
client.close
|
||||
end
|
||||
|
||||
def serve(client)
|
||||
# basic validation steps
|
||||
begin
|
||||
# 1. JSON validation
|
||||
data = client.gets
|
||||
data = JSON.load(data)
|
||||
rescue JSON::ParserError
|
||||
err(client,:BADJSON)
|
||||
def require_keys(dict,key_dict)
|
||||
raise KeyError, "not a dict" unless dict.kind_of? Hash
|
||||
key_dict.each_pair { |k,v|
|
||||
unless (dict.has_key? k) and (dict[k].kind_of? v) then
|
||||
raise KeyError, "key #{k} of type #{v} required"
|
||||
end
|
||||
# 2. Method validation
|
||||
method = data[:method]
|
||||
err(client,:NOMETHOD) if not @methods.include?(method)
|
||||
# last but not least: calling the given method
|
||||
if not self.methods.include?(method) then
|
||||
err(client,:NOTIMPL)
|
||||
return
|
||||
}
|
||||
end
|
||||
|
||||
server = Hyde::Server.new Port: 8000 do
|
||||
path "user" do
|
||||
post "new" do |ctx|
|
||||
req,res = ctx.request,ctx.response
|
||||
begin
|
||||
data = JSON::Parser.new(req.body).parse
|
||||
require_keys(data,{
|
||||
username: String,
|
||||
protocol_id: String
|
||||
})
|
||||
Users.add(Heimdall::User.new data[:username], data[:protocol_id])
|
||||
rescue JSON::ParserError => jsonerror
|
||||
res.body = JSON::fast_generate({
|
||||
error: "#{jsonerror}"
|
||||
})
|
||||
res['Content-Type'] = "application/json"
|
||||
res.status = 400
|
||||
rescue KeyError => keyerror
|
||||
res.body = JSON::fast_generate({
|
||||
error: "#{keyerror}"
|
||||
})
|
||||
res['Content-Type'] = "application/json"
|
||||
res.status = 400
|
||||
end
|
||||
end
|
||||
send(client,self.method(method).call(client,data))
|
||||
end
|
||||
|
||||
def join(client,data)
|
||||
chaind = data["chanid"]
|
||||
chanid = rand(89999999)+10000000 if not chanid
|
||||
if not @channelPool[chanid] then
|
||||
@channelPool[chanid] = Heimdall::Channel.new
|
||||
@channelListeners
|
||||
end
|
||||
return {
|
||||
:code=>PROTO_CODE[:SUCCESS],
|
||||
:method=>"join",
|
||||
:chanid=>chanid,
|
||||
}
|
||||
end
|
||||
end
|
||||
|
||||
puts "Starting heimdall v#{$VERSION} server on port #{$PORT}..."
|
||||
server = TCPServer.new 9128
|
||||
loop do
|
||||
Thread.start(server.accept) do |client|
|
||||
puts "Received client connection on #{client.addr}"
|
||||
|
||||
end
|
||||
end
|
||||
server.start
|
||||
|
|
Loading…
Reference in New Issue